雅荷心语博客
雅荷心语
心之所向便是光
  • 首页
  • 前端之旅
  • 后端之路
  • 软件工具
  • 心灵鸡汤
  • 心路历程
  • 视频资料
  • 关于我们
    • 关于我
    • 关于我
  • 微信平台
    • 业务合作
  • 首页
  • 前端之旅
  • 后端之路
  • 软件工具
  • 心灵鸡汤
  • 心路历程
  • 视频资料
  • 关于我们
    • 关于我
    • 关于我
  • 微信平台
    • 业务合作
  • 关注本站
    • 微信
    • 微博
    • 腾讯微博
    • Twitter
    • Facebook
    • RSS订阅
Hi, 请登录     我要注册     找回密码

使用ws开发简易的聊天室

2023-07-24 分类:前端之旅 阅读(585) 评论(0)

ws 是什么?

https://www.npmjs.com/package/ws

ws 是一个 nodejs 的 webscoket 库, 用来创建 websocket 服务。

下面用简单的思路来实现一个聊天室应用的 nodejs 端服务。

JavaScript
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
import WebSocket, { WebSocketServer } from 'ws';
import express from 'express';
import chalk from 'chalk';
 
const pageConfig = {
ping: 5000, // 心跳检测
msgPort: 3000, //内部推送端口
socketPort: 8020, // socket端口
}
const app = express();
const wss = new WebSocketServer({ port: pageConfig.socketPort });
const groups = new Map();
wss.on('connection', function connection(ws) {
ws.on('message', function incoming(message) {
console.log('received: %s', message);
const data = JSON.parse(message);
ws.uid = data?.uid ?? '';
const action = data?.action ?? '';
const room = data?.room ?? '';
sameUser(ws); // 同一个用户在不同的客户端打开; 则把这些用户加入到一个房间内
if (action === 'login') {
// 登陆系统; 返回令牌
replyData({}, action, ws);
}
// 加入房间
if (action === 'join') {
// 加入到房间;
room && addGroup(room, ws);
}
// 数据转发到其他room
if (action === 'transmit') {
// 将本次数据转发到此分组所有用户
sendMessageToGroup(room, data, action);
}
// 其他操作
if (action === 'other') {}
});
  ws.on('close', function close() {
  console.log(ws.userId, '离开了');
    groups.forEach(function (value, key) {
      if (value.has(ws)) {
        value.delete(ws);
        if (value.size === 0) {
          groups.delete(key);
        }
      }
    });
  });
});
/**
* 回复消息给客户端
* @param data
* @param action
* @param ws
*/
const replyData = (data, action = '', ws) => {
const config = JSON.stringify({
txt: data,
action,
time: new Date().getTime(),
uid: ws.uid
});
console.log(chalk.green(`🚀 回复消息`), action, config);
ws.send(config);
}
/**
* 加入到房间
* @param group
* @param ws
*/
const addGroup = (group, ws) => {
if (group) {
if (!groups.has(group)) {
groups.set(group, new Set());
}
groups.get(group).add(ws);
console.log(chalk.red(ws.uid), chalk.blue('加入了房间'), group);
}
}
/**
* 多个客户端相同 UID 用户处理
* @param ws
*/
const sameUser = (ws) => {
// 查找所有在线用户里 uid 等于自己的, 加一个分组里;
// Set 会自动过滤重复的
wss.clients.forEach(function each(client) {
// 把所有的自己都加入到 room Id 是 uid 的组里;
if (client.readyState === WebSocket.OPEN && client.uid === ws.uid) {
addGroup(ws.uid, client);
}
});
}
/**
* 推送消息给房间(分组)
* @param group
* @param message
* @param action
*/
const sendMessageToGroup = (group, message, action = '') => {
if (groups.has(group)) {
groups.get(group).forEach(function each(client) {
if (client.readyState === WebSocket.OPEN) {
client.send(message);
replyData(message, action, client);
}
});
}
}
/**
* 给所有人发消息
* @param txt
*/
const sendAll = (txt) => {
wss.clients.forEach(function each(client) {
if (client.readyState === WebSocket.OPEN) {
client.send(txt);
}
});
}
app.get('/', (req, res) => {
const params = req.query;
sendAll(params?.txt ?? '你好');
res.send('Hello, this is a GET request!');
});
// 给房间发消息
app.get('/send', (req, res) => {
const params = req?.query ?? {};
const action = params?.action ?? '';
sendMessageToGroup(params?.room, params?.txt ?? '你好', action);
res.send('Hello, Send Success');
});
// 获取房间个数
app.get('/count', (req, res) => {
let size = 0;
groups.forEach(function (value, key) {
const room = req?.query?.room ?? '';
if (key === room) size = value?.size;
});
res.send(size.toString());
})
// 心跳检测;
setInterval(() => {
wss.clients.forEach(function each(client) {
if (client.readyState === WebSocket.OPEN) {
replyData(`Hello ~ 你还在吗?`, 'ping', client);
} else {
// 删除已关闭的客户端
groups.forEach(function (value, key) {
console.log(`名称`, key, '在线个数', value.size);
if (value.has(client)) {
value.delete(client);
if (value.size === 0) {
groups.delete(key);
}
}
});
}
});
}, pageConfig.ping);
 
