为什么在调用 .block() 时反应器上下文会抛出“上下文不包含键”？

如何解决为什么在调用 .block() 时反应器上下文会抛出“上下文不包含键”？

我有一个注入上下文的 WebFilter

public class MyWebFilter implements WebFilter {
  @Override
  public Mono<Void> filter(ServerWebExchange exchange,WebFilterChain chain) {
    return chain.filter(exchange)
      .contextWrite(ctx -> ctx.put("context","some-context"));
  }
}

在我的Controller中，我可以读取并成功返回

@RestController
public class MyController {
  @GetMapping(path = "test")
  public Mono<String> test() {
    return Mono.deferContextual(ctx -> Mono.just(ctx.get("context")));
  }
}

但是，如果我在其上调用 .block()：

Mono.deferContextual(ctx -> Mono.just(ctx.get("context"))).block();

它抛出这个异常

java.util.NoSuchElementException：上下文不包含键：上下文

我无法理解响应式世界中的执行顺序。

解决方法

您会希望 Spring 尽可能“现实地”处理这个问题。创建一个测试并使用 WebTestClient。 （不要调用 .block() ever 否则你会用六个用户对你的服务器进行 DoS。不要直接注入/调用你的控制器方法，你需要所有的代理、过滤器等）

这是 kotlin，但您应该能够弄清楚：

@SpringBootApplication
class WebfluxScratchpadApplication {
    @Bean
    fun filter(): WebFilter = WebFilter { serverWebExchange,webFilterChain ->
        webFilterChain.filter(serverWebExchange)
            .contextWrite {
                it.put("myContextKey","!!")
            }
    }
}

@RestController
class MyController {
    @GetMapping("/foo")
    fun foo() = Mono.just("foo")
        .flatMap { s ->
            Mono.deferContextual { ctx ->
                Mono.just(String.format("%s%s",s,ctx["myContextKey"]))
            }
        }
}

fun main(args: Array<String>) {
    runApplication<WebfluxScratchpadApplication>(*args)
}

@WebFluxTest
class WebfluxScratchpadApplicationTests {

    @Autowired
    lateinit var client: WebTestClient

    @Test
    fun stuff() {
        Assertions.assertEquals(
            "foo!!",client.get()
                .uri("/foo")
                .exchange()
                .expectBody<String>()
                .returnResult()
                .responseBody!!
        )
    }
}