SpringBoot+Netty+WebSocket实现实时通信
阅读原文时间:2023年07月09日阅读:2

这篇随笔暂时不讲原理,首先搭建起一个简单的可以实现通信的Demo。之后的一系列随笔会进行一些原理上的分享。

不过在这之前大家最好了解一下Netty的线程模型和NIO编程模型,会对它的整体逻辑有所了解。

更新一篇关于NIO的博客:手动搭建I/O网络通信框架3:NIO编程模型,升级改造聊天室

首先创建好项目后在pom.xml引入Netty依赖

io.netty netty-all

用Netty搭建一个WebSocket服务器整体上需要三样东西,不管是不是用的SpringBoot框架,这三样东西是必不可少的。

1.启动服务器的类(NettyServer),会进行一些初步的配置工作。

2.助手类(Handler),有自己定义的助手类,也有Netty提供的一些基本的助手类,比如对Http、WebSocket支持的助手类。

3.初始化器(Initializer),我们下面使用的是主从线程模型,从线程组里会分配出不同channel去处理不同客户端的请求,而每个channel里就会有各种助手类去实现一些功能。初始化器的作用就是对各种助手类进行绑定。

服务器启动类:

public class NettyServer {
private static int port;

public NettyServer(int port) {  
    this.port = port;  
}  
public static void start() throws InterruptedException {//在main方法里调用这个方法,并用构造函数设置端口号  
    //创建主线程组,接收请求  
    EventLoopGroup bossGroup = new NioEventLoopGroup();  
    //创建从线程组,处理主线程组分配下来的io操作  
    EventLoopGroup workerGroup = new NioEventLoopGroup();  
    //创建netty服务器  
    try {  
        ServerBootstrap serverBootstrap = new ServerBootstrap();  
        serverBootstrap.group(bossGroup, workerGroup)//设置主从线程组  
                .channel(NioServerSocketChannel.class)//设置通道  
                .childHandler(new NettyServerInitializer());//子处理器,用于处理workerGroup中的操作  
        //启动server  
        ChannelFuture channelFuture = serverBootstrap.bind(port).sync();  
        //监听关闭channel  
        channelFuture.channel().closeFuture().sync();  
    } finally {  
        bossGroup.shutdownGracefully();//关闭主线程  
        workerGroup.shutdownGracefully();//关闭从线程  
    }  
}  

}

初始化器:

public class NettyServerInitializer extends ChannelInitializer {
@Override
protected void initChannel(SocketChannel socketChannel) throws Exception {
ChannelPipeline pipeline= socketChannel.pipeline();
//以下三个是Http的支持
//http解码器
pipeline.addLast(new HttpServerCodec());
//支持写大数据流
pipeline.addLast(new ChunkedWriteHandler());
//http聚合器
pipeline.addLast(new HttpObjectAggregator(1024*62));
//websocket支持,设置路由
pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));
//添加自定义的助手类
pipeline.addLast(new NettyHandler());
}
}

自定义助手类:

这个类就是业务的核心,客户端的请求会在这里处理。比如客户端连接、客户端发送消息、给客户端发送消息等等。

自定义助手类需要重写的方法可以根据自己的需求重写,这里就不把每个方法都重写一遍了,完整的大家可以去找找文档看看。

如果需要在助手类中用到@Autowire注解,可以参考这个博客,网上有很多说明,这里就不再重复了https://blog.csdn.net/weixin_30828379/article/details/95009595

public class NettyHandler extends SimpleChannelInboundHandler {//TextWebSocketFrame是netty用于处理websocket发来的文本对象
  //所有正在连接的channel都会存在这里面,所以也可以间接代表在线的客户端
public static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
  //在线人数
public static int online;
//接收到客户都发送的消息
@Override
public void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
SendAllMessages(ctx,send_message);//send_message是我的自定义类型,前后端分离往往需要统一数据格式,可以先把对象转成json字符串再发送给客户端
}
//客户端建立连接
@Override
public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
channelGroup.add(ctx.channel());
online=channelGroup.size();
System.out.println(ctx.channel().remoteAddress()+"上线了!");
}
//关闭连接
@Override
public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
channelGroup.remove(ctx.channel());
online=channelGroup.size();
System.out.println(ctx.channel().remoteAddress()+"断开连接");
}

//出现异常  
@Override  
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {  
    cause.printStackTrace();  
}

//给某个人发送消息  
private void SendMessage(ChannelHandlerContext ctx, Send\_Message msg) {  
    ctx.channel().writeAndFlush(new TextWebSocketFrame(JSON.toJSONString(msg)));  
}

//给每个人发送消息,除发消息人外  
private void SendAllMessages(ChannelHandlerContext ctx,Send\_Message msg) {  
    for(Channel channel:channelGroup){  
        if(!channel.id().asLongText().equals(ctx.channel().id().asLongText())){  
            channel.writeAndFlush(new TextWebSocketFrame(JSON.toJSONString(msg)));  
        }  
    }  
}  

}

前端创建WebSocket对象进行访问

通过socket就可以用一些api进行发送消息,接收消息的操作。然后把接收的数据按各自的需求展示出来就行了,前端部分就不再赘述了。

下面8088端口记得要在main方法里面设置。

if (window.WebSocket) {
var host = window.location.hostname;
var url = "ws://" + host + ":8088/ws";
var socket = new WebSocket(url);
}else{
alert("你的浏览器不支持WebSocket。请不要使用低版本的IE浏览器。");
}