已解决 - 如何为 Spring 中的 WebSocket 升级请求自定义 HTTP 响应？

Posted 2021-03-19

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

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;

状态码可以用，但正文是空的。怎么发啊？

EDIT：原来问题是我用Firefox控制台检查响应，它没有给我任何显示。如果我使用cURL进行同样的请求，那么一切都能正常工作，我看到了我写到响应体的消息！我有一个WebSocket端点。

答案

要更新响应体，你需要写你的自定义RequestUpateStrategy.As从spring官方文档中的AbstractHandshakeHandler说。

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

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

对于每个servlet容器都有RequestUpgradeStrategy的实现。文件