使用nodejs Websocket 建立one to one即时通讯服务

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

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