kyo.stats.Stats.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kyo-core-opt_3 Show documentation
kyo-core-opt
There is a newer version: 0.8.5
package kyo.stats

import kyo._
import kyo.ios._
import kyo.locals.Locals
import kyo.stats.internal
import kyo.stats.internal.TraceReceiver

abstract class Stats {

  def scope(name: String): Stats

  def initCounter(
      name: String,
      description: String = "empty",
      unit: String = "",
      attributes: Attributes = Attributes.empty
  ): Counter

  def initHistogram(
      name: String,
      description: String = "empty",
      unit: String = "",
      attributes: Attributes = Attributes.empty
  ): Histogram

  def initGauge(
      name: String,
      description: String = "empty",
      unit: String = "",
      attributes: Attributes = Attributes.empty
  )(f: => Double): Gauge

  def traceSpan[T, S](
      name: String,
      attributes: Attributes = Attributes.empty
  )(v: => T > S): T > (IOs with S)
}

object Stats {
  val noop: Stats =
    new Stats {

      def scope(name: String) = this

      def initCounter(
          name: String,
          description: String,
          unit: String,
          a: Attributes
      ) =
        Counter.noop

      def initHistogram(
          name: String,
          description: String,
          unit: String,
          a: Attributes
      ) =
        Histogram.noop

      def initGauge(
          name: String,
          description: String = "",
          unit: String = "",
          a: Attributes = Attributes.empty
      )(f: => Double) =
        Gauge.noop

      def traceSpan[T, S](
          name: String,
          attributes: Attributes = Attributes.empty
      )(v: => T > S): T > (IOs with S) = v
    }

  private val traceReceiver = Locals.init[TraceReceiver](TraceReceiver.get)

  def traceListen[T, S](receiver: TraceReceiver)(v: T > S): T > (IOs with S) =
    traceReceiver.get.map { curr =>
      traceReceiver.let(TraceReceiver.all(List(curr, receiver)))(v)
    }

  def initScope(first: String, rest: String*): Stats =
    scope(first :: rest.toList)

  private def scope(path: List[String]): Stats =
    new Stats {
      def scope(name: String) =
        Stats.scope(name :: path)

      def initCounter(
          name: String,
          description: String,
          unit: String,
          a: Attributes
      ) =
        internal.MetricReceiver.get.counter(path.reverse, name, description, unit, a)

      def initHistogram(
          name: String,
          description: String,
          unit: String,
          a: Attributes
      ) =
        internal.MetricReceiver.get.histogram(path.reverse, name, description, unit, a)

      def initGauge(
          name: String,
          description: String = "",
          unit: String = "",
          a: Attributes = Attributes.empty
      )(f: => Double) =
        internal.MetricReceiver.get.gauge(path.reverse, name, description, unit, a)(f)

      def traceSpan[T, S](
          name: String,
          attributes: Attributes = Attributes.empty
      )(v: => T > S): T > (IOs with S) =
        traceReceiver.get.map(internal.Span.trace(_, path.reverse, name, attributes)(v))

      override def toString = s"Stats(scope = ${path.reverse})"
    }
}