這篇文章將為大家詳細講解有關基于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進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
