Add coroutine variant of WebExceptionHandler

See gh-32931
This commit is contained in:
Sébastien Deleuze
2024-06-11 10:09:17 +02:00
parent 3d8488a7d8
commit 611367e4bb
2 changed files with 43 additions and 0 deletions

View File

@@ -0,0 +1,15 @@
package org.springframework.web.server
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.reactor.mono
import reactor.core.publisher.Mono
import kotlin.coroutines.CoroutineContext
abstract class CoWebExceptionHandler : WebExceptionHandler {
final override fun handle(exchange: ServerWebExchange, ex: Throwable): Mono<Void> {
val context = exchange.attributes[CoWebFilter.COROUTINE_CONTEXT_ATTRIBUTE] as CoroutineContext?
return mono(context ?: Dispatchers.Unconfined) { coHandle(exchange, ex) }.then()
}
protected abstract suspend fun coHandle(exchange: ServerWebExchange, ex: Throwable)
}