From b32f871adec935090f54c56418e97d175243d0de Mon Sep 17 00:00:00 2001
From: Ivan Topolnjak <ivantopo@gmail.com>
Date: Fri, 5 Dec 2014 02:22:25 +0100
Subject: + newrelic: react correctly to restart and shutdown events from the
 New Relic collector.

---
 kamon-core/src/main/scala/kamon/TimeUnits.scala      | 20 ++++++++++++++++++++
 .../src/main/scala/kamon/metric/Subscriptions.scala  | 10 +++++-----
 2 files changed, 25 insertions(+), 5 deletions(-)

(limited to 'kamon-core/src/main/scala/kamon')

diff --git a/kamon-core/src/main/scala/kamon/TimeUnits.scala b/kamon-core/src/main/scala/kamon/TimeUnits.scala
index 44f5b4c3..f2933a11 100644
--- a/kamon-core/src/main/scala/kamon/TimeUnits.scala
+++ b/kamon-core/src/main/scala/kamon/TimeUnits.scala
@@ -1,10 +1,30 @@
 package kamon
 
+/**
+ *  Epoch time stamp in seconds.
+ */
+class Timestamp(val seconds: Long) extends AnyVal {
+  def <(that: Timestamp): Boolean = this.seconds < that.seconds
+  def >(that: Timestamp): Boolean = this.seconds > that.seconds
+  def ==(that: Timestamp): Boolean = this.seconds == that.seconds
+  def >=(that: Timestamp): Boolean = this.seconds >= that.seconds
+  def <=(that: Timestamp): Boolean = this.seconds <= that.seconds
+
+  override def toString: String = String.valueOf(seconds) + ".seconds"
+}
+
+object Timestamp {
+  def now: Timestamp = new Timestamp(System.currentTimeMillis() / 1000)
+  def earlier(l: Timestamp, r: Timestamp): Timestamp = if (l <= r) l else r
+  def later(l: Timestamp, r: Timestamp): Timestamp = if (l >= r) l else r
+}
+
 /**
  *  Epoch time stamp in milliseconds.
  */
 class MilliTimestamp(val millis: Long) extends AnyVal {
   override def toString: String = String.valueOf(millis) + ".millis"
+  def toTimestamp: Timestamp = new Timestamp(millis / 1000)
 }
 
 object MilliTimestamp {
diff --git a/kamon-core/src/main/scala/kamon/metric/Subscriptions.scala b/kamon-core/src/main/scala/kamon/metric/Subscriptions.scala
index c6571507..a22e1c21 100644
--- a/kamon-core/src/main/scala/kamon/metric/Subscriptions.scala
+++ b/kamon-core/src/main/scala/kamon/metric/Subscriptions.scala
@@ -21,7 +21,7 @@ import kamon.metric.Subscriptions._
 import kamon.util.GlobPathFilter
 import scala.concurrent.duration.{ FiniteDuration, Duration }
 import java.util.concurrent.TimeUnit
-import kamon.Kamon
+import kamon.{ MilliTimestamp, Kamon }
 import kamon.metric.TickMetricSnapshotBuffer.FlushBuffer
 
 class Subscriptions extends Actor {
@@ -30,7 +30,7 @@ class Subscriptions extends Actor {
   val flushMetricsSchedule = scheduleFlushMessage()
   val collectionContext = Kamon(Metrics).buildDefaultCollectionContext
 
-  var lastTick: Long = System.currentTimeMillis()
+  var lastTick: MilliTimestamp = MilliTimestamp.now
   var oneShotSubscriptions: Map[ActorRef, MetricSelectionFilter] = Map.empty
   var permanentSubscriptions: Map[ActorRef, MetricSelectionFilter] = Map.empty
 
@@ -65,7 +65,7 @@ class Subscriptions extends Actor {
   }
 
   def flush(): Unit = {
-    val currentTick = System.currentTimeMillis()
+    val currentTick = MilliTimestamp.now
     val snapshots = collectAll()
 
     dispatchSelectedMetrics(lastTick, currentTick, permanentSubscriptions, snapshots)
@@ -86,7 +86,7 @@ class Subscriptions extends Actor {
     builder.result()
   }
 
-  def dispatchSelectedMetrics(lastTick: Long, currentTick: Long, subscriptions: Map[ActorRef, MetricSelectionFilter],
+  def dispatchSelectedMetrics(lastTick: MilliTimestamp, currentTick: MilliTimestamp, subscriptions: Map[ActorRef, MetricSelectionFilter],
     snapshots: Map[MetricGroupIdentity, MetricGroupSnapshot]): Unit = {
 
     for ((subscriber, filter) ← subscriptions) {
@@ -108,7 +108,7 @@ object Subscriptions {
   case object FlushMetrics
   case class Unsubscribe(subscriber: ActorRef)
   case class Subscribe(category: MetricGroupCategory, selection: String, subscriber: ActorRef, permanently: Boolean = false)
-  case class TickMetricSnapshot(from: Long, to: Long, metrics: Map[MetricGroupIdentity, MetricGroupSnapshot])
+  case class TickMetricSnapshot(from: MilliTimestamp, to: MilliTimestamp, metrics: Map[MetricGroupIdentity, MetricGroupSnapshot])
 
   trait MetricSelectionFilter {
     def accept(identity: MetricGroupIdentity): Boolean
-- 
cgit v1.2.3