Spring AOP中能否为suspend函数连接点指定协程上下文？

你遇到的核心问题是：通过返回Mono的AOP通知无法正确将自定义协程上下文传递给目标suspend函数，因为目标函数的协程上下文在被Spring AOP代理时就已经绑定，后续通过awaitFirst调用无法覆盖原有上下文。

更优实现方式 #

利用Spring AOP对suspend类型通知的支持（你使用的快照版本已修复#22462问题，满足这个条件），直接将通知声明为suspend函数，结合Kotlin协程的withContext切换上下文，就能让目标函数在指定的自定义上下文里执行，无需逐个修改目标方法。

步骤1：实现自定义协程上下文元素

确保你的CustomContext实现了协程上下文元素接口：

class CustomContext : AbstractCoroutineContextElement(CustomContext) {
    companion object Key : CoroutineContext.Key<CustomContext>
    // 可添加自定义属性或方法，比如用户ID、请求标识等
}

步骤2：修改AOP通知为suspend函数

将@Around通知改为suspend类型，在withContext中执行连接点：

suspend fun joinPoint() {
    // 在目标函数中可直接获取自定义上下文
    val currentContext = coroutineContext[CustomContext]
    // 业务逻辑
}

@Around("execution (* *.joinPoint(..))")
suspend fun advice(pjp: ProceedingJoinPoint): Any? {
    val customContext = CustomContext()
    // 切换协程上下文后执行目标函数
    return withContext(customContext) {
        pjp.proceed()
    }
}

方案原理 #

当AOP通知是suspend函数时，Spring会正确处理协程上下文的传递逻辑：withContext会将当前协程的上下文替换为自定义上下文，而pjp.proceed()调用目标suspend函数时，会自动继承这个新的上下文，目标函数内部通过coroutineContext就能获取到CustomContext实例。

为什么之前的方案无效 #

你之前的通知返回Mono，并用mono(customContext)包裹，但pjp.proceed()返回的Mono是Spring对目标suspend函数的包装，它的协程上下文已经绑定了原有的执行上下文，后续在新上下文里调用awaitFirst只是等待结果，并不会改变目标函数执行时的上下文环境。

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