如何在 PerConnectionWebSocketHandler 中装饰 websocket 处理程序？

Question

我正在像这样使用 PerConnectionWebSocketHandler：

@Bean
public WebSocketHandler myHandler() {
    return new PerConnectionWebSocketHandler(MyHandler.class));
}

其中 MyHandler 是一个实现 Spring WebSocketHandler 接口的类。

我想添加一个异常处理装饰器，可以像这样创建单个实例：

new ExceptionWebSocketHandlerDecorator(new MyHandler() );

我需要 ExceptionWebSocketHandlerDecorator 由 PerConnectionWebSocketHandler 包装。当 ExceptionWebSocketHandlerDecorator 在外面时，MyHandler 抛出的异常导致 MyHandler.afterConnectionClosed 没有被调用。

如何将它与 PerConnectionWebSocketHandler 结合使用？

Answer 1

WebSocketHandlers 可以相互嵌套：

@Bean
public WebSocketHandler myHandler() {
    return new PerConnectionWebSocketHandler(MyHandler.class));
}

@Bean
public WebSocketHandler myHandlerDecorator() {
    return new ExceptionWebSocketHandlerDecorator(myHandler());
}

这是针对情况，当您打算使用 BeanFactory 自动连接您的目标 MyHandler 或通过简单的方式：

@Bean
public WebSocketHandler myHandlerDecorator() {
    return new ExceptionWebSocketHandlerDecorator(
                   new PerConnectionWebSocketHandler(MyHandler.class)));
}

如果没有。

此外，您还可以考虑链接到LoggingWebSocketHandlerDecorator。

更新

根据您的评论，我建议您编写自己的ExceptionWebSocketHandler 或在您的MyHandler 中实现所有ExceptionWebSocketHandlerDecorator 逻辑。这是因为PerConnectionWebSocketHandler 需要具有默认构造函数的类，即使它被委托给this.beanFactory.createBean(this.handlerType)。

更新2

这种方法的困难在于我有多个处理程序，

那么，你可以实现自己的PerConnectionExceptionWebSocketHandler！与ExceptionWebSocketHandlerDecorator 和PerConnectionWebSocketHandler 的逻辑相同。或者只是扩展 PerConnectionWebSocketHandler 并覆盖您需要的内容。