Netty4 WebSocket服务器中正确处理401响应，确保浏览器接收
在使用Netty 4构建WebSocket服务器时，如果身份验证失败，需要正确地向浏览器发送401 (Unauthorized) 响应。 直接关闭连接可能导致浏览器无法识别错误原因。以下改进后的代码确保浏览器能够接收并处理401响应：
问题：在WebSocket握手阶段，如果身份验证失败，服务器直接关闭连接，浏览器无法接收到预期的401响应。
解决方案：在关闭连接前，必须确保401响应已完全发送到客户端。
改进后的代码：
private void httpResponse401(ChannelHandlerContext ctx, FullHttpRequest request) {
FullHttpResponse response = new DefaultFullHttpResponse(
request.protocolVersion(),
HttpResponseStatus.UNAUTHORIZED);
response.headers().set(HttpHeaderNames.CONTENT_LENGTH, 0);
ctx.writeAndFlush(response).addListener(future -> {
if (future.isSuccess()) {
// 响应成功发送后关闭连接
ctx.close();
} else {
// 响应发送失败，也关闭连接
ctx.close();
// 可选：记录错误日志
logger.error("Failed to send 401 response: {}", future.cause());
}
});
ReferenceCountUtil.release(request);
}
关键改进：
-
addListener用于确保响应发送完成:writeAndFlush方法是异步的。addListener会在响应完全写入通道后执行回调函数，确保在关闭连接前，401响应已发送。 - 错误处理: 添加了错误处理，如果响应发送失败，也会关闭连接并可选地记录错误日志，方便调试。 这对于生产环境至关重要。
- 清晰的代码结构: 代码结构更加清晰，易于理解和维护。
通过此改进，浏览器将接收401响应，并能够根据该状态码进行相应的处理，例如显示错误信息或重定向到登录页面。 记住添加必要的日志记录，以便在出现问题时能够快速诊断。
