ms/controllerblueprints/modules/blueprint-core/src/main/kotlin/org/onap/ccsdk/cds/controllerblueprints/core/MDCContext.kt

   1 /*
   2  * Copyright © 2018-2019 AT&T Intellectual Property.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.onap.ccsdk.cds.controllerblueprints.core
  18
  19 import kotlinx.coroutines.ThreadContextElement
  20 import org.slf4j.MDC
  21 import kotlin.coroutines.AbstractCoroutineContextElement
  22 import kotlin.coroutines.CoroutineContext
  23
  24 typealias MDCContextMap = Map<String, String>?
  25
  26 class MDCContext(private val contextMap: MDCContextMap = MDC.getCopyOfContextMap()) :
  27         ThreadContextElement<MDCContextMap>, AbstractCoroutineContextElement(Key) {
  28     /**
  29      * Key of [MDCContext] in [CoroutineContext].
  30      */
  31     companion object Key : CoroutineContext.Key<MDCContext>
  32
  33     override fun updateThreadContext(context: CoroutineContext): MDCContextMap {
  34         val oldState = MDC.getCopyOfContextMap()
  35         setCurrent(contextMap)
  36         return oldState
  37     }
  38
  39     override fun restoreThreadContext(context: CoroutineContext, oldState: MDCContextMap) {
  40         setCurrent(oldState)
  41     }
  42
  43     private fun setCurrent(contextMap: MDCContextMap) {
  44         if (contextMap == null) {
  45             MDC.clear()
  46         } else {
  47             MDC.setContextMap(contextMap)
  48         }
  49     }
  50 }