Amazon SWF应用SLF4J MDC部分日志缺失问题求助

核心排查方向 #

• 线程上下文切换问题：Amazon SWF的工作流/活动执行可能跨线程调度，而MDC是线程绑定的。若当前线程put了MDC，但后续代码被SWF调度到其他线程执行，新线程不会自动继承原线程的MDC上下文。
• SWF框架的上下文传递缺失：AWS Flow Framework等SWF相关框架不会自动传递MDC，需要手动在活动/子工作流调用时显式传递上下文。
• MDC值被意外清除：排查代码路径中是否存在MDC.clear()、MDC.remove(Constants.MDC_TRACE_ID)这类操作，可能在put值后、打日志前被执行。
• 异步日志配置缺陷：如果使用异步日志Appender，需确保异步线程能继承原线程的MDC上下文，否则会丢失追踪ID。

针对性解决步骤 #

1. 显式传递MDC到SWF活动/子工作流
   在发起活动或子工作流时，将当前MDC的TRACE_ID作为参数传入，在目标执行单元的入口处重新写入MDC：

   // 发起活动时传递traceId
   ActivityOptions options = new ActivityOptions.Builder().build();
   String traceId = MDC.get(Constants.MDC_TRACE_ID);
   activityClient.processActivity(traceId, activityInput);
   
   // 活动入口处恢复MDC
   public void processActivity(String traceId, ActivityInput input) {
       MDC.put(Constants.MDC_TRACE_ID, traceId);
       try {
           log.info("Starting workflow: {}", input);
           // 后续业务逻辑
       } finally {
           // 避免线程池复用导致上下文污染，执行后清除当前MDC项
           MDC.remove(Constants.MDC_TRACE_ID);
       }
   }
   

2. 改造线程池支持MDC上下文继承
   如果使用自定义线程池处理SWF任务，通过包装Runnable实现MDC上下文传递：

   public class MdcWrappedRunnable implements Runnable {
       private final Runnable target;
       private final Map<String, String> mdcSnapshot;
   
       public MdcWrappedRunnable(Runnable target) {
           this.target = target;
           this.mdcSnapshot = MDC.getCopyOfContextMap();
       }
   
       @Override
       public void run() {
           Map<String, String> originalContext = MDC.getCopyOfContextMap();
           try {
               if (mdcSnapshot != null) {
                   MDC.setContextMap(mdcSnapshot);
               }
               target.run();
           } finally {
               // 恢复原线程的MDC上下文
               if (originalContext != null) {
                   MDC.setContextMap(originalContext);
               } else {
                   MDC.clear();
               }
           }
       }
   }
   

   提交任务时用该类包裹原Runnable即可。

3. 修正异步日志的MDC配置
   以Logback为例，配置AsyncAppender时开启MDC复制：

   <appender name="ASYNC_LOG" class="ch.qos.logback.classic.AsyncAppender">
       <appender-ref ref="STDOUT"/>
       <includeCallerData>true</includeCallerData>
       <mdcCopy>true</mdcCopy>
   </appender>
   

4. 排查代码中的MDC清除操作
   全局搜索MDC.clear()、MDC.remove调用，检查目标代码路径中是否有意外清除逻辑。可以在MDC.put后立即打印MDC.get(Constants.MDC_TRACE_ID)，确认值是否存在，再逐步排查后续代码对MDC的修改。

内容的提问来源于stack exchange，提问作者T.G.