! all: improve the metric recorders infrastructure

1 files changed, 59 insertions, 0 deletions

diff --git a/kamon-system-metrics/src/main/scala/kamon/system/sigar/SigarMetricsUpdater.scala b/kamon-system-metrics/src/main/scala/kamon/system/sigar/SigarMetricsUpdater.scala
new file mode 100644
index 00000000..8a430427
--- /dev/null
+++ b/kamon-system-metrics/src/main/scala/kamon/system/sigar/SigarMetricsUpdater.scala
@@ -0,0 +1,59 @@
+package kamon.system.sigar
+
+import akka.actor.{ Props, Actor }
+import kamon.Kamon
+import kamon.metric.instrument.InstrumentFactory
+import kamon.metric.{ Entity, EntityRecorder, MetricsExtension, Metrics }
+import kamon.system.sigar.SigarMetricsUpdater.UpdateSigarMetrics
+import org.hyperic.sigar.Sigar
+
+import scala.concurrent.duration.FiniteDuration
+
+class SigarMetricsUpdater(refreshInterval: FiniteDuration) extends Actor {
+  val sigar = new Sigar
+  val metricsExtension = Kamon(Metrics)(context.system)
+
+  val sigarMetrics = List(
+    CpuMetrics.register(metricsExtension),
+    FileSystemMetrics.register(metricsExtension),
+    LoadAverageMetrics.register(metricsExtension),
+    MemoryMetrics.register(metricsExtension),
+    NetworkMetrics.register(metricsExtension),
+    ProcessCpuMetrics.register(metricsExtension))
+
+  val refreshSchedule = context.system.scheduler.schedule(refreshInterval, refreshInterval, self, UpdateSigarMetrics)(context.dispatcher)
+
+  def receive = {
+    case UpdateSigarMetrics ⇒ updateMetrics()
+  }
+
+  def updateMetrics(): Unit = {
+    sigarMetrics.foreach(_.update(sigar))
+  }
+
+  override def postStop(): Unit = {
+    refreshSchedule.cancel()
+    super.postStop()
+  }
+}
+
+object SigarMetricsUpdater {
+  def props(refreshInterval: FiniteDuration): Props =
+    Props(new SigarMetricsUpdater((refreshInterval)))
+
+  case object UpdateSigarMetrics
+}
+
+trait SigarMetric extends EntityRecorder {
+  def update(sigar: Sigar): Unit
+}
+
+abstract class SigarMetricRecorderCompanion(metricName: String) {
+  def register(metricsExtension: MetricsExtension): SigarMetric = {
+    val instrumentFactory = metricsExtension.instrumentFactory("system-metric")
+    metricsExtension.register(Entity(metricName, "system-metric"), apply(instrumentFactory)).recorder
+  }
+
+  def apply(instrumentFactory: InstrumentFactory): SigarMetric
+}
+