org.jetbrains.kotlinx.jupyter.ext.integration.ElapsedTimeIntegration.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-jupyter-lib-ext Show documentation

Extended functionality for Kotlin kernel

The newest version!

package org.jetbrains.kotlinx.jupyter.ext.integration

import org.jetbrains.kotlinx.jupyter.api.annotations.JupyterLibrary
import org.jetbrains.kotlinx.jupyter.api.libraries.JupyterIntegration
import kotlin.time.Duration
import kotlin.time.Duration.Companion.milliseconds

@JupyterLibrary
object ElapsedTimeIntegration : JupyterIntegration() {
    private var lastTime: Long = -1L

    var enabled = false
    var format: (Duration) -> Any = { duration -> "Elapsed time: $duration" }

    override fun Builder.onLoaded() {
        beforeCellExecution {
            if (enabled) {
                lastTime = currentTime()
            }
        }

        afterCellExecution { _, _ ->
            if (enabled && lastTime != -1L) {
                val elapsed = currentTime() - lastTime
                val duration = elapsed.milliseconds
                display(format(duration), null)
            }
        }
    }

    private fun currentTime(): Long = System.currentTimeMillis()
}