org.jetbrains.kotlinx.jupyter.config.logging.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-jupyter-shared-compiler Show documentation

Implementation of REPL compiler and preprocessor for Jupyter dialect of Kotlin (IDE-compatible)

There is a newer version: 0.12.0-290

Show newest version

package org.jetbrains.kotlinx.jupyter.config

import org.jetbrains.kotlinx.jupyter.exceptions.renderException
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import java.io.PrintStream
import java.util.concurrent.locks.ReentrantLock

fun getLogger(name: String = "ikotlin"): Logger = LoggerFactory.getLogger(name)

object KernelStreams {
    private val rLock = ReentrantLock()

    private var _out: PrintStream = System.out
    private var _err: PrintStream = System.err

    val out: PrintStream get() = _out
    val err: PrintStream get() = _err

    fun setStreams(lock: Boolean, output: PrintStream, error: PrintStream) {
        if (lock) rLock.lock()
        _out = output
        _err = error
        if (!lock) rLock.unlock()
    }
}

fun Logger.errorForUser(stream: PrintStream = KernelStreams.err, message: String, throwable: Throwable? = null) {
    if (throwable == null) error(message)
    else error(message, throwable)

    if (message.isNotEmpty()) {
        stream.print("[ERROR] ")
        stream.println(message)
    }
    throwable?.let { stream.println(it.renderException()) }
    stream.flush()
}

fun  Logger.catchAll(stream: PrintStream = KernelStreams.err, msg: String = "", body: () -> T): T? = try {
    body()
} catch (e: Throwable) {
    this.errorForUser(stream, msg, e)
    null
}