Spring Boot未来是否支持TraceId与MDC向虚拟线程的传播？

核心结论 #

目前Spring Boot官方尚未将虚拟线程的TraceId/MDC自动传播功能纳入明确的开发 roadmap，你需要自行实现上下文传播逻辑，同时可关注Spring Boot官方的后续更新。

自行实现的可行方案 #

1. 针对StructuredTaskScope.fork()

手动捕获当前线程的MDC上下文和TraceId，在fork的任务执行前后手动注入和清理：

// 捕获当前上下文
String traceId = MDC.get("traceId");
Map<String, String> mdcContext = MDC.getCopyOfContextMap();

// 在fork任务中注入上下文
try (var scope = new StructuredTaskScope.ShutdownOnFailure()) {
    scope.fork(() -> {
        MDC.setContextMap(mdcContext);
        try {
            // 执行下游REST调用逻辑
            return downstreamServiceA.call();
        } finally {
            MDC.clear();
        }
    });

    scope.fork(() -> {
        MDC.setContextMap(mdcContext);
        try {
            // 执行另一个下游REST调用逻辑
            return downstreamServiceB.call();
        } finally {
            MDC.clear();
        }
    });

    scope.join();
    scope.throwIfFailed();
}

可以将上下文捕获、注入、清理的逻辑封装成工具类，减少重复代码。

2. 针对Executors.newVirtualThreadPerTaskExecutor()

自定义ExecutorService包装类，在提交任务时自动携带MDC上下文：

ExecutorService virtualExecutor = Executors.newVirtualThreadPerTaskExecutor();
ExecutorService contextPropagatingExecutor = new ExecutorService() {
    @Override
    public void execute(Runnable command) {
        Map<String, String> currentMdc = MDC.getCopyOfContextMap();
        virtualExecutor.execute(() -> {
            MDC.setContextMap(currentMdc);
            try {
                command.run();
            } finally {
                MDC.clear();
            }
        });
    }

    @Override
    public <T> Future<T> submit(Callable<T> task) {
        Map<String, String> currentMdc = MDC.getCopyOfContextMap();
        return virtualExecutor.submit(() -> {
            MDC.setContextMap(currentMdc);
            try {
                return task.call();
            } finally {
                MDC.clear();
            }
        });
    }

    @Override
    public <T> Future<T> submit(Runnable task, T result) {
        Map<String, String> currentMdc = MDC.getCopyOfContextMap();
        return virtualExecutor.submit(() -> {
            MDC.setContextMap(currentMdc);
            try {
                task.run();
                return result;
            } finally {
                MDC.clear();
            }
        });
    }

    @Override
    public Future<?> submit(Runnable task) {
        Map<String, String> currentMdc = MDC.getCopyOfContextMap();
        return virtualExecutor.submit(() -> {
            MDC.setContextMap(currentMdc);
            try {
                task.run();
            } finally {
                MDC.clear();
            }
        });
    }

    // invokeAll、invokeAny等方法可按需按相同逻辑实现
};

Spring Boot未来支持情况 #

Spring框架团队已经关注到虚拟线程的上下文传播需求，包括ThreadLocal和MDC的自动传播，但截至Spring Boot 3.2.x版本，尚未提供内置支持。后续可以通过关注Spring Boot的官方issue跟踪列表和版本发布说明，了解相关功能的迭代进度。

内容的提问来源于stack exchange，提问作者michaldo