app.listen(pageConfig.msgPort, () => {
console.log(chalk.blue(`内部推送服务已启动: ${pageConfig.msgPort}`));
});

 

服务包含了心跳检测, 多分组, 内部推送服务等!

分享到:更多 ()

相关推荐

  • AI 编辑器 cursor 如何禁止自动更新
  • AI 编辑器 cursor 工具栏改成和 vscode 一样的左侧 竖向展示
  • nodejs 脚本打包为可执行文件
  • 初学 python 笔记
  • nodejs 同时运行多个脚本
  • 让你的照片动起来
  • vue工程项目动态加载umd.js实践
  • 使用 shell 检测目标服务器是否异常
关于我

小天明 北京·朝阳 前端搬砖工程师

碎碎念):(来自公众号)

热门文章

  • 踩坑记录——iphone上safari开启隐身模式时localStorage变为只读-雅荷心语博客踩坑记录——iphone上safari开启隐身模式时localStorage变为只读2017-02-21评论(4)
  • 程序员是怎样一群人-雅荷心语博客程序员是怎样一群人2015-12-08评论(3)
  • 百度你个大毒瘤 - 吐糟博客这几天打不开事情-雅荷心语博客百度你个大毒瘤 – 吐糟博客这几天打不开事情2015-12-28评论(2)
  • PHP 非对称加密 openssl 加密及解密方法-雅荷心语博客PHP 非对称加密 openssl 加密及解密方法2016-05-17评论(2)
  • PHPStorm10 下载安装破解汉化-雅荷心语博客PHPStorm10 下载安装破解汉化2015-12-15评论(2)
2025年7月
一 二 三 四 五 六 日
« 六    
 123456
78910111213
14151617181920
21222324252627
28293031  

最新评论

  • 前端小武 8年前 (2017-04-06)说:
    我看到了layer
  • 丁艳平 8年前 (2017-03-03)说:
  • Dawn 9年前 (2016-09-16)说:
    call_user_func_array最后的例子是错哦,你用bc方法去调用类里 另外一个方法就知道问题所在了。情况1.调用非静态方法 第一个参数应该传[类的实例,调用方法] (既然有类实例了直接-&
  • Dawn 9年前 (2016-06-21)说:
    tp框架设置了全局捕获异常的,这也没什么。坑的是 他捕获了异常。然后全部返回404。。。不知道的 还以为自己网站被删除了
  • Dawn 9年前 (2016-05-17)说:
    构造函数里的判断 用异常机制可能更好一些

其他类型

  • 芊云全景
  • 配音兔AI配音神器

博客类型

  • 芊云全景
  • 配音兔AI配音神器

左邻右舍

  • 易水寒
  • 楼教主
  • 芊云全景
  • 贤心
  • 配音兔AI配音神器

雅荷心语博客 -心之所向便是光

联系我们关于我们

© 2025 雅荷心语博客   网站地图