java - 如何在 Spring 中为 WebSocket 升级请求自定义 HTTP 响应？

Question

我在我的 WebSocketConfigurer 实现中有一个这样配置的 WebSocket 端点

public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new ExceptionWebSocketHandlerDecorator(myWebSocketHandler), "/ws/")
                .setAllowedOrigins("*")
                .addInterceptors(myHandshakeInterceptor);
    }

myWebSocketHandler#beforeHandshake 中有一些检查，如果服务器决定拒绝升级连接，我想向用户发送一条错误消息。这个我试过了

response.setStatusCode(HttpStatus.BAD_GATEWAY);
try {
    response.getBody().write("Error".getBytes());
    response.getBody().flush();
} catch (IOException e) {
    log.error("Error writing response", e);
}
return false;

状态码有效，但正文是空的。如何发送？

编辑:原来问题是我使用 Firefox 控制台检查响应，但它没有显示任何内容。如果我使用 cURL 发出相同的请求，那么一切正常，我会看到我写入响应正文的消息!

Answer 1

要更新响应主体，您需要编写自定义 RequestUpateStrategy。从 AbstractHandshakeHandler 的 spring 官方文档中可以看出:

The actual upgrade is delegated to a server-specific RequestUpgradeStrategy, which will update the response as necessary and initialize the WebSocket.

如果您使用 Tomcat 作为 servlet 容器，您可以扩展 TomcatRequestUpgradeStrategy class 和 provider 自己的逻辑怎么处理网络套接字升级。

对于每个 servlet 容器都有 RequestUpgradeStrategy 的实现 documentation