klogger.kt

package me.lucapette.ext.klogger

import mu.KLogger
import java.util.WeakHashMap

data class TimeDelta(val sinceStart: Long, val sinceLast: Long)

data class TimeLog(val start: Long, var current: Long) {
    fun since(): TimeDelta {
        val now = System.currentTimeMillis()

        return TimeDelta(now - start, now - current)
    }

    fun update()  {
        current = System.currentTimeMillis()
    }
}

val store = WeakHashMap<String, TimeLog>()

fun KLogger.time(label: String) {
    val now = System.currentTimeMillis()
    store[label] = TimeLog(now, now)
}

fun KLogger.timeLog(label: String, message: String) {
    val timeLog = store[label] ?: return
    val delta = timeLog.since()
    info { "[$label ${delta.sinceStart} ${delta.sinceLast}] $message" }
    timeLog.update()
}

fun KLogger.timeEnd(label: String) {
    val timeLog = store[label] ?: return
    val delta = timeLog.since()
    info { "[$label ${delta.sinceStart} ${delta.sinceLast}]" }
    store.remove(label)
}