如何在线程池中传递Slf4J MDC内容

最新推荐文章于 2025-06-03 16:26:20 发布

原创最新推荐文章于 2025-06-03 16:26:20 发布 · 5.2k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#MDC #线程池 #slf4j #TheadPoolTaskExecutor

编码中遇到的问题同时被 2 个专栏收录

3 篇文章

订阅专栏

springMvc

2 篇文章

订阅专栏

本文介绍了一种在Spring框架中实现线程池的方法，通过继承ThreadPoolTaskExecutor并覆盖其execute和submit方法，实现在任务执行前和执行后对MDC（Mapped Diagnostic Context）的上下文进行操作，确保了在多线程环境中日志记录的一致性和可追溯性。

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

下面给出在spring中如何实现
线程池继承ThreadPoolTaskExecutor

public class MdcThreadPoolTaskExecutor extends ThreadPoolTaskExecutor {
    
        private static final long serialVersionUID = 1L;
        private boolean useFixedContext = false;
        private Map<String, String> fixedContext;
    
        MdcThreadPoolTaskExecutor() {
            super();
        }
    
        public MdcThreadPoolTaskExecutor(Map<String, String> fixedContext) {
            super();
            this.fixedContext = fixedContext;
            useFixedContext = (fixedContext != null);
        }
    
        private Map<String, String> getContextForTask() {
            return useFixedContext ? fixedContext : MDC.getCopyOfContextMap();
        }
    
        /**
         * All executions will have MDC injected. {@code ThreadPoolExecutor}'s submission methods ({@code submit()} etc.)
         * all delegate to this.
         */
        @Override
        public void execute(Runnable command) {
            super.execute(wrapExecute(command, getContextForTask()));
        }
    
        @Override
        public <T> Future<T> submit(Callable<T> task) {
            return super.submit(wrapSubmit(task, getContextForTask()));
        }
    
        private <T> Callable<T> wrapSubmit(Callable<T> task, final Map<String, String> context) {
            return () -> {
                Map<String, String> previous = MDC.getCopyOfContextMap();
                if (context == null) {
                    MDC.clear();
                } else {
                    MDC.setContextMap(context);
                }
                try {
                    return task.call();
                } finally {
                    if (previous == null) {
                        MDC.clear();
                    } else {
                        MDC.setContextMap(previous);
                    }
                }
            };
        }
    
        private Runnable wrapExecute(final Runnable runnable, final Map<String, String> context) {
            return () -> {
                Map<String, String> previous = MDC.getCopyOfContextMap();
                if (context == null) {
                    MDC.clear();
                } else {
                    MDC.setContextMap(context);
                }
                try {
                    runnable.run();
                } finally {
                    if (previous == null) {
                        MDC.clear();
                    } else {
                        MDC.setContextMap(previous);
                    }
                }
            };
        }
    }