2 * Copyright © 2018-2019 AT&T Intellectual Property.
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 package org.onap.ccsdk.cds.controllerblueprints.core
19 import kotlinx.coroutines.ThreadContextElement
21 import kotlin.coroutines.AbstractCoroutineContextElement
22 import kotlin.coroutines.CoroutineContext
24 typealias MDCContextMap = Map<String, String>?
26 class MDCContext(private val contextMap: MDCContextMap = MDC.getCopyOfContextMap()) :
27 ThreadContextElement<MDCContextMap>, AbstractCoroutineContextElement(Key) {
29 * Key of [MDCContext] in [CoroutineContext].
31 companion object Key : CoroutineContext.Key<MDCContext>
33 override fun updateThreadContext(context: CoroutineContext): MDCContextMap {
34 val oldState = MDC.getCopyOfContextMap()
35 setCurrent(contextMap)
39 override fun restoreThreadContext(context: CoroutineContext, oldState: MDCContextMap) {
43 private fun setCurrent(contextMap: MDCContextMap) {
44 if (contextMap == null) {
47 MDC.setContextMap(contextMap)