com.gu.management.JvmMetrics.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of management_2.11 Show documentation
management
The newest version!
package com.gu.management

import management.{ MemoryPoolMXBean, ManagementFactory }
import collection.JavaConversions._
import org.slf4j.{ Logger, LoggerFactory }

object JvmMetrics {
  private val logger = LoggerFactory.getLogger("com.gu.management.JvmMetrics")

  lazy val all = numThreads.toList ::: totalThreads.toList ::: gcRates ::: memoryUsage

  lazy val numThreads =
    try {
      ManagementFactory.getThreadMXBean.getThreadCount
      Some(
        new GaugeMetric(
          group = "jvm",
          name = "num_threads",
          title = "Number of active threads",
          description = "Number of threads currently active as reported by the jvm",
          getValue = () => ManagementFactory.getThreadMXBean.getThreadCount
        )
      )
    } catch {
      case e: Exception =>
        logger.debug("Failed to initialise active threads metric")
        None
    }

  lazy val totalThreads =
    try {
      ManagementFactory.getThreadMXBean.getTotalStartedThreadCount
      Some(
        new GaugeMetric(
          group = "jvm",
          name = "total_threads",
          title = "Thread started threads",
          description = "Threads started since the application started as reported by the jvm",
          getValue = () => ManagementFactory.getThreadMXBean.getTotalStartedThreadCount
        )
      )
    } catch {
      case e: Exception =>
        logger.debug("Failed to initialise total threads metric")
        None
    }

  lazy val gcRates =
    try {
      ManagementFactory.getGarbageCollectorMXBeans.toList map { gc =>
        new TimingMetric(
          group = "jvm",
          name = "gc_" + gc.getName.toLowerCase.replace(' ', '_'),
          title = "GC " + gc.getName,
          description = "Collection rates for the " + gc.getName + " garbage collector"
        ) {
          override def count = gc.getCollectionCount
          override def totalTimeInMillis = gc.getCollectionTime
        }
      }
    } catch {
      case e: Exception =>
        logger.trace("Failed to initialise gc metrics", e)
        Nil
    }

  lazy val memoryUsage =
    try {
      ManagementFactory.getMemoryPoolMXBeans().toList flatMap { memPool: MemoryPoolMXBean =>
        {
          Option(memPool.getUsage) match {
            case Some(_) => Some(new GaugeMetric(
              group = "jvm",
              name = "%s memory usage".format(memPool.getName),
              title = "%s Memory Usage".format(memPool.getName),
              description = "%s memory usage as percentage of max value".format(memPool.getName),
              getValue = () => {
                val currentUsage = memPool.getUsage
                math.round(currentUsage.getUsed.toFloat / currentUsage.getMax * 100)
              }
            ))
            case None => None
          }
        }
      }
    } catch {
      case e: Exception =>
        logger.trace("Failed to initialise memory usage metrics", e)
        Nil
    }

}