From 0000a65ab4479a2a40e2d6468036438e9705b4aa Mon Sep 17 00:00:00 2001
From: vlad <vlad@driver.xyz>
Date: Tue, 13 Jun 2017 10:25:55 -0700
Subject: Initial extraction of Driver non-specific utilities

---
 .../common/concurrent/MdcExecutionContext.scala    | 35 ++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 src/main/scala/xyz/driver/common/concurrent/MdcExecutionContext.scala

(limited to 'src/main/scala/xyz/driver/common/concurrent/MdcExecutionContext.scala')

diff --git a/src/main/scala/xyz/driver/common/concurrent/MdcExecutionContext.scala b/src/main/scala/xyz/driver/common/concurrent/MdcExecutionContext.scala
new file mode 100644
index 0000000..cd2b394
--- /dev/null
+++ b/src/main/scala/xyz/driver/common/concurrent/MdcExecutionContext.scala
@@ -0,0 +1,35 @@
+package xyz.driver.common.concurrent
+
+import org.slf4j.MDC
+
+import scala.concurrent.{ExecutionContext, ExecutionContextExecutor}
+
+object MdcExecutionContext {
+  def from(orig: ExecutionContext): ExecutionContext = new MdcExecutionContext(orig)
+}
+
+class MdcExecutionContext(orig: ExecutionContext) extends ExecutionContextExecutor {
+
+  def execute(runnable: Runnable): Unit = {
+    val parentMdcContext = MDC.getCopyOfContextMap
+
+    orig.execute(new Runnable {
+      def run(): Unit = {
+        val saveMdcContext = MDC.getCopyOfContextMap
+        setContextMap(parentMdcContext)
+
+        try {
+          runnable.run()
+        } finally {
+          setContextMap(saveMdcContext)
+        }
+      }
+    })
+  }
+
+  private[this] def setContextMap(context: java.util.Map[String, String]): Unit =
+    Option(context).fold(MDC.clear())(MDC.setContextMap)
+
+  def reportFailure(t: Throwable): Unit = orig.reportFailure(t)
+
+}
-- 
cgit v1.2.3