components/model-catalog/blueprint-model/archetype-blueprint/src/main/resources/archetype-resources/Tests/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/uat/logging/SmartColorDiscriminator.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

package org.onap.ccsdk.cds.blueprintsprocessor.uat.logging

import ch.qos.logback.classic.spi.ILoggingEvent
import ch.qos.logback.core.sift.AbstractDiscriminator
import org.onap.ccsdk.cds.blueprintsprocessor.uat.logging.ColorMarker
import org.onap.ccsdk.cds.blueprintsprocessor.uat.logging.LogColor.MDC_COLOR_KEY

class SmartColorDiscriminator : AbstractDiscriminator<ILoggingEvent>() {

    var defaultValue: String = "white"

    override fun getKey(): String {
        return MDC_COLOR_KEY
    }

    fun setKey() {
        throw UnsupportedOperationException("Key not settable. Using $MDC_COLOR_KEY")
    }

    override fun getDiscriminatingValue(e: ILoggingEvent): String =
        (e.marker as? ColorMarker)?.name
            ?: e.mdcPropertyMap?.get(MDC_COLOR_KEY)
            ?: defaultValue
}