kyo.stats.internal.UnsafeGauge.scala Maven / Gradle / Ivy

Go to download
package kyo.stats.internal

class UnsafeGauge(run: () => Double) {
    def collect(): Double = run()
}

class UnsafeCounterGauge(run: () => Long) {
    private var last = 0L
    def collect(): Long = {
        val value = run()
        if (value < 0) {
            (Long.MaxValue + value) + 2
        } else {
            value
        }
    }

    private def findDelta(a: Long, b: Long) = {
        (Long.MaxValue - a) + b
    }

    private[kyo] def delta() = {
        val curr  = collect()
        val delta = if (curr >= last) curr - last else findDelta(last, curr)
        last = curr
        delta
    }

    private[kyo] def getLast(): Long = last
}