테스트 환경(Kotest) + MDCContext 사용 시 kotlin.coroutines.coroutineContext로 컨텍스트 요소 조회 시 null 반환 (실제 환경에서는 정상)

Question

안녕하세요, 코루틴 가지고 씨름하다가 좀 이상한 현상을 발견해서 질문 남깁니다.다른 게 아니라, 테스트 환경이랑 실제 서버 환경에서 코루틴 컨텍스트 요소 조회 결과가 다르게 나오네요. 특히 MDCContext랑 제가 만든 커스텀 컨텍스트 요소를 같이 쓸 때 문제가 생기는 것 같아요.하고자 했던 것:API 요청 처리하다가 백그라운드에서 비동기로 뭔가 처리해야 할 때가 있잖아요? 이때 요청 추적 정보(TraceContextElement라는 걸 만들었어요)랑 MDC 로깅 정보를 다음 코루틴으로 잘 넘겨주고 싶었습니다. kotlinx-coroutines-slf4j의 MDCContext도 같이 써야 하고요. 특이사항은 fire-and-frogot 패턴에서도 이런 Tracing 컨텍스트가 유지되어 로깅을 하고 싶었습니다.사용한 코드(간략하게):1. 커스텀 컨텍스트 요소:/ TraceContext를 감싸는 요소 class TraceContextElement(val context: TraceContext) : AbstractCoroutineContextElement(Key) { companion object Key : CoroutineContext.Key<TraceContextElement> } // 컨텍스트 조회를 위한 확장 함수 fun CoroutineContext.traceContext(): TraceContext? = this[TraceContextElement]?.context // 간략화된 TraceContext 데이터 클래스 data class TraceContext(val traceId: String?, /* 다른 필드들 */ )2. *코루틴 실행 함수:* 두 가지 버전으로 만들어 봤어요. // --- 실행기 1: CoroutineScope 확장 람다 사용 --- fun CoroutineScope.launchWithTrace( traceManager: TraceManager, // 테스트에서는 Mocking block: suspend CoroutineScope.() -> Unit // 람다가 CoroutineScope의 확장 함수 형태 ): Job { val mdcContextMap = MDC.getCopyOfContextMap() ?: emptyMap() val traceContext = traceManager.createTraceContext() // 테스트용 TraceContext 반환 // 두 컨텍스트와 함께 launch return launch(MDCContext() + TraceContextElement(traceContext)) { try { if (mdcContextMap.isNotEmpty()) MDC.setContextMap(mdcContextMap) block.invoke(this) // 여기서 'this'는 CoroutineScope } finally { MDC.clear() } } } // --- 실행기 2: 일반 suspend 람다 사용 --- fun CoroutineScope.launchWithTrace2( traceManager: TraceManager, // 테스트에서는 Mocking block: suspend () -> Unit // 람다가 일반적인 suspend 함수 형태 ): Job { val mdcContextMap = MDC.getCopyOfContextMap() ?: emptyMap() val traceContext = traceManager.createTraceContext() // 테스트용 TraceContext 반환 // 두 컨텍스트와 함께 launch return launch(MDCContext() + TraceContextElement(traceContext)) { try { if (mdcContextMap.isNotEmpty()) MDC.setContextMap(mdcContextMap) block() // 일반 함수 호출 } finally { MDC.clear() } } } (두 함수 다 내부적으로는 MDCContext()랑 TraceContextElement를 컨텍스트에 넣고 코루틴을 실행해요. TraceManager는 테스트에서 Mocking 하고요.)문제상황!:Kotest로 짠 테스트 코드에서 Dispatchers.Default를 써서 돌려보면요, launchWithTrace (버전 1) 사용 시: test("launchWithTrace는 잘 돼요") { // ... (scope, mock 설정) ... scope.launchWithTrace(mockTraceManager) { // 확장 람다 방식 val ctx = coroutineContext.traceContext() // 여기서 this.coroutineContext println("launchWithTrace 컨텍스트: $ctx") // 👉 정상 출력: TraceContext(traceId=test-id) ctx shouldNotBe null // OK // ... } // ... } 이건 문제없이 잘 돌아가고, coroutineContext (스코프의 컨텍스트)로 TraceContextElement를 잘 가져옵니다. launchWithTrace2 (버전 2) 사용 시: test("launchWithTrace2는 이상해요 - 실패") { // ... (scope, mock 설정) ... scope.launchWithTrace2(mockTraceManager) { // 일반 람다 방식 val ctx = kotlin.coroutines.coroutineContext.traceContext() // 최상위 프로퍼티 사용 println("launchWithTrace2 컨텍스트: $ctx") // 👉 출력: null // ctx shouldNotBe null // 여기서 실패해요! // ... } // ... } 이게 문제입니다. 똑같이 컨텍스트에 넣었는데, 람다 안에서 kotlin.coroutines.coroutineContext로 접근하면 TraceContextElement를 못 찾고 null이 나와요.그런데 더 이상한 건:실제 Spring Boot 서버를 띄워서 API를 쏴보면요, launchWithTrace2 (버전 2)를 써도 kotlin.coroutines.coroutineContext.traceContext() 호출 시 정상적으로 TraceContextElement를 가져옵니다. (보통 Dispatchers.IO랑 같이 쓰긴 합니다)궁금한 점:대체 왜 테스트 환경(Kotest, MockK, Dispatchers.Default, MDCContext 조합)에서만, 일반 suspend () -> Unit 람다 안에서 kotlin.coroutines.coroutineContext로 접근하면 컨텍스트 요소를 못 찾는 걸까요?혹시 kotlin.coroutines.coroutineContext 이게 MDCContext나 Dispatchers.Default 때문에 생기는 스레드 전환 같은 거랑 뭔가 꼬이는 부분이 테스트 환경에만 있는 걸까요? 아니면 Continuation에서 컨텍스트를 가져오는 방식이랑 CoroutineScope 객체에서 직접 가져오는 방식 사이에 우리가 모르는 차이가 있는 건지... 경험 있으신 분들의 조언을 구합니다!혹시 비슷한 문제 겪어보셨거나 원인을 짐작 가는 분 계시면 도움 부탁드립니다!사용 환경:Kotlin: 1.9.xkotlinx-coroutines-core: 1.8.xkotlinx-coroutines-slf4j: 1.8.xKotest: 5.x.xMockK: 1.13.xJVM: 21

