aboutsummaryrefslogtreecommitdiff
path: root/kamon-core/src/main/scala/kamon/metric/Scaler.scala
blob: f8f51c00cb25bc604edf0c909496d10a1d5fe8c8 (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
package kamon.metric

import kamon.util.MeasurementUnit
import kamon.util.MeasurementUnit.Dimension

class Scaler(targetTimeUnit: MeasurementUnit, targetInformationUnit: MeasurementUnit, dynamicRange: DynamicRange) {
  require(targetTimeUnit.dimension == Dimension.Time, "timeUnit must be in the time dimension.")
  require(targetInformationUnit.dimension == Dimension.Information, "informationUnit must be in the information dimension.")

  val scaleHistogram = new HdrHistogram("scaler", Map.empty, MeasurementUnit.none, dynamicRange)

  def scaleDistribution(metric: MetricDistribution): MetricDistribution = {
    metric.measurementUnit match {
      case MeasurementUnit(Dimension.Time, magnitude) if(magnitude != targetTimeUnit.magnitude) =>
        scaleMetricDistributionToTarget(metric, targetTimeUnit)

      case MeasurementUnit(Dimension.Information, magnitude) if(magnitude != targetTimeUnit.magnitude) =>
        scaleMetricDistributionToTarget(metric, targetInformationUnit)

      case _ => metric
    }
  }

  def scaleMetricValue(metric: MetricValue): MetricValue = {
    metric.measurementUnit match {
      case MeasurementUnit(Dimension.Time, magnitude) if(magnitude != targetTimeUnit.magnitude) =>
        scaleMetricValueToTarget(metric, targetTimeUnit)

      case MeasurementUnit(Dimension.Information, magnitude) if(magnitude != targetTimeUnit.magnitude) =>
        scaleMetricValueToTarget(metric, targetInformationUnit)

      case _ => metric
    }
  }

  private def scaleMetricDistributionToTarget(metric: MetricDistribution, targetUnit: MeasurementUnit): MetricDistribution = {
    metric.distribution.bucketsIterator.foreach(b => {
      val scaledValue = MeasurementUnit.scale(b.value, metric.measurementUnit, targetUnit)
      scaleHistogram.record(Math.ceil(scaledValue).toLong, b.frequency)
    })

    scaleHistogram.snapshot().copy(
      name = metric.name,
      tags = metric.tags,
      measurementUnit = targetUnit,
      dynamicRange = dynamicRange
    )
  }

  private def scaleMetricValueToTarget(metric: MetricValue, targetUnit: MeasurementUnit): MetricValue = {
    val scaledValue = MeasurementUnit.scale(metric.value, metric.measurementUnit, targetUnit)

    metric.copy(
      value = Math.ceil(scaledValue).toLong,
      measurementUnit = targetUnit
    )
  }
}