说明:本篇博客是基于学习某网有关视频教学。目的
:创建一个websocket服务器,获取客户端传来的数据,同时向客户端发送数据
一、服务端
1、Main主类
public class WSServer { public static void main(String[] args) throws Exception { // 定义一对线程组 // 主线程组, 用于接受客户端的连接, EventLoopGroup mainGroup = new NioEventLoopGroup(); // 从线程组, 负责IO交互工作 EventLoopGroup subGroup = new NioEventLoopGroup(); try { //netty服务器的创建, 辅助工具类,用于服务器通道的一系列配置 ServerBootstrap server = new ServerBootstrap(); //绑定两个线程组 server.group(mainGroup, subGroup) //指定NIO的模式 .channel(NioServerSocketChannel.class) //子处理器,用于处理workerGroup .childHandler(new WSServerInitialzer()); // 启动server,并且设置8088为启动的端口号,同时启动方式为同步 ChannelFuture future = server.bind(8088).sync(); // 监听关闭的channel,设置位同步方式 future.channel().closeFuture().sync(); } finally { //退出线程组 mainGroup.shutdownGracefully(); subGroup.shutdownGracefully(); } } }
2、WSServerInitialzer类(子处理器)
public class WSServerInitialzer extends ChannelInitializer<SocketChannel> { @Override protected void initChannel(SocketChannel ch) throws Exception { ChannelPipeline pipeline = ch.pipeline(); // websocket 基于http协议,所以要有http编解码器 服务端用HttpServerCodec pipeline.addLast(new HttpServerCodec()); // 对写大数据流的支持 pipeline.addLast(new ChunkedWriteHandler()); /** * 我们通常接收到的是一个http片段,如果要想完整接受一次请求的所有数据,我们需要绑定HttpObjectAggregator,然后我们 * 就可以收到一个FullHttpRequest-是一个完整的请求信息。 *对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()); } }
3、ChatHandler(助手类)
/** * @Description: 处理消息的handler * TextWebSocketFrame: 在netty中,是用于为websocket专门处理文本的对象,frame是消息的载体 * 这里已经指定了类型 如果这里是Object 那么下面还需判断是不是TextWebSocketFrame类型 */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("我是服务器,我收到你的消息为:" + 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,所以下面的remove不用我们再手写// clients.remove(ctx.channel()); System.out.println("客户端断开,channle对应的长id为:" + ctx.channel().id().asLongText()); System.out.println("客户端断开,channle对应的短id为:" + ctx.channel().id().asShortText()); } }
到这里服务端已经写好了,和之前搭建的服务器大致没什么区别,主要区别在于ChannelPipeline添加了不同的Handel,助手类对websocket做了些处理工作。
二、客户端
客户端这边是采用Hbuilderx工具创建的前端项目,代码如下
<!DOCTYPE html><html> <head> <meta charset="utf-8" /> <title></title> </head> <body> <div>发送消息:</div> <input type="text" id="msgContent"/> <input type="button" value="点我发送" ="CHAT.chat()"/> <div>接受消息:</div> <div id="receiveMsg" style="background-color: gainsboro;"></div> <script type="application/javascript"> window.CHAT = { socket: null, init: function() { <!--判断浏览器是否支持 websocket--> if (window.WebSocket) { <!--连接服务器websocket IP+端口号 /ws是服务器WebSocketServerProtocolHandler添加的--> 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. = 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>
三、测试
通过测试可以总结
1、页面初始化的时候就已经成功和服务端websocket建立连接成功。 2、服务端收到客户端数据,并向客户端发送数据。 3、当关闭页面的时候,既相当于关闭了该websocket连接,服务端会自动移除。
点击查看更多内容
为 TA 点赞
评论
共同学习,写下你的评论
评论加载中...
作者其他优质文章
正在加载中
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