캐티 · Answer

스레드 매니저에서 새로운 스레드를 다루는 방식이 다를 수가 있겟쩌. 이런 경우에는 스레드아이디랑 잡아이디를 확인해보아야 하는 거여요. 일반적으로 스택 변수에 접근한다거나 .. c언어도 이럼 앵간한 경우에 접근이 어렵고 물론 포인터를 가지고 있으면 이케이케 하는 수는 있는데 코틀린은 많이 다르겠쩌.

스토리 · Answer

저도 직접 디버깅을 해봐야 알겠지만,Spring boot는 Mono를 호출할 때, Dispatchers.Unconfined를 사용 합니다.https://github.com/spring-projects/spring-framework/blob/main/spring-webflux/src/main/kotlin/org/springframework/web/reactive/function/server/CoRouterFunctionDsl.kt#L742이로 인해서, launchWithTrace2를 사용하실 때 scope와 최상위 컨텍스트의 스레드가 같았을 것 입니다.근데 테스트 환경이니, Dispatchers.Default로 인해 두 스레드가 달라서 null이 발생하는 것으로 보입니다.

테스트 환경(Kotest) + MDCContext 사용 시 kotlin.coroutines.coroutineContext로 컨텍스트 요소 조회 시 null 반환 (실제 환경에서는 정상)

2개의 답변

#kotlin 목록