WebSocketServerHandler 示例代码

以下是一个简单的 WebSocketServerHandler 示例代码，使用 Java 中的 Netty 框架实现：

public class WebSocketServerHandler extends SimpleChannelInboundHandler<WebSocketFrame> {

    private WebSocketServerHandshaker handshaker;

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, WebSocketFrame frame) throws Exception {
        // 判断是否是关闭连接请求
        if (frame instanceof CloseWebSocketFrame) {
            handshaker.close(ctx.channel(), (CloseWebSocketFrame) frame.retain());
            return;
        }

        // 判断是否是 Ping 消息
        if (frame instanceof PingWebSocketFrame) {
            ctx.channel().write(new PongWebSocketFrame(frame.content().retain()));
            return;
        }

        // 只处理文本消息
        if (!(frame instanceof TextWebSocketFrame)) {
            throw new UnsupportedOperationException(String.format("%s frame types not supported", frame.getClass().getName()));
        }

        // 处理文本消息
        String message = ((TextWebSocketFrame) frame).text();
        System.out.printf("Received message: %s%n", message);
        ctx.channel().write(new TextWebSocketFrame("Server received: " + message));
    }

    @Override
    public void channelReadComplete(ChannelHandlerContext ctx) throws Exception {
        ctx.flush();
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        cause.printStackTrace();
        ctx.close();
    }

    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        System.out.println("New client connected: " + ctx.channel().remoteAddress());
    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        System.out.println("Client disconnected: " + ctx.channel().remoteAddress());
    }

    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        if (evt instanceof WebSocketServerProtocolHandler.HandshakeComplete) {
            // WebSocket 握手完成，保存握手信息
            handshaker = ((WebSocketServerProtocolHandler.HandshakeComplete) evt).handshaker();
            System.out.println("WebSocket handshake completed.");
        }
    }
}

该类继承自 Netty 中的 SimpleChannelInboundHandler 类，用于处理 WebSocket 消息。在 channelRead0 方法中，根据消息类型进行不同的处理。如果是关闭连接请求，则关闭连接；如果是 Ping 消息，则回复 Pong 消息；如果是文本消息，则打印接收到的消息，并回复一条消息。

在 handlerAdded 和 handlerRemoved 方法中，分别打印客户端连接和断开连接的信息。

在 userEventTriggered 方法中，处理 WebSocket 握手完成事件，保存握手信息。