add metric scaler

1 files changed, 58 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/util/MetricScaler.scala b/kamon-core/src/main/scala/kamon/util/MetricScaler.scala
new file mode 100644
index 00000000..ca77a9a1
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/util/MetricScaler.scala
@@ -0,0 +1,58 @@
+package kamon.util
+
+import kamon.metric.{DynamicRange, HdrHistogram, MetricDistribution, MetricValue}
+import kamon.util.MeasurementUnit.Dimension
+
+class MetricScaler(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
+    )
+  }
+}