基于Node.js的WebSocket通信实现

2024-02-27 20:26:04 47

node的依赖包

node中实现Websocket的依赖包有很多，websocket、ws均可，本文选取ws来实现，首先安装依赖

npminstallws

聊天室实例

假如A,B,C,D用户均通过客户端连接到Websocket服务，其中每个人发的消息都需要将其通过Websocket转发给其他人，此场景类似于服务端将A的消息广播给组内其他用户。

服务端实现

首先来看服务端程序，具体的工作流程分以下几步：

创建一个WebSocketServer的服务，同时监听8080端口的连接请求。
每当有新的客户端连接该WebSocket成功时，便将该连接push到连接池的数组中。
监听message事件，当该事件发生时，遍历连接池，以连接为单位将该消息转发到对应的客户端
监听close事件，当该事件发生时，将该连接移出连接池

服务端代码

varWebSocketServer=require('ws').Server,
wss=newWebSocketServer({port:8080});

//连接池
varclients=[];

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){
returnws1!==ws
})
});
});

客户端实现

如何发现用户？

通过上述的demo可以看到，WebSocket都是基于连接的，也就是说我们知道data是从那个connection发过来，但并不知道使用客户端的是李雷或者韩梅梅，这可如何是好？再想另一种场景，李雷只想给韩梅梅发消息，不想将消息广播给其他客户端，此时我们就需要在Server端能够标识用户身份和连接的对应关系。

于是，需要在客户端连接到WebSocket之后，紧接着再发一次请求，告诉Server我的user_id是多少，Server将此user_id与connection之间的关系存储在hashmap中，至此就建立了user_id与connection的对应关系。当需要发送消息给对应的客户端，从此hashmap中取出对应用户的connection信息，调用其send方法发出消息即可。

依赖包

npminstallhashmap

服务端实现

varWebSocketServer=require('ws').Server,webSocketServer=newWebSocketServer({port:8080});
varHashMap=require('hashmap');

//recordtheclient
varuserConnectionMap=newHashMap();
varconnectNum=0;

//connection
webSocketServer.on('connection',function(ws){
++connectNum;
console.log('Aclienthasconnected.currentconnectnumis:'+connectNum);
ws.on('message',function(message){
varobjMessage=JSON.parse(message);
varstrType=objMessage['type'];

switch(strType){
case'online':
userConnectionMap.set(objMessage['from'],ws);
break;
default:
vartargetConnection=userConnectionMap.get(objMessage['to']);
if(targetConnection){
targetConnection.send(message);
}
}
});

ws.on('close',function(message){
varobjMessage=JSON.parse(message);
userConnectionMap.remove(objMessage['from']);
});
});

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。

基于Node.js的WebSocket通信实现

热门推荐

随机推荐