如何自定义 websocket 握手和/或添加过滤器？

Posted 2023-02-27

【中文标题】如何自定义 websocket 握手和/或添加过滤器？

【英文标题】：How to customize websocket handshake and/or add a filter?

【发布时间】：2019-02-21 12:39:50

【问题描述】：

我需要以某种方式验证 websocket 请求。

有没有办法使用带有 reactor-netty 的 spring webflux 来实现？

spring-boot 2.0.5.Release

reactor-netty 0.7.9.Release

【参考方案1】：

这些版本包含使用 websocket 自定义某些东西的有限可能性。 我发现的唯一方法是扩展 HandshakeWebSoketService：

public class CustomHandshakeWebSocketService extends HandshakeWebSocketService 

  public CustomHandshakeWebSocketService(
      RequestUpgradeStrategy upgradeStrategy
  ) 
    super(upgradeStrategy);
  

  @Override
  public Mono<Void> handleRequest(
      ServerWebExchange exchange,
      WebSocketHandler handler
  ) 
    ServerHttpRequest request = exchange.getRequest();
    if ( //do some validation and if valid delegate to chain) 

      return super.handleRequest(exchange, handler);
    
    // If not valid, return error
    return Mono
        .error(new ResponseStatusException(HttpStatus.BAD_REQUEST, "Invalid request"));
  

然后我们必须注册那个自定义服务：

@Bean
public WebSocketHandlerAdapter handlerAdapter() 
  return new WebSocketHandlerAdapter(webSocketService());


@Bean
public WebSocketService webSocketService() 
 ReactorNettyRequestUpgradeStrategy strategy = new ReactorNettyRequestUpgradeStrategy();
    return new CustomHandshakeWebSocketService(strategy);