netty结合websocket使用
首先需要在后台建立netty服务器启动类;
package com.cxy; 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; /*** * @ClassName: WsNetty * @Description: * @Auther: cxy * @Date: 2019/2/5:14:15 * @version : V1.0 */ public class WsNetty { public static void main(String[] args) throws InterruptedException { /* 主从线程组模型 */ EventLoopGroup mainGroup =new NioEventLoopGroup(); EventLoopGroup subGroup=new NioEventLoopGroup(); try {
//创建核心类 ServerBootstrap serverBootstrap =new ServerBootstrap(); serverBootstrap.group(mainGroup,subGroup) .channel(NioServerSocketChannel.class) .childHandler(new WSServerInitialzer());//添加助手类 ChannelFuture channelFuture= serverBootstrap.bind(8088).sync(); channelFuture.channel().closeFuture().sync(); } catch (InterruptedException e) { e.printStackTrace(); }finally { 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 java.time.LocalDateTime; 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 (Channel channel: clients) { // channel.writeAndFlush( // new TextWebSocketFrame( // "[服务器在]" + LocalDateTime.now() // + "接受到消息, 消息为:" + 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()); } }
前端页面:
<!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.chat1()" /> <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://127.0.0.1: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("浏览器不支持协议"); } }, chat1:function(){ var msg =document.getElementById("msgContent"); chat.socket.send(msg.value); } }; chat.init(); </script> </body> </html>
测试结果:
笔记转移,由于在有道云的笔记转移,写的时间可能有点久,如果有错误的地方,请指正