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
|
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
}
|