From 594c7a1729789eae7037918cde7287bdc4111b70 Mon Sep 17 00:00:00 2001
From: Ivan Topolnjak <ivantopo@gmail.com>
Date: Wed, 3 Dec 2014 02:10:46 +0100
Subject: = core: first simple approach to providing traces and a subscription
 mechanism.

---
 .../src/main/scala/kamon/trace/Sampler.scala       | 45 ++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 kamon-core/src/main/scala/kamon/trace/Sampler.scala

(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
new file mode 100644
index 00000000..60a400f8
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/trace/Sampler.scala
@@ -0,0 +1,45 @@
+package kamon.trace
+
+import java.util.concurrent.atomic.AtomicLong
+
+import scala.concurrent.forkjoin.ThreadLocalRandom
+
+trait Sampler {
+  def shouldTrace: Boolean
+  def shouldReport(traceElapsedNanoTime: Long): Boolean
+}
+
+object NoSampling extends Sampler {
+  def shouldTrace: Boolean = false
+  def shouldReport(traceElapsedNanoTime: Long): Boolean = false
+}
+
+object SampleAll extends Sampler {
+  def shouldTrace: Boolean = true
+  def shouldReport(traceElapsedNanoTime: Long): Boolean = true
+}
+
+class RandomSampler(chance: Int) extends Sampler {
+  require(chance > 0, "kamon.trace.random-sampler.chance cannot be <= 0")
+  require(chance <= 100, "kamon.trace.random-sampler.chance cannot be > 100")
+
+  def shouldTrace: Boolean = ThreadLocalRandom.current().nextInt(100) <= chance
+  def shouldReport(traceElapsedNanoTime: Long): Boolean = true
+}
+
+class OrderedSampler(interval: Int) extends Sampler {
+  require(interval > 0, "kamon.trace.ordered-sampler.interval cannot be <= 0")
+
+  private val counter = new AtomicLong(0L)
+  def shouldTrace: Boolean = counter.incrementAndGet() % interval == 0
+  // TODO: find a more efficient way to do this, protect from long overflow.
+  def shouldReport(traceElapsedNanoTime: Long): Boolean = true
+}
+
+class ThresholdSampler(thresholdInNanoseconds: Long) extends Sampler {
+  require(thresholdInNanoseconds > 0, "kamon.trace.threshold-sampler.minimum-elapsed-time cannot be <= 0")
+
+  def shouldTrace: Boolean = true
+  def shouldReport(traceElapsedNanoTime: Long): Boolean = traceElapsedNanoTime >= thresholdInNanoseconds
+}
+
-- 
cgit v1.2.3