Netty进阶篇之websocket发送消息（8）

时间 2019-11-20

原文原文链接

序言：Netty进阶篇之简单版websocket发消息(7) 大概和下面的代码就成类似，只不过上一篇博客更加简单一些。javascript

一、pom文件

<dependency>
    <groupId>io.netty</groupId>
    <artifactId>netty-all</artifactId>
    <version>4.1.25.Final</version>
</dependency>

二、index.html

<!DOCTYPE html>
<html>
   <head>
      <meta charset="utf-8" />
      <title></title>
   </head>
   <body>
      <div>发送消息:</div>
      <input type="text" id="msgContent"/>
      <input type="button" value="点我发送" onclick="CHAT.chat()"/>
      
      <div>接受消息：</div>
      <div id="receiveMsg" style="background-color: gainsboro;"></div>
      
      <script type="application/javascript">
         
         window.CHAT = {
            socket: null,
            init: function() {
               if (window.WebSocket) {
                  CHAT.socket = new WebSocket("ws://192.168.31.160:8088/ws");
                  CHAT.socket.onopen = function() {
                     console.log("链接创建成功...");
                  },
                  CHAT.socket.onclose = function() {
                     console.log("链接关闭...");
                  },
                  CHAT.socket.onerror = function() {
                     console.log("发生错误...");
                  },
                  CHAT.socket.onmessage = function(e) {
                     console.log("接受到消息：" + e.data);
                     var receiveMsg = document.getElementById("receiveMsg");
                     var html = receiveMsg.innerHTML;
                     receiveMsg.innerHTML = html + "<br/>" + e.data;
                  }
               } else {
                  alert("浏览器不支持websocket协议...");
               }
            },
            chat: function() {
               var msg = document.getElementById("msgContent");
               CHAT.socket.send(msg.value);
            }
         };
         CHAT.init();
      </script>
   </body>
</html>

三、main函数

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;

public class WSServer {

   public static void main(String[] args) throws Exception {
      
      //定义一对线程组 
      // 主线程组, 用于接受客户端的链接，可是不作任何处理，跟老板同样，不作事      
      EventLoopGroup mainGroup = new NioEventLoopGroup();
      // 从线程组, 老板线程组会把任务丢给他，让手下线程组去作任务
      EventLoopGroup subGroup = new NioEventLoopGroup();
      
      try {
         // netty服务器的建立, ServerBootstrap 是一个启动类
         ServerBootstrap server = new ServerBootstrap();
         server.group(mainGroup, subGroup)    // 设置主从线程组
            .channel(NioServerSocketChannel.class)    // 设置nio的双向通道
            .childHandler(new WSServerInitialzer());  //// 子处理器，用于处理workerGroup
         
         // 启动server，而且设置8088为启动的端口号，同时启动方式为同步
         ChannelFuture future = server.bind(8088).sync();
         
         future.channel().closeFuture().sync();
      } finally {
         // 监听关闭的channel，设置位同步方式
         mainGroup.shutdownGracefully();
         subGroup.shutdownGracefully();
      }
   }
}

四、初始化类

import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
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;

public class WSServerInitialzer extends ChannelInitializer<SocketChannel> {

   @Override
   protected void initChannel(SocketChannel ch) throws Exception {
      ChannelPipeline pipeline = ch.pipeline();
      
      // websocket 基于http协议，因此要有http编解码器
      pipeline.addLast(new HttpServerCodec());
      // 对写大数据流的支持 
      pipeline.addLast(new ChunkedWriteHandler());
      // 对httpMessage进行聚合，聚合成FullHttpRequest或FullHttpResponse
      // 几乎在netty中的编程，都会使用到此hanler
      pipeline.addLast(new HttpObjectAggregator(1024*64));
      
      // ====================== 以上是用于支持http协议    ======================
      
      // ====================== 如下是支持httpWebsocket ======================
      
      /**
       * websocket 服务器处理的协议，用于指定给客户端链接访问的路由 : /ws
       * 本handler会帮你处理一些繁重的复杂的事
       * 会帮你处理握手动做： handshaking（close, ping, pong） ping + pong = 心跳
       * 对于websocket来说，都是以frames进行传输的，不一样的数据类型对应的frames也不一样
       */
      pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));
      
      // 自定义的handler
      pipeline.addLast(new ChatHandler());
   }
}

五、助手类

import io.netty.channel.Channel;
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;

/**
 * 
 * @Description: 处理消息的handler
 * TextWebSocketFrame： 在netty中，是用于为websocket专门处理文本的对象，frame是消息的载体
 */
public class ChatHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {

   // 用于记录和管理全部客户端的channle
   private static ChannelGroup clients = 
         new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
   
   @Override
   protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) 
         throws Exception {
      // 获取客户端传输过来的消息
      String content = msg.text();
      System.out.println("接受到的数据：" + content);

      //相似于for循环
      clients.writeAndFlush(
            new TextWebSocketFrame(
                  "[服务器在]" + LocalDateTime.now() 
                  + "接受到消息, 消息为：" + content));
   }

   /**
    * 当客户端链接服务端以后（打开链接）
    * 获取客户端的channle，而且放到ChannelGroup中去进行管理
    */
   @Override
   public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
      clients.add(ctx.channel());
   }

   @Override
   public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
      // 当触发handlerRemoved，ChannelGroup会自动移除对应客户端的channel
//    clients.remove(ctx.channel());
      System.out.println("客户端断开，channle对应的长id为：" 
                     + ctx.channel().id().asLongText());
      System.out.println("客户端断开，channle对应的短id为：" 
                     + ctx.channel().id().asShortText());
   }
}

六、测试

运行index.htmlhtml

发送消息，以下server端返回消息java