已解决 - 如何在 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 的春季官方文档中所说：

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

如果您使用 Tomcat 作为 servlet 容器，您可以扩展 TomcatRequestUpgradeStrategyclass 并提供您自己的逻辑如何处理 websocket 升级。

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