Propogate MDC context into newly created thread pool

ThreadPoolTaskExecutorWithMdcPropagation.java

public class ThreadPoolTaskExecutorWithMdcPropagation extends ThreadPoolExecutor {

    private static final Logger LOGGER = LoggerFactory.getLogger(ThreadPoolTaskExecutorWithMdcPropagation.class);

    private final Map<String, String> parentMdcContextMap;

    public ThreadPoolTaskExecutorWithMdcPropagation(int corePoolSize, int maxPoolSize, long keepAliveTime, TimeUnit keepAliveTimeUnit) {
        super(corePoolSize, maxPoolSize, keepAliveTime, keepAliveTimeUnit, new LinkedBlockingQueue<>());
        parentMdcContextMap = MDC.getCopyOfContextMap();
    }

    @Override
    protected void beforeExecute(Thread workerThread, Runnable runnable) {
        try {
            if (parentMdcContextMap != null && !parentMdcContextMap.isEmpty()) {
                MDC.setContextMap(parentMdcContextMap);
            }
        } catch (Exception e) {
            LOGGER.error("Exception occurred while passing MDC context to thread.", e);
        }
        super.beforeExecute(workerThread, runnable);
    }
}