基于Node.js的WebSocket通信實現(xiàn)

更新時間：2017年03月11日 10:47:17 作者：董浩_

這篇文章主要介紹了基于Node.js的WebSocket通信實現(xiàn)，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

node的依賴包

node中實現(xiàn)Websocket的依賴包有很多，websocket、ws均可，本文選取ws來實現(xiàn)，首先安裝依賴

npm install ws

聊天室實例

假如A,B,C,D用戶均通過客戶端連接到Websocket服務，其中每個人發(fā)的消息都需要將其通過Websocket轉(zhuǎn)發(fā)給其他人，此場景類似于服務端將A的消息廣播給組內(nèi)其他用戶。

服務端實現(xiàn)

首先來看服務端程序，具體的工作流程分以下幾步：

創(chuàng)建一個WebSocketServer的服務，同時監(jiān)聽8080端口的連接請求。
每當有新的客戶端連接該WebSocket成功時，便將該連接push到連接池的數(shù)組中。
監(jiān)聽message事件，當該事件發(fā)生時，遍歷連接池，以連接為單位將該消息轉(zhuǎn)發(fā)到對應的客戶端
監(jiān)聽close事件，當該事件發(fā)生時，將該連接移出連接池

服務端代碼

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信息，調(diào)用其send方法發(fā)出消息即可。

依賴包

npm install hashmap

服務端實現(xiàn)

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']);
  });
});

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: