io.github.numichi.reactive.logger.hook.MDCContextHookCache.kt Maven / Gradle / Ivy

Go to download

package io.github.numichi.reactive.logger.hook

enum class Position {
    BEFORE,
    AFTER,
}

internal object MDCContextHookCache {
    private val store = mutableMapOf()

    fun clear() {
        store.clear()
    }

    fun getHooks(): Map {
        return store
    }

    fun existsHook(position: Position): Boolean {
        return store.containsKey(position)
    }

    fun addHook(hook: MDCContextHook) {
        if (hook.position == Position.BEFORE) {
            store[Position.BEFORE] = hook
        } else {
            store[Position.AFTER] = hook
        }
    }

    fun removeHook(position: Position) {
        store.remove(position)
    }
}