无法在 reactor-netty 版本 0.9.10 上设置 WriteTimeout

Question

我使用具有 reactor-netty 版本 0.9.10 的 Spring WebFlux 版本 2.3.0.RELEASE 编写了一个反应式 API。作为 API 的 SLA 的一部分，如果服务器占用的时间超过规定的配置 WriteTimeout，我想使请求超时。

分享下面我为 NettyReactiveWebServerFactory 实现定制器的代码。

@Bean
  public WebServerFactoryCustomizer serverFactoryCustomizer() {
    return new NettyTimeoutCustomizer();
  }

  class NettyTimeoutCustomizer implements WebServerFactoryCustomizer<NettyReactiveWebServerFactory> {

    @Override
    public void customize(NettyReactiveWebServerFactory factory) {
      int connectionTimeout = 1000;
      int writeTimeout = 1;
          factory.addServerCustomizers(server -> server.tcpConfiguration(tcp ->
              tcp.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, connectionTimeout)
                  .doOnConnection(connection ->
                      connection.addHandlerLast(new WriteTimeoutHandler(writeTimeout)))));
    }
  }

尽管有定制器，但 WriteTimeout 对 API 不起作用。

Answer 1

不是定义WebServerFactoryCustomizer bean，而是创建一个NettyReactiveWebServerFactory 的bean 来覆盖Spring 的自动配置。

  @Bean
  public NettyReactiveWebServerFactory nettyReactiveWebServerFactory() {
    NettyReactiveWebServerFactory webServerFactory = new NettyReactiveWebServerFactory();
    webServerFactory.addServerCustomizers(new MyCustomizer());
    return webServerFactory;
  }

现在MyCustomizer 看起来像这样：

public class MyCustomizer implements NettyServerCustomizer {

  @Override
  public HttpServer apply(HttpServer httpServer) {
    return httpServer.tcpConfiguration(tcpServer -> tcpServer.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 1000)
        .bootstrap(serverBootstrap -> serverBootstrap.childHandler(new ChannelInitializer<Channel>() {
          @Override
          protected void initChannel(Channel channel) throws Exception {
            channel.pipeline().addLast("writeTimeoutHandler", new WriteTimeoutHandler(1));
          }
        }))
    );
  }
}

这是official API doc中建议的方式