Netty WebSocket服務器返回401，瀏覽器無響應的解決策略

在使用Netty構建WebSocket服務器并進行token驗證時，如果token無效，服務器返回401狀態(tài)碼并關閉連接，瀏覽器卻可能無任何反應。本文分析此問題并提供解決方案。

問題描述

使用Netty開發(fā)WebSocket服務器，需要驗證token。token驗證失敗時，服務器返回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(ChannelFutureListener.CLOSE);
    ReferenceCountUtil.release(request);
}

前端JavaScript代碼：

var socket = new WebSocket("ws://127.0.0.1:18080/ws?token=xxxx");

服務器雖然返回401并關閉連接，但瀏覽器卻沒有任何反應。

解決方案

瀏覽器無響應是因為401響應在WebSocket握手階段被忽略了。 我們需要在握手階段或連接建立后進行處理。

方案一：在握手階段返回401響應

在WebSocket握手階段（HTTP請求階段），token驗證失敗則直接返回HTTP 401響應，避免建立WebSocket連接。 修改服務器端代碼：

if (!validateToken(request)) {
    FullHttpResponse response = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.UNAUTHORIZED);
    response.headers().set(HttpHeaderNames.CONTENT_TYPE, "text/plain; charset=utf-8");
    response.headers().set(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes());
    ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE);
    return;
}

方案二：連接建立后發(fā)送自定義消息

如果需要在WebSocket連接建立后再處理401，可以在連接建立后立即發(fā)送自定義消息通知前端。 在Netty的WebSocketServerProtocolHandler中添加自定義處理器：

channel.pipeline().addLast(new WebSocketServerProtocolHandler("/ws") {
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        if (evt == WebSocketServerProtocolHandler.ServerHandshakeStateEvent.HANDSHAKE_COMPLETE) {
            if (!validateToken()) {
                ctx.writeAndFlush(new TextWebSocketFrame("401 Unauthorized")).addListener(ChannelFutureListener.CLOSE);
            }
        }
        super.userEventTriggered(ctx, evt);
    }
});

前端JavaScript代碼需要監(jiān)聽onmessage事件：

socket.onmessage = function(event) {
    if (event.data === "401 Unauthorized") {
        console.log("連接關閉，原因：401 Unauthorized");
        socket.close();
    }
};

通過以上方法，瀏覽器就能正確處理401狀態(tài)碼，前端也能收到相應的反饋信息，從而解決瀏覽器無響應的問題。 選擇哪種方案取決于具體的應用場景和需求。

以上是WebSocket服務器返回401后瀏覽器無反應的原因是什么？如何解決？的詳細內容。更多信息請關注PHP中文網其他相關文章！