package com.iMagine.iMagine_pro.aop;

import com.iMagine.iMagine_common.utils.UUIDUtil;
import com.iMagine.iMagine_pro.utils.TokenUtil;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.MDC;
import org.springframework.stereotype.Component;


/**
 * @author 名一
 * @ClassName TraceIdAspect
 * @description: 日志請(qǐng)求鏈路id切面處理
 * @datetime 2024年 12月 16日 14:28
 * @version: 1.0
 */
@Slf4j
@Aspect
@Component
public class TraceIdAspect {
    
    /** 鏈路追蹤id */
    public final static String TRACE_ID = "TRACE_ID";
    /** 用戶(hù) */
    public final static String USER = "USER";
    
    /**
     * 鏈路id切點(diǎn)定義
     */
    @Pointcut("execution(* com.iMagine.iMagine_pro.controller.*.*(..))")
    public void TraceIdCut() {

    }

    /**
     * 鏈路id添加
     */
    @Before("TraceIdCut()")
    public void cutProcessBefore() {
        MDC.put(TRACE_ID, UUIDUtil.getUUID());
        //以下代碼為記錄用戶(hù)信息，方便更直觀(guān)的識(shí)別日志操作人信息。若不需要可以將下面部分刪除
        String nickname = TokenUtil.getNicknameByToken();
        if (null == nickname){
            nickname = "游客訪(fǎng)問(wèn)";
        }
        MDC.put(USER, nickname);
    }

    /**
     * 鏈路id清除
     */
    @After("TraceIdCut()")
    public void cutProcessAfter() {
        MDC.clear();
    }
}

package com.iMagine.iMagine_common.utils;

import java.util.UUID;

/**
 * @author 名一
 * @ClassName UUIDUtil
 * @description: UUID工具類(lèi)
 * @datetime 2024年 04月 23日 11:49
 * @version: 1.0
 */
public class UUIDUtil {

    /**
     * 獲取UUID
     *
     * @return
     */
    public static String getUUID() {
        //生產(chǎn)uuid并去掉uuid的短橫線(xiàn)
        return UUID.randomUUID().toString().replace("-", "");
    }
}

package com.iMagine.iMagine_common.utils;

import com.iMagine.iMagine_common.constant.SysConstant;
import org.slf4j.MDC;

import java.util.Map;

/**
 * @author 名一
 * @ClassName ThreadMdcUtil
 * @description: 多線(xiàn)程鏈路追蹤工具類(lèi)
 * @datetime 2024年 12月 16日 14:57
 * @version: 1.0
 */
public class ThreadMdcUtil {

    // 獲取唯一性標(biāo)識(shí)
    public static String generateTraceId() {
        return UUIDUtil.getUUID();
    }

    public static void setTraceIdIfAbsent() {
        if (MDC.get(SysConstant.TRACE_ID) == null) {
            MDC.put(SysConstant.TRACE_ID, generateTraceId());
        }
    }

    /**
     * 用于父線(xiàn)程向線(xiàn)程池中提交任務(wù)時(shí)，將自身MDC中的數(shù)據(jù)復(fù)制給子線(xiàn)程
     *
     * @param runnable  要執(zhí)行的線(xiàn)程
     * @param context   父線(xiàn)程的mdc
     * @return          鏈路id傳遞后的任務(wù)
     */
    public static Runnable wrap(final Runnable runnable, final Map<String, String> context) {
        return () -> {
            if (context == null) {
                MDC.clear();
            } else {
                MDC.setContextMap(context);
            }
            setTraceIdIfAbsent();
            try {
                runnable.run();
            } finally {
                MDC.clear();
            }
        };
    }
}

/**
 * 線(xiàn)程池添加 [ai server 相關(guān)操作(psot)]任務(wù)
 *
 * @param task 添加的任務(wù)
 */
public static void addSendPostThreadToThreadPool(SendPostThread task) {
    log.info("addSendPostThreadToThreadPool pool:{}, task:{}", pool, task);
    try {
        pool.execute(ThreadMdcUtil.wrap(task, MDC.getCopyOfContextMap()));
    } catch (Exception e) {
        log.error("addSendPostThreadToThreadPool error pool:{}, task:{}, e:", pool, task, e);
    }
}

<!-- 日志輸出格式 （其中%X{TRACE_ID}是mdc里邊鏈路請(qǐng)求id的key，user-%X{USER}是用戶(hù)信息的key）-->
<property name="ENCODER_PATTERN" value="%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %X{TRACE_ID} user-%X{USER} %-5level %logger{32}-[%line] - %msg%n"/>