Netty应用（五）架设WebSocket服务

Springboot有封装好的Websocket包，只需要几个简单的类和注解就能提供服务。在Java程序中，我们可以使用Netty来实现Websocket服务。

依赖：

compile 'io.netty:netty-all:4.1.63.Final'

同样还是基本的三个文件。

1. 自定义处理器

package com.chris.ws.server.handler;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.util.concurrent.GlobalEventExecutor;

/**
 * @author Chris Chan
 * Create on 2021/4/18 15:14
 * Use for:
 * Explain:
 */
public class WebSocketServerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
        String text = msg.text();
        System.out.println(text);
        channelGroup.writeAndFlush(new TextWebSocketFrame(ctx.channel().remoteAddress() + " : " + text));
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        channelGroup.add(ctx.channel());
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        channelGroup.remove(ctx.channel());
    }
}

负责消息转发，以及客户端上下线的处理。为了代码简洁，我这里不写过多的逻辑。

2. 初始化器。

在Netty应用中，这个地方是最重要的，我们要在这里设置各种编解码器、处理器等等。此时我们需要添加与Websocket相关的编解码器和处理器。

package com.chris.ws.server.handler;

import io.netty.channel.ChannelInitializer;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;

/**
 * @author Chris Chan
 * Create on 2021/4/18 15:10
 * Use for:
 * Explain:
 */
public class WebSocketServerInitializer extends ChannelInitializer<SocketChannel> {
    @Override
    protected void initChannel(SocketChannel ch) throws Exception {
        ch.pipeline()
                .addLast("HttpServerCodec", new HttpServerCodec())
                .addLast("ChunkedWriteHandler", new ChunkedWriteHandler())
                .addLast("HttpObjectAggregator", new HttpObjectAggregator(8192))
                .addLast("WebSocketServerProtocolHandler", new WebSocketServerProtocolHandler("/chat"))
                .addLast("WebSocketServerHandler", new WebSocketServerHandler());
    }
}

Websocket是一个网络请求，所以需要包含对请求响应的处理器。"/chat"是我们规定的后缀，客户端连接必须带着这个后缀才能到达。

3. 测试类。大同小异的一个类。

package com.chris.ws.server;

import com.chris.ws.server.handler.WebSocketServerInitializer;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;

/**
 * @author Chris Chan
 * Create on 2021/4/18 15:09
 * Use for:
 * Explain: WebSocket的支持
 */
public class WebSocketServer {
    public static void main(String[] args) {
        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();

        try {
            ServerBootstrap serverBootstrap = new ServerBootstrap();
            serverBootstrap
                    .group(bossGroup, workerGroup)
                    .channel(NioServerSocketChannel.class)
                    .handler(new LoggingHandler(LogLevel.INFO))
                    .childHandler(new WebSocketServerInitializer());
            serverBootstrap
                    .bind(8001)
                    .sync()
                    .channel()
                    .closeFuture()
                    .sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }

    }
}

好了，我们运行起来，可以测试了。

我们在网上搜索 "在线websocket测试" ，找一个在线测试工具，比如这个：

http://www.jsons.cn/websocket/

输入字符串连接

ws://localhost:8001/chat

发送消息

我们也可以自己写一个JS的客户端，创建一个ws对象，实现四个回调和一个主动发送消息的函数即可。这里我们就不贴源码了，网上参考资料很多。

https://www.runoob.com/html/html5-websocket.html

https://www.cnblogs.com/llljpf/p/10830651.html

https://blog.csdn.net/github_39532240/article/details/85330232