在Netty4 WebSocket服务器中，如何确保浏览器正确接收到401响应？

使用Netty 在构建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响应，并可以根据状态代码进行相应的处理，如显示错误信息或重定向登录页面。 记住添加必要的日志记录，以便在出现问题时快速诊断。

以上是Nettty4 如何确保浏览器在WebSocket服务器中正确接收401响应？详情请关注图灵教育的其他相关文章！