From 7e74030ea824a6b6a561880103f69f3e8e409bba Mon Sep 17 00:00:00 2001 From: João Ferreira Date: Fri, 26 Jun 2015 17:47:00 +0100 Subject: + core: kamon.trace.token-generator conf --- kamon-core/src/main/scala/kamon/trace/Sampler.scala | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'kamon-core/src/main/scala/kamon/trace/Sampler.scala') 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()) + } +} + -- cgit v1.2.3