這篇文章將為大家詳細(xì)講解有關(guān)基于Node.js如何實(shí)現(xiàn)WebSocket通信，小編覺(jué)得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

創(chuàng)新互聯(lián)服務(wù)項(xiàng)目包括萍鄉(xiāng)網(wǎng)站建設(shè)、萍鄉(xiāng)網(wǎng)站制作、萍鄉(xiāng)網(wǎng)頁(yè)制作以及萍鄉(xiāng)網(wǎng)絡(luò)營(yíng)銷(xiāo)策劃等。多年來(lái)，我們專(zhuān)注于互聯(lián)網(wǎng)行業(yè)，利用自身積累的技術(shù)優(yōu)勢(shì)、行業(yè)經(jīng)驗(yàn)、深度合作伙伴關(guān)系等，向廣大中小型企業(yè)、政府機(jī)構(gòu)等提供互聯(lián)網(wǎng)行業(yè)的解決方案，萍鄉(xiāng)網(wǎng)站推廣取得了明顯的社會(huì)效益與經(jīng)濟(jì)效益。目前，我們服務(wù)的客戶(hù)以成都為中心已經(jīng)輻射到萍鄉(xiāng)省份的部分城市，未來(lái)相信會(huì)繼續(xù)擴(kuò)大服務(wù)區(qū)域并繼續(xù)獲得客戶(hù)的支持與信任！

node的依賴(lài)包

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

npm install ws

聊天室實(shí)例

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

服務(wù)端實(shí)現(xiàn)

首先來(lái)看服務(wù)端程序，具體的工作流程分以下幾步：

1. 創(chuàng)建一個(gè)WebSocketServer的服務(wù)，同時(shí)監(jiān)聽(tīng)8080端口的連接請(qǐng)求。

2. 每當(dāng)有新的客戶(hù)端連接該WebSocket成功時(shí)，便將該連接push到連接池的數(shù)組中。

3. 監(jiān)聽(tīng)message事件，當(dāng)該事件發(fā)生時(shí)，遍歷連接池，以連接為單位將該消息轉(zhuǎn)發(fā)到對(duì)應(yīng)的客戶(hù)端

4. 監(jiān)聽(tīng)close事件，當(dāng)該事件發(fā)生時(shí)，將該連接移出連接池

服務(wù)端代碼

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) {
    // 連接關(guān)閉時(shí)，將其移出連接池
    clients = clients.filter(function(ws1){
      return ws1 !== ws
    })
  });
});

客戶(hù)端實(shí)現(xiàn)

如何發(fā)現(xiàn)用戶(hù)？

通過(guò)上述的demo可以看到，WebSocket都是基于連接的，也就是說(shuō)我們知道data是從那個(gè)connection發(fā)過(guò)來(lái)，但并不知道使用客戶(hù)端的是李雷或者韓梅梅，這可如何是好？再想另一種場(chǎng)景，李雷只想給韓梅梅發(fā)消息，不想將消息廣播給其他客戶(hù)端，此時(shí)我們就需要在Server端能夠標(biāo)識(shí)用戶(hù)身份和連接的對(duì)應(yīng)關(guān)系。

于是，需要在客戶(hù)端連接到WebSocket之后，緊接著再發(fā)一次請(qǐng)求，告訴Server我的user_id是多少，Server將此user_id與connection之間的關(guān)系存儲(chǔ)在hashmap中，至此就建立了user_id與connection的對(duì)應(yīng)關(guān)系。當(dāng)需要發(fā)送消息給對(duì)應(yīng)的客戶(hù)端，從此hashmap中取出對(duì)應(yīng)用戶(hù)的connection信息，調(diào)用其send方法發(fā)出消息即可。

依賴(lài)包

npm install hashmap

服務(wù)端實(shí)現(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']);
  });
});

關(guān)于“基于Node.js如何實(shí)現(xiàn)WebSocket通信”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺(jué)得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)建站www.cdcxhl.com，海內(nèi)外云服務(wù)器15元起步，三天無(wú)理由+7*72小時(shí)售后在線(xiàn)，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國(guó)服務(wù)器、虛擬主機(jī)、免備案服務(wù)器”等云主機(jī)租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡(jiǎn)單易用、服務(wù)可用性高、性?xún)r(jià)比高”等特點(diǎn)與優(yōu)勢(shì)，專(zhuān)為企業(yè)上云打造定制，能夠滿(mǎn)足用戶(hù)豐富、多元化的應(yīng)用場(chǎng)景需求。

分享文章：基于Node.js如何實(shí)現(xiàn)WebSocket通信-創(chuàng)新互聯(lián)
本文路徑：http://weahome.cn/article/ceihic.html