MDC helper function for Kotlin

WithMdc.kt

import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.slf4j.MDCContext
import kotlinx.coroutines.withContext
import org.slf4j.MDC

inline fun <T> withMdc(vararg attributes: Pair<String, Any?>, block: () -> T): T {
    val oldAttributes = MDC.getCopyOfContextMap() ?: emptyMap()
    attributes.forEach { (key, value) ->
        value?.apply {
            MDC.put(key, toString())
        }
    }

    return try {
        block()
    } finally {
        MDC.setContextMap(oldAttributes)
    }
}

suspend fun <T> withMdcContext(vararg attributes: Pair<String, Any?>, block: suspend CoroutineScope.() -> T): T {
    val mdcContext = withMdc(*attributes) { MDCContext() }
    return withContext(mdcContext, block)
}

Usage example:

    withMdc(
        "userId" to user.id,
        "timestamp" to LocalDateTime.now(),
    ) {
        logger.info("User called a method")
        someOtherMethodWhereYouWantToPropagateTheContext()
    }