异步线程池内部MDC仍继承主线程值的解决方案

最新推荐文章于 2025-07-22 20:00:31 发布

weixin_34121282

最新推荐文章于 2025-07-22 20:00:31 发布

阅读量6.3k

点赞数 1

CC 4.0 BY-SA版权

文章标签： java python runtime

原文链接：https://my.oschina.net/u/3434392/blog/2885718

本文探讨了在Spring中使用@Sync注解时，如何解决ThreadPoolTaskExecutor在切换线程时无法继承MDC上下文的问题。通过分析源码，提出了解决方案——重写ThreadPoolTaskExecutor并在callable执行前注入父线程的MDC内容。此外，文章还简要介绍了AsyncAnnotationBeanPostProcessor和AsyncExecutionInterceptor在异步处理中的作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2019独角兽企业重金招聘Python工程师标准>>>

增加日志输出标识


import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;


@Retention(RetentionPolicy.RUNTIME)
@Target({ ElementType.METHOD })
public @interface MDC {

}

切面：

import java.util.UUID;

import lombok.extern.slf4j.Slf4j;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.slf4j.MDC;

@Slf4j
public abstract class MdcInterceptor {
    protected final static String LOG_ID = "logId";

    @Pointcut("execution(* com.noob.api..*.*(..)) || @annotation(com.noob.aspectj.MDC)")
    public void pointCut() {
        // do nothings
    };

    @Around(value = "pointCut()")
    public Object invoke(ProceedingJoinPoint point) throws Throwable {
        Object result;
        try {
            setTradeId();
            result = point.proceed(point.getArgs());
            removeTraceId();
        } catch (Throwable throwable) {
            throw throwable;
        } finally {
            removeTraceId();
        }
        return result;
    }

    /**
     * 璁剧疆traceId
     */
    public static void setTradeId() {
        try {
            MDC.put(LOG_ID, UUID.randomUUID().toString().replace("-", ""));
        } catch (Exception e) {
            log.error("set log no exception", e);
        }
    }

    /**
     * remove traceId
     */
    public static void removeTraceId() {
        try {
            MDC.remove(LOG_ID);
        } catch (Exception e) {
            log.error("remove log no exception", e);
        }
    }
}

@Sync继承MDC上下文

在使用@Sync时，Spring默认提供的TaskExecutor下的ThreadPoolTaskExecutor无法支持MDC在线程切换时上下文内容的继承。因为MDC本质上使用ThreadLocal来保存上下文。

需要注意的是：切面的织入一定要符合【外部调入】的原则！

解决方法：

重写ThreadPoolTaskExecutor，在callable(真实处理过程)前注入父线程内容。

import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.Future;
import java.util.UUID;
import lombok.extern.slf4j.Slf4j;

import org.apache.commons.collections.MapUtils;
import org