com.twitter.finagle.stats.OstrichStatsReceiver.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finagle-ostrich4_2.11 Show documentation

finagle-ostrich4

The newest version!

package com.twitter.finagle.stats

import com.twitter.ostrich.stats.{Stats, StatsCollection}

class OstrichStatsReceiver(
  val repr: StatsCollection = Stats,
  val delimiter: String = "/"
) extends StatsReceiverWithCumulativeGauges {

  def this() = this(Stats, "/")
  // To avoid breaking the Java API:
  def this(repr: StatsCollection) = this(repr, "/")

  override def toString: String = "OstrichStatsReceiver"

  override protected[this] def registerGauge(name: Seq[String], f: => Float) {
    repr.addGauge(variableName(name)) { f.toDouble }
  }

  override protected[this] def deregisterGauge(name: Seq[String]) {
    repr.clearGauge(variableName(name))
  }

  override def counter(name: String*) = new Counter {
    private[this] val counter = repr.getCounter(variableName(name))

    override def incr(delta: Int) { counter.incr(delta) }
  }

  override def stat(name: String*) = new Stat {
    private[this] val metric = repr.getMetric(variableName(name))

    override def add(value: Float) {
      metric.add(value.toInt)
    }
  }

  private[this] def variableName(name: Seq[String]) = name mkString delimiter
}