使用nodejs Websocket 建立one to one即时通讯服务
程序员文章站
2022-06-05 22:04:03
...
需求:使用nodejs-websocket 建立一对一的即时通信服务
环境:nodejs
设计思想:保存消息通道,通过消息通道返回消息.
首先,要知道websocket 是什么?
WebSocket建立在TCP协议之上,并且与HTTP协议有着良好的兼容性,最大的特点就是服务器可以主动向客户端推送消息,客户端也可以主动向服务器发送消息。
其次,nodejs-websocket 又是什么?
看这里https://www.npmjs.com/package/nodejs-websocket
既然都看到这里了,说明对基础已经掌握,so对node项目初始化、安装nodejs-websocket不再赘述。
const ws = require('nodejs-websocket')
const PORT = 8000
// 通道
let AllConn = [{ userId: 'test001' }]
// 消息记录
let messageList = []
// 创建一个服务
var server = ws.createServer(function (conn) {
console.log('New connection')
console.log(AllConn.length)
// 客户端有消息的时候的回到函数
conn.on('text', function (str) {
const msg = JSON.parse(str)
// 建立通道
if (msg.type === 'get-pass') {
// 判断是否存在
let exist = AllConn.some(item => {
return item.userId === msg.userId // true or false
})
if (exist) {
// 存在替换key
AllConn.forEach((item) => {
if (item.userId === msg.userId) {
item.wskey = conn.key
}
})
} else {
AllConn.push({
'userId': msg.userId,
'wskey': conn.key
})
conn.sendText(JSON.stringify({ code: '1', msg: '通道创建成功' }))
}
}
// 通过接收消息人标识,查找通道
if (msg.type === 'get-enter') {
// 组织需要发送的信息包
let conversations = {
formId: msg.userId,
toId: msg.toId,
msgUserId: msg.msgUserId, // 消息id = userId
content: msg.content,
createTime: msg.createTime,
timeStr: getMyTime(msg.createTime),
read: false
}
// 接收到消息返回
conn.sendText(JSON.stringify(conversations))
AllConn.forEach(item => {
// 通道存在
if (item.userId === msg.toId) {
// 获取通道key,发送数据包
broadcast(conversations, item.wskey)
}
})
}
})
conn.on('close', (code, res) => {
// 客户端断开连接
console.log(conn.key + '已经断开')
AllConn.forEach((item, index, arr) => {
if (item.wskey === conn.key) {
// 删除通道记录
arr.splice(index, 1)
}
})
})
conn.on('error', function (err) {
console.log('错误:' + JSON.stringify(err))
})
}).listen(PORT)
/**
* @description:广播功能
* @param {Object} str 消息包
* @param {Object} wskey 通道key
* @return:
*/
function broadcast (str, wskey) {
// 遍历所有通道
server.connections.forEach(conn => {
const key = conn.key
// 查找指定通道
if (key === wskey) {
conn.sendText(JSON.stringify(str))
}
})
}
/**
* @description: 时间戳转时间
* @param {type}
* @return:
*/
function getMyTime (timestamp) {
let date = new Date(timestamp)
let hours = date.getHours()
let minutes = date.getMinutes()
return `${hours >= 10 ? hours : '0' + hours}:${minutes >= 10 ? minutes : '0' + minutes}`
}
如哪位大佬发现不完美、有误及更好的方法.批评指正.
上一篇: C语言自问自答
下一篇: python3实现爬取淘宝美食代码分享