package com.xzx.gc.common.utils;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.CharsetUtil;
import com.xzx.gc.common.constant.Constants;
import lombok.experimental.UtilityClass;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.MDC;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

@UtilityClass
@Slf4j
public class LogUtils {


    /**
     * trace id
     */
    public static final String TRACE_USER_ID = "USER_ID";
    public static final String TRACE_ORDER_ID = "ORDER_ID";


    /**
     * 写入攻击型IP
     * @param ip   ip
     * @param reason  原因
     * @throws IOException
     */
    public boolean writeIp(String ip,String reason) {
        String urlPrefix=Constants.IMG_UPLOAD_PATH;
        urlPrefix=urlPrefix+"/ip.txt";
        List<String> list=new ArrayList<>();
        boolean exist = FileUtil.exist(urlPrefix);
        if(!exist){
            try {
                new File(urlPrefix).createNewFile();
                log.info("***************初始化记录攻击IP日志文件成功路径:{}*************",urlPrefix);
                list.add("IP地址                                 时间                                原因");
                //日志头
                FileUtil.writeLines(list,urlPrefix,CharsetUtil.CHARSET_UTF_8);
            } catch (IOException e) {
                log.error("记录IP日志文件创建失败",e);
                return  false;
            }
        }

        list.clear();
        list.add(""+ip+"                    "+DateUtil.now() +"                    "+reason+"");
        FileUtil.appendUtf8Lines(list,urlPrefix);
        log.info("记录攻击IP:{}成功,原因：{}",ip,reason);
        return true;
    }


    /**
     * 设置 traceId
     * @param traceId traceId
     */
    public  void setTraceId( String traceId, String value) {
        MDC.put(traceId, value);
    }

    /**
     * 移除 traceId
     */
    public  void removeTraceId( String traceId) {
        MDC.remove(traceId);
    }

    /**
     * 获取批次号
     * @return 批次号
     */
    public  String getTraceId(String traceId) {
        return MDC.get(traceId);
    }

    public void clearTraceId(){
        MDC.clear();
    }

}