aboutsummaryrefslogtreecommitdiff
path: root/kamon-core/src/main/scala/kamon/metric/instrument/MinMaxCounter.scala
blob: 70094b7b7afc703bd55b0599128f9ae4192ca4f2 (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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package kamon.metric.instrument

import java.lang.Math.abs
import java.time.Duration
import java.util.concurrent.atomic.{AtomicLong, AtomicReference}

import kamon.jsr166.LongMaxUpdater
import kamon.util.MeasurementUnit

trait MinMaxCounter {
  def dynamicRange: DynamicRange
  def sampleInterval: Duration
  def measurementUnit: MeasurementUnit

  def increment(): Unit
  def increment(times: Long): Unit
  def decrement(): Unit
  def decrement(times: Long): Unit
  def sample(): Unit
}


class PaddedMinMaxCounter(name: String, tags: Map[String, String], underlyingHistogram: Histogram with DistributionSnapshotInstrument,
    val sampleInterval: Duration) extends SnapshotableMinMaxCounter {

  private val min = new LongMaxUpdater(0L)
  private val max = new LongMaxUpdater(0L)
  private val sum = new AtomicLong()

  def dynamicRange: DynamicRange =
    underlyingHistogram.dynamicRange

  def measurementUnit: MeasurementUnit =
    underlyingHistogram.measurementUnit

  private[kamon] def snapshot(): DistributionSnapshot =
    underlyingHistogram.snapshot()

  def increment(): Unit =
    increment(1L)

  def increment(times: Long): Unit = {
    val currentValue = sum.addAndGet(times)
    max.update(currentValue)
  }

  def decrement(): Unit =
    decrement(1L)

  def decrement(times: Long): Unit = {
    val currentValue = sum.addAndGet(-times)
    min.update(-currentValue)
  }

  def sample(): Unit = {
    val currentValue = {
      val value = sum.get()
      if (value <= 0) 0 else value
    }

    val currentMin = {
      val rawMin = min.maxThenReset(-currentValue)
      if (rawMin >= 0)
        0
      else
        abs(rawMin)
    }

    val currentMax = max.maxThenReset(currentValue)

    underlyingHistogram.record(currentValue)
    underlyingHistogram.record(currentMin)
    underlyingHistogram.record(currentMax)
  }
}