Merge pull request #223 from jtjeferreira/customTraceTokenGenerator

+ core: kamon.trace.token-generator conf

1 files changed, 14 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/trace/Sampler.scala b/kamon-core/src/main/scala/kamon/trace/Sampler.scala
index 5abba221..ac780d5e 100644
--- a/kamon-core/src/main/scala/kamon/trace/Sampler.scala
+++ b/kamon-core/src/main/scala/kamon/trace/Sampler.scala
@@ -16,9 +16,14 @@
 
 package kamon.trace
 
+import java.net.InetAddress
+import java.util.concurrent.atomic.AtomicLong
+
 import kamon.util.{ NanoInterval, Sequencer }
 import scala.concurrent.forkjoin.ThreadLocalRandom
 
+import scala.util.Try
+
 trait Sampler {
   def shouldTrace: Boolean
   def shouldReport(traceElapsedTime: NanoInterval): Boolean
@@ -71,3 +76,12 @@ class ThresholdSampler(thresholdInNanoseconds: Long) extends Sampler {
   def shouldReport(traceElapsedTime: NanoInterval): Boolean = traceElapsedTime.nanos >= thresholdInNanoseconds
 }
 
+class DefaultTokenGenerator extends Function0[String] {
+  private val _hostnamePrefix = Try(InetAddress.getLocalHost.getHostName).getOrElse("unknown-localhost")
+  private val _tokenCounter = new AtomicLong
+
+  def apply(): String = {
+    _hostnamePrefix + "-" + String.valueOf(_tokenCounter.incrementAndGet())
+  }
+}
+