aboutsummaryrefslogtreecommitdiff
path: root/kamon-newrelic/src/main/scala/kamon/newrelic/Metric.scala
blob: 6fb645f52b79d4627d46749f79d1475478768b8f (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
package kamon.newrelic

import kamon.metric.instrument._
import kamon.metric.MetricKey
import kamon.util.{ MapMerge, Timestamp }

case class MetricID(name: String, scope: Option[String])
case class MetricData(callCount: Long, total: Double, totalExclusive: Double, min: Double, max: Double, sumOfSquares: Double) {
  def merge(that: MetricData): MetricData =
    MetricData(
      callCount + that.callCount,
      total + that.total,
      totalExclusive + that.totalExclusive,
      math.min(min, that.min),
      math.max(max, that.max),
      sumOfSquares + that.sumOfSquares)
}

object Metric {

  def scaleFunction(uom: UnitOfMeasurement): Double  Double = uom match {
    case time: Time  time.scale(Time.Seconds)
    case other       a  a
  }

  def apply(snapshot: InstrumentSnapshot, snapshotUnit: UnitOfMeasurement, name: String, scope: Option[String]): Metric = {
    snapshot match {
      case hs: Histogram.Snapshot 
        var total: Double = 0D
        var sumOfSquares: Double = 0D
        val scaler = scaleFunction(snapshotUnit)

        val scaledMin = scaler(hs.min)
        val scaledMax = scaler(hs.max)

        hs.recordsIterator.foreach { record 
          val scaledValue = scaler(record.level)

          total += scaledValue * record.count
          sumOfSquares += (scaledValue * scaledValue) * record.count
        }

        (MetricID(name, scope), MetricData(hs.numberOfMeasurements, total, total, scaledMin, scaledMax, sumOfSquares))

      case cs: Counter.Snapshot 
        (MetricID(name, scope), MetricData(cs.count, cs.count, cs.count, 0, cs.count, cs.count * cs.count))
    }
  }
}

case class TimeSliceMetrics(from: Timestamp, to: Timestamp, metrics: Map[MetricID, MetricData]) {
  import MapMerge.Syntax

  def merge(that: TimeSliceMetrics): TimeSliceMetrics = {
    val mergedFrom = Timestamp.earlier(from, that.from)
    val mergedTo = Timestamp.later(to, that.to)
    val mergedMetrics = metrics.merge(that.metrics, (l, r)  l.merge(r))

    TimeSliceMetrics(mergedFrom, mergedTo, mergedMetrics)
  }
}

case class MetricBatch(runID: Long, timeSliceMetrics: TimeSliceMetrics)