wip, trying to get something that could be tested

1 files changed, 30 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/package.scala b/kamon-core/src/main/scala/kamon/package.scala
index 92c48017..b65022bc 100644
--- a/kamon-core/src/main/scala/kamon/package.scala
+++ b/kamon-core/src/main/scala/kamon/package.scala
@@ -1,3 +1,6 @@
+import java.util.concurrent.atomic.AtomicLong
+import java.util.concurrent.{Executors, ThreadFactory}
+
 import com.typesafe.config.Config
 
 import scala.collection.concurrent.TrieMap
@@ -7,6 +10,33 @@ package object kamon {
 
 
   /**
+    * Creates a thread factory that assigns the specified name to all created Threads.
+    */
+  def threadFactory(name: String): ThreadFactory =
+    new ThreadFactory {
+      val defaultFactory = Executors.defaultThreadFactory()
+
+      override def newThread(r: Runnable): Thread = {
+        val thread = defaultFactory.newThread(r)
+        thread.setName(name)
+        thread
+      }
+    }
+
+  def numberedThreadFactory(name: String): ThreadFactory =
+    new ThreadFactory {
+      val count = new AtomicLong()
+      val defaultFactory = Executors.defaultThreadFactory()
+
+      override def newThread(r: Runnable): Thread = {
+        val thread = defaultFactory.newThread(r)
+        thread.setName(name + "-" + count.incrementAndGet().toString)
+        thread
+      }
+    }
+
+
+  /**
     *  Workaround to the non thread-safe [[scala.collection.concurrent.TrieMap#getOrElseUpdate()]] method. More details on
     *  why this is necessary can be found at [[https://issues.scala-lang.org/browse/SI-7943]].
     */