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响应，并能够根据该状态码进行相应的处理，例如显示错误信息或重定向到登录页面。 记住添加必要的日志记录，以便在出现问题时能够快速诊断。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
