node的依賴包
node中實現(xiàn)Websocket的依賴包有很多，websocket、ws均可，本文選取ws來實現(xiàn)，首先安裝依賴
npm install ws
聊天室實例
假如A,B,C,D用戶均通過客戶端連接到Websocket服務，其中每個人發(fā)的消息都需要將其通過Websocket轉發(fā)給其他人，此場景類似于服務端將A的消息廣播給組內其他用戶。
服務端實現(xiàn)
首先來看服務端程序，具體的工作流程分以下幾步：
服務端代碼
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 }) });});客戶端實現(xiàn)
<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>如何發(fā)現(xiàn)用戶？
通過上述的demo可以看到，WebSocket都是基于連接的，也就是說我們知道data是從那個connection發(fā)過來，但并不知道使用客戶端的是李雷或者韓梅梅，這可如何是好？再想另一種場景，李雷只想給韓梅梅發(fā)消息，不想將消息廣播給其他客戶端，此時我們就需要在Server端能夠標識用戶身份和連接的對應關系。
于是，需要在客戶端連接到WebSocket之后，緊接著再發(fā)一次請求，告訴Server我的user_id是多少，Server將此user_id與connection之間的關系存儲在hashmap中，至此就建立了user_id與connection的對應關系。當需要發(fā)送消息給對應的客戶端，從此hashmap中取出對應用戶的connection信息，調用其send方法發(fā)出消息即可。
依賴包
npm install hashmap
服務端實現(xiàn)
var WebSocketServer = require('ws').Server, webSocketServer = new WebSocketServer({port: 8080});var HashMap = require('hashmap');// record the clientvar userConnectionMap = new HashMap();var connectNum = 0;// connectionwebSocketServer.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']); });});以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持VeVb武林網(wǎng)。
新聞熱點
疑難解答
