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     /**
  30      * Key of [MDCContext] in [CoroutineContext].
  31      */
  32     companion object Key : CoroutineContext.Key<MDCContext>
  33
  34     override fun updateThreadContext(context: CoroutineContext): MDCContextMap {
  35         val oldState = MDC.getCopyOfContextMap()
  36         setCurrent(contextMap)
  37         return oldState
  38     }
  39
  40     override fun restoreThreadContext(context: CoroutineContext, oldState: MDCContextMap) {
  41         setCurrent(oldState)
  42     }
  43
  44     private fun setCurrent(contextMap: MDCContextMap) {
  45         if (contextMap == null) {
  46             MDC.clear()
  47         } else {
  48             MDC.setContextMap(contextMap)
  49         }
  50     }
  51 }