blob: 9f8db3ef6305fa513f05017275c8808ab491cec9 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
/** Code ported from "de.geekonaut" %% "slickmdc" % "1.0.0"
* License: @see https://github.com/AVGP/slickmdc/blob/master/LICENSE
* Blog post: @see http://50linesofco.de/post/2016-07-01-slick-and-slf4j-mdc-logging-in-scala.html
*/
package xyz.driver.core.logging
import org.slf4j.MDC
import scala.concurrent.ExecutionContext
/**
* Execution context proxy for propagating SLF4J diagnostic context from caller thread to execution thread.
*/
class MdcExecutionContext(executionContext: ExecutionContext) extends ExecutionContext {
override def execute(runnable: Runnable): Unit = {
val callerMdc = MDC.getCopyOfContextMap
executionContext.execute(new Runnable {
def run(): Unit = {
// copy caller thread diagnostic context to execution thread
// scalastyle:off
if (callerMdc != null) MDC.setContextMap(callerMdc)
try {
runnable.run()
} finally {
// the thread might be reused, so we clean up for the next use
MDC.clear()
}
}
})
}
override def reportFailure(cause: Throwable): Unit = executionContext.reportFailure(cause)
}
|