kyo.stats.Counter.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._

case class Counter(unsafe: Counter.Unsafe) extends AnyVal {
  def inc: Unit > IOs =
    IOs(unsafe.inc())
  def add(v: Long): Unit > IOs =
    IOs(unsafe.add(v))
  def add(v: Long, b: Attributes): Unit > IOs =
    IOs(unsafe.add(v, b))
  def attributes(b: Attributes): Counter =
    Counter(unsafe.attributes(b))
}

object Counter {

  abstract class Unsafe {
    def inc(): Unit
    def add(v: Long): Unit
    def add(v: Long, b: Attributes): Unit
    def attributes(b: Attributes): Unsafe
  }

  val noop: Counter =
    Counter(
        new Unsafe {
          def inc()                       = ()
          def add(v: Long)                = ()
          def add(v: Long, b: Attributes) = ()
          def attributes(b: Attributes)   = this
        }
    )

  def all(l: List[Counter]): Counter =
    l.filter(_ != noop) match {
      case Nil =>
        noop
      case h :: Nil =>
        h
      case l =>
        Counter(
            new Unsafe {
              def inc() = add(1)
              def add(v: Long) = {
                var c = l
                while (c ne Nil) {
                  c.head.unsafe.add(v)
                  c = c.tail
                }
              }
              def add(v: Long, b: Attributes) = {
                var c = l
                while (c ne Nil) {
                  c.head.unsafe.add(v, b)
                  c = c.tail
                }
              }
              def attributes(b: Attributes) =
                all(l.map(c => c.attributes(b))).unsafe
            }
        )
    }
}