基于Node.js的WebSocket通信实现
更新时间：2017年03月11日 10:47:17 作者：董浩_
这篇文章主要介绍了基于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 fs模块(文件模块)创建、删除目录(文件)读取写入文件流的方法
这篇文章主要介绍了Node.js fs模块(文件模块)创建、删除目录(文件)读取写入文件流的方法，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2019-09-09
进程间通信是是Node.js的一个十分重要的部分,这篇文章主要给大家介绍了关于两个Node.js进程是如何通信的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下2021-10-10
node实现shell命令管理工具及commander.js学习
这篇文章主要为大家介绍了node实现shell命令管理工具及commander.js学习，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-09-09
这篇文章主要介绍了node中Stream的详细介绍，流是一个数据传输手段，是端到端信息交换的一种方式，而且是有顺序的,是逐块读取数据、处理内容，用于顺序读取输入或写入输出2022-09-09
这篇文章主要为大家介绍了node.js模拟实现自动发送邮件验证码的实例过程，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-04-04
这篇文章主要为大家介绍了前端面试之输入npm run后发生了什么及执行原理解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-07-07
这篇文章主要给大家介绍了关于Windows7系统下如何安装nodejs16以上版本的相关资料,很多时候node.js的版本存在兼容,文中通过图文介绍的非常详细,需要的朋友可以参考下2023-07-07
这篇文章主要给大家介绍了关于Node.js学习教程之Module模块的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Node.js具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-09-09
net和http模块都是node核心模块之一，他们都可以搭建自己的服务端和客户端，以响应请求和发送请求。这篇文章主要介绍了Nodejs核心模块之net和http的使用详解，感兴趣的小伙伴们可以参考一下2019-04-04
静态资源服务器指的是不会被服务器的动态运行所改变或者生成的文件,本文主要为大家详细介绍了Node.js服务器静态资源处理的相关知识,需要的可以了解下2024-04-04
最新评论