欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

使用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}`
}

如哪位大佬发现不完美、有误及更好的方法.批评指正.