Skip to content

Commit

Permalink
Add coroutine variant of WebExceptionHandler
Browse files Browse the repository at this point in the history
  • Loading branch information
earlgrey02 committed Jun 1, 2024
1 parent bdc4ecd commit 10e7dc7
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
package org.springframework.web.server

import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.reactor.mono
import reactor.core.publisher.Mono

abstract class CoWebExceptionHandler : WebExceptionHandler {
final override fun handle(exchange: ServerWebExchange, ex: Throwable): Mono<Void> =
mono(Dispatchers.Unconfined) { coHandle(exchange, ex) }.then()


protected abstract suspend fun coHandle(exchange: ServerWebExchange, ex: Throwable)
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
package org.springframework.web.server

import org.assertj.core.api.Assertions.assertThat
import org.junit.jupiter.api.Test
import org.springframework.web.testfixture.http.server.reactive.MockServerHttpRequest
import org.springframework.web.testfixture.server.MockServerWebExchange
import reactor.test.StepVerifier

class CoWebExceptionHandlerTest {
@Test
fun handle() {
val exchange = MockServerWebExchange.from(MockServerHttpRequest.get("https://example.com"))
val ex = RuntimeException()

val handler = MyCoWebExceptionHandler()
val result = handler.handle(exchange, ex)

StepVerifier.create(result).verifyComplete()

assertThat(exchange.attributes["foo"]).isEqualTo("bar")
}
}

private class MyCoWebExceptionHandler : CoWebExceptionHandler() {
override suspend fun coHandle(exchange: ServerWebExchange, ex: Throwable) {
exchange.attributes["foo"] = "bar"
}
}

0 comments on commit 10e7dc7

Please sign in to comment.