1 files changed, 86 insertions, 0 deletions
diff --git a/kamon-core/src/main/scala/kamon/metrics/HighDynamicRangeRecorder.scala b/kamon-core/src/main/scala/kamon/metrics/HighDynamicRangeRecorder.scala
new file mode 100644
index 00000000..e31d0e11
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/metrics/HighDynamicRangeRecorder.scala
@@ -0,0 +1,86 @@
+/*
+ * =========================================================================================
+ * Copyright © 2013 the kamon project <http://kamon.io/>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+ * except in compliance with the License. You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the
+ * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+ * either express or implied. See the License for the specific language governing permissions
+ * and limitations under the License.
+ * =========================================================================================
+ */
+
+package org.HdrHistogram
+
+import java.util.concurrent.atomic.AtomicLongFieldUpdater
+import scala.annotation.tailrec
+import kamon.metrics.{ DefaultMetricSnapshot, MetricSnapshot, MetricRecorder }
+import com.typesafe.config.Config
+import org.HdrHistogram.HighDynamicRangeRecorder.Configuration
+
+/**
+ *  This implementation aims to be used for real time data collection where data snapshots are taken often over time.
+ *  The snapshotAndReset() operation extracts all the recorded values from the histogram and resets the counts, but still
+ *  leave it in a consistent state even in the case of concurrent modification while the snapshot is being taken.
+ */
+class HighDynamicRangeRecorder(configuration: Configuration)
+    extends AtomicHistogram(1L, configuration.highestTrackableValue, configuration.significantValueDigits) with MetricRecorder {
+
+  import HighDynamicRangeRecorder.totalCountUpdater
+
+  def record(value: Long): Unit = recordValue(value)
+
+  def collect(): MetricSnapshot = {
+    val entries = Vector.newBuilder[MetricSnapshot.Measurement]
+    val countsLength = counts.length()
+
+    @tailrec def iterate(index: Int, previousValue: Long, nrOfRecordings: Long, bucketLimit: Long, increment: Long): Long = {
+      if (index < countsLength) {
+        val currentValue = previousValue + increment
+        val countAtValue = counts.getAndSet(index, 0)
+
+        if (countAtValue > 0)
+          entries += MetricSnapshot.Measurement(currentValue, countAtValue)
+
+        if (currentValue == bucketLimit)
+          iterate(index + 1, currentValue, nrOfRecordings + countAtValue, (bucketLimit << 1) + 1, increment << 1)
+        else
+          iterate(index + 1, currentValue, nrOfRecordings + countAtValue, bucketLimit, increment)
+      } else {
+        nrOfRecordings
+      }
+    }
+
+    val nrOfRecordings = iterate(0, -1, 0, subBucketMask, 1)
+
+    def tryUpdateTotalCount: Boolean = {
+      val previousTotalCount = getTotalCount
+      val newTotalCount = previousTotalCount - nrOfRecordings
+
+      totalCountUpdater.compareAndSet(this, previousTotalCount, newTotalCount)
+    }
+
+    while (!tryUpdateTotalCount) {}
+
+    DefaultMetricSnapshot(nrOfRecordings, entries.result())
+  }
+
+}
+
+object HighDynamicRangeRecorder {
+  val totalCountUpdater = AtomicLongFieldUpdater.newUpdater(classOf[AtomicHistogram], "totalCount")
+
+  def apply(configuration: Configuration): HighDynamicRangeRecorder = new HighDynamicRangeRecorder(configuration)
+
+  case class Configuration(highestTrackableValue: Long, significantValueDigits: Int)
+
+  case object Configuration {
+    def fromConfig(config: Config): Configuration = {
+      Configuration(config.getLong("highest-trackable-value"), config.getInt("significant-value-digits"))
+    }
+  }
+}