Kotlin中如何检测函数递归调用？协程环境下避免runByChannel嵌套死锁

在Kotlin协程环境中，替代ThreadLocal的最佳方案是自定义CoroutineContext元素——协程上下文会全程跟随协程执行，不受线程调度切换的影响，完美适配你的场景。

步骤1：定义标记用的CoroutineContext元素 #

首先创建一个专属的上下文标记，用来标识当前协程是否处于runByChannel的执行链中：

// 定义标记的Key，用于在CoroutineContext中查找
private object RunByChannelMarkerKey : CoroutineContext.Key<RunByChannelMarker>

// 标记元素本身，仅作为存在性判断的标识
private data class RunByChannelMarker(val unused: Unit = Unit) : CoroutineContext.Element {
    override val key: CoroutineContext.Key<*>
        get() = RunByChannelMarkerKey
}

步骤2：修改runByChannel函数实现 #

在runByChannel中加入标记检查和上下文注入逻辑：

fun <T> runByChannel(block: suspend () -> T): CompletableDeferred<T> {
    // 检查当前协程上下文是否已有标记，若有则说明是递归调用，直接抛出异常或处理错误
    currentCoroutineContext()[RunByChannelMarkerKey]?.let {
        throw IllegalStateException("递归调用runByChannel会导致死锁，禁止此类操作")
        // 若不想抛出异常，也可以返回失败的Deferred：
        // return CompletableDeferred<T>().apply { completeExceptional(IllegalStateException(...)) }
    }

    val resultDeferred = CompletableDeferred<T>()
    
    // 向Channel发送任务时，将标记注入协程上下文
    yourChannel.send {
        withContext(currentCoroutineContext() + RunByChannelMarker()) {
            try {
                val result = block()
                resultDeferred.complete(result)
            } catch (e: Throwable) {
                resultDeferred.completeExceptional(e)
            }
        }
    }

    return resultDeferred
}

方案原理说明 #

• CoroutineContext是协程的核心属性，会自动跟随协程的调度、子协程继承传递，无论协程切换到哪个线程，上下文里的标记都会保留。
• 当runByChannel执行Lambda时，我们通过withContext将标记加入当前协程上下文，后续Lambda内部任何位置调用runByChannel，都会检测到这个标记，从而阻止递归调用。

额外注意事项 #

• 如果Lambda内部需要启动独立的子协程（比如launch），且允许该子协程调用runByChannel，可以在启动子协程时移除标记：

  launch(currentCoroutineContext().minusKey(RunByChannelMarkerKey)) {
      // 这里可以安全调用runByChannel
      runByChannel { ... }
  }
  

• 错误处理逻辑可根据业务需求调整，比如返回错误结果而非抛出异常，避免影响整体流程。

内容的提问来源于stack exchange，提问作者progquester