! all: improve the metric recorders infrastructure

1 files changed, 28 insertions, 0 deletions

diff --git a/kamon-system-metrics/src/main/scala/kamon/system/jmx/ThreadsMetrics.scala b/kamon-system-metrics/src/main/scala/kamon/system/jmx/ThreadsMetrics.scala
new file mode 100644
index 00000000..b33eb3e6
--- /dev/null
+++ b/kamon-system-metrics/src/main/scala/kamon/system/jmx/ThreadsMetrics.scala
@@ -0,0 +1,28 @@
+package kamon.system.jmx
+
+import java.lang.management.ManagementFactory
+
+import kamon.metric.GenericEntityRecorder
+import kamon.metric.instrument.InstrumentFactory
+
+class ThreadsMetrics(instrumentFactory: InstrumentFactory) extends GenericEntityRecorder(instrumentFactory) {
+  val threadsBean = ManagementFactory.getThreadMXBean
+
+  gauge("daemon-thread-count", () ⇒ {
+    threadsBean.getDaemonThreadCount.toLong
+  })
+
+  gauge("peak-thread-count", () ⇒ {
+    threadsBean.getPeakThreadCount.toLong
+  })
+
+  gauge("thread-count", () ⇒ {
+    threadsBean.getThreadCount.toLong
+  })
+
+}
+
+object ThreadsMetrics extends JmxSystemMetricRecorderCompanion("threads") {
+  def apply(instrumentFactory: InstrumentFactory): ThreadsMetrics =
+    new ThreadsMetrics(instrumentFactory)
+}