CountDownLatch 无法重写，正确做法是用 TraceableExecutorService 自动传递 MDC traceId 或手动快照还原 contextMap，确保子线程启动时 traceId 已就位。
CountDownLatch 本身是不可重写核心逻辑的 final 类，它基于 AQS 实现，内部状态（state）完全封装，不提供钩子、不开放继承、也不支持装饰器式“包裹类”来拦截 countDown() 或 await() 的执行路径。所谓“重写 CountDownLatch 的核心包裹类”在 Java 并发 API 层面不存在合法、安全、可维护的实现方式。
但你的实际诉求很明确：
✅ 在多线程协作场景中（如用 CountDownLatch 等待多个异步任务完成），
✅ 保证子线程能继承并延续主线程的 traceId（例如 MDC 中的 trace-id），
✅ 并且这个传递必须在计数归零前就完成——即子线程启动时、执行业务前，traceId 就已就位，而非事后补救。
这不是靠“改造 CountDownLatch”，而是靠正确传递上下文 + 合理使用线程执行单元。以下是真正可行、生产验证过的做法：
✅ 正确做法一：用 TraceableExecutorService 替代裸线程池（推荐）
Spring Cloud Sleuth / Micrometer Tracing 已内置支持：
- 当你注册一个
ThreadPoolTaskExecutorBean，Sleuth 会自动将其包装为TraceableExecutorService； - 所有通过
execute()/submit()提交的任务，都会自动复制当前线程的 MDC（含 traceId）到子线程； -
CountDownLatch只负责同步，不再承担上下文传递职责。
@Bean
public TaskExecutor traceableExecutor() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(4);
executor.setMaxPoolSize(10);
executor.setThreadNamePrefix("traceable-");
executor.initialize();
return executor; // Sleuth 自动增强
}使用时：
CountDownLatch latch = new CountDownLatch(3);
for (int i = 0; i < 3; i++) {
executor.submit(() -> {
// ✅ 此处 MDC 已含主线程 traceId
log.info("子任务执行中..."); // 日志自动带 trace-id
doWork();
latch.countDown();
});
}
latch.await(); // 主线程等待✅ 正确做法二：手动透传 MDC（无 Spring 环境也适用）
若未用 Sleuth，可在提交任务前快照 MDC，并在子线程中显式还原：
Map<String, String> contextMap = MDC.getCopyOfContextMap(); // 快照
executor.submit(() -> {
if (contextMap != null) {
MDC.setContextMap(contextMap); // 还原
}
try {
log.info("子任务执行（traceId 已就位）");
doWork();
} finally {
MDC.clear(); // 避免线程复用污染
}
latch.countDown();
});⚠️ 注意：必须在 countDown() 前完成日志或业务调用，确保 traceId 在关键路径上生效。
❌ 不可行方案（需规避）
-
继承/代理 CountDownLatch：类是 final，且
await()/countDown()不触发可扩展回调，强行反射修改state或 AQS 队列会破坏线程安全性； -
在
await()返回后补 traceId：此时子线程早已执行完毕，traceId 注入失去意义； -
用
InheritableThreadLocal覆盖默认行为：JDK 8+ 中ThreadPoolExecutor默认不继承InheritableThreadLocal，需自定义ThreadFactory，但不如 MDC 显式可靠； -
修改
CountDownLatch源码重新编译：违反 JDK 合规性，无法升级，极易引发 AQS 状态紊乱。
✅ 补充：确保 traceId 在 await 期间也可见（可选）
如果主线程在 latch.await() 期间也要打日志（比如超时监控），记得 MDC 是线程局部的，主线程的 traceId 始终存在，无需额外操作。
如需超时保护，直接用带参 await(long, TimeUnit) 即可：
if (!latch.await(10, TimeUnit.SECONDS)) {
log.warn("等待子任务超时，当前 traceId: {}", MDC.get("trace-id"));
}不复杂但容易忽略。
