这篇文章将为大家详细讲解有关基于Node.js如何实现WebSocket通信,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。
node的依赖包
node中实现Websocket的依赖包有很多,websocket、ws均可,本文选取ws来实现,首先安装依赖
npm install ws
聊天室实例
假如A,B,C,D用户均通过客户端连接到Websocket服务,其中每个人发的消息都需要将其通过Websocket转发给其他人,此场景类似于服务端将A的消息广播给组内其他用户。
服务端实现
首先来看服务端程序,具体的工作流程分以下几步:
创建一个WebSocketServer的服务,同时监听8080端口的连接请求。
每当有新的客户端连接该WebSocket成功时,便将该连接push到连接池的数组中。
监听message事件,当该事件发生时,遍历连接池,以连接为单位将该消息转发到对应的客户端
监听close事件,当该事件发生时,将该连接移出连接池
服务端代码
var WebSocketServer = require('ws').Server, wss = new WebSocketServer({port: 8080}); // 连接池 var clients = []; wss.on('connection', function(ws) { // 将该连接加入连接池 clients.push(ws); ws.on('message', function(message) { // 广播消息 clients.forEach(function(ws1){ if(ws1 !== ws) { ws1.send(message); } }) }); ws.on('close', function(message) { // 连接关闭时,将其移出连接池 clients = clients.filter(function(ws1){ return ws1 !== ws }) }); });
客户端实现
<html> <input type="text" id="text"> <input type="button" onclick="sendMessage()" value="online"> <script> var ws = new WebSocket("ws://localhost:8080"); ws.onopen = function (e) { console.log('Connection to server opened'); } ws.onmessage = function(event) { console.log('Client received a message', event); }; ws.onclose = function (e) { console.log('connection closed.'); } function sendMessage() { ws.send(document.getElementById('text').value); } </script> </html>
如何发现用户?
通过上述的demo可以看到,WebSocket都是基于连接的,也就是说我们知道data是从那个connection发过来,但并不知道使用客户端的是李雷或者韩梅梅,这可如何是好?再想另一种场景,李雷只想给韩梅梅发消息,不想将消息广播给其他客户端,此时我们就需要在Server端能够标识用户身份和连接的对应关系。
于是,需要在客户端连接到WebSocket之后,紧接着再发一次请求,告诉Server我的user_id是多少,Server将此user_id与connection之间的关系存储在hashmap中,至此就建立了user_id与connection的对应关系。当需要发送消息给对应的客户端,从此hashmap中取出对应用户的connection信息,调用其send方法发出消息即可。
依赖包
npm install hashmap
服务端实现
var WebSocketServer = require('ws').Server, webSocketServer = new WebSocketServer({port: 8080}); var HashMap = require('hashmap'); // record the client var userConnectionMap = new HashMap(); var connectNum = 0; // connection webSocketServer.on('connection', function(ws) { ++ connectNum; console.log('A client has connected. current connect num is : ' + connectNum); ws.on('message', function(message) { var objMessage = JSON.parse(message); var strType = objMessage['type']; switch(strType) { case 'online' : userConnectionMap.set(objMessage['from'], ws); break; default: var targetConnection = userConnectionMap.get(objMessage['to']); if (targetConnection) { targetConnection.send(message); } } }); ws.on('close', function(message) { var objMessage = JSON.parse(message); userConnectionMap.remove(objMessage['from']); }); });
关于“基于Node.js如何实现WebSocket通信”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。