如何在执行长耗时计算的Spring WebFlux Kotlin控制器中避免使用GlobalScope

原有代码无法立即返回的原因 #

coroutineScope 是挂起函数，设计上会等待其内部启动的所有子协程执行完成后才会返回，所以你原来的写法必然会阻塞到长耗时计算结束才能返回runId，和你的预期不符。

符合结构化并发规范的实现方案 #

你不需要使用GlobalScope，只需要自定义一个独立于请求协程生命周期的CoroutineScope，和Spring应用的生命周期绑定即可，这是该场景下的标准实现方案。

步骤1：定义自定义协程作用域的Spring Bean

import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.SupervisorJob
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Configuration

@Configuration
class CoroutineConfig {
    // SupervisorJob 保证单个任务失败不会影响其他后台任务
    @Bean
    fun longRunningTaskScope(): CoroutineScope = CoroutineScope(SupervisorJob() + Dispatchers.Default)
}

步骤2：注入作用域启动后台任务

import org.springframework.web.reactive.function.server.coRouter
import org.springframework.stereotype.Component

@Component
class ComputationRoutes(
    // 注入自定义的长任务协程作用域
    private val longRunningTaskScope: CoroutineScope
) {
    fun route() = coRouter {
        POST("/big-computation") { request: ServerRequest ->
            val params = request.awaitBody<LongRunningComputationParams>()
            val runId = GlobalResultStorage.prepareRun(params)
            // 用独立作用域启动后台任务，和请求处理协程无父子绑定，不会被等待
            longRunningTaskScope.launch {
                runCatching { longRunningComputation(params) }
                    .onSuccess { GlobalResultStorage.addResult(runId, it) }
                    .onFailure { 
                        // 建议记录错误，标记任务失败，避免调用方无限轮询
                        GlobalResultStorage.markRunFailed(runId, it.localizedMessage)
                    }
            }
            ok().bodyValueAndAwait(runId)
        }
    }
}

方案优势 #

• 完全符合结构化并发规范：可以统一管控所有后台长耗时任务，Spring应用优雅停机时，可主动调用longRunningTaskScope.cancel()终止所有仍在运行的计算任务，避免进程无法正常退出。
• 灵活性更高：你可以为该作用域统一配置异常处理器、调度器参数，也可以将任务的Job对象存入结果存储中，支持调用方主动取消任务。
• 完全满足业务需求：后台任务和请求协程生命周期完全解耦，请求会直接返回runId，无需等待计算完成。

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