blob: 5939efb3d35fea4d80c85cb019664dbd43992710 (
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
/** 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
package database
import java.util.concurrent._
import java.util.concurrent.atomic.AtomicInteger
import scala.concurrent._
import com.typesafe.scalalogging.StrictLogging
import slick.util.AsyncExecutor
import logging.MdcExecutionContext
/** Taken from the original Slick AsyncExecutor and simplified
* @see https://github.com/slick/slick/blob/3.1/slick/src/main/scala/slick/util/AsyncExecutor.scala
*/
object MdcAsyncExecutor extends StrictLogging {
/** Create an AsyncExecutor with a fixed-size thread pool.
*
* @param name The name for the thread pool.
* @param numThreads The number of threads in the pool.
*/
def apply(name: String, numThreads: Int): AsyncExecutor = {
new AsyncExecutor {
val tf = new DaemonThreadFactory(name + "-")
lazy val executionContext = {
new MdcExecutionContext(ExecutionContext.fromExecutor(Executors.newFixedThreadPool(numThreads, tf)))
}
def close(): Unit = {}
}
}
def default(name: String = "AsyncExecutor.default"): AsyncExecutor = apply(name, 20)
private class DaemonThreadFactory(namePrefix: String) extends ThreadFactory {
private[this] val group =
Option(System.getSecurityManager).fold(Thread.currentThread.getThreadGroup)(_.getThreadGroup)
private[this] val threadNumber = new AtomicInteger(1)
def newThread(r: Runnable): Thread = {
val t = new Thread(group, r, namePrefix + threadNumber.getAndIncrement, 0)
if (!t.isDaemon) t.setDaemon(true)
if (t.getPriority != Thread.NORM_PRIORITY) t.setPriority(Thread.NORM_PRIORITY)
t
}
}
}
|