kana/fmhub.js

export default class {
  constructor() {
    this.channels = new Map()
    this.users = new Map()
  }
  订阅频道(fid, uid) {
    console.log(`用户 ${uid} 订阅了 ${fid}`)
    let channel = this.channels.get(fid) || new Map()
    if (!channel.size) this.channels.set(fid, channel)
    channel.set(uid, true)
  }
  取消订阅(fid, uid) {
    console.log(`用户 ${uid} 取消订阅 ${fid} 频道`)
    let channel = this.channels.get(fid)
    channel.delete(uid)
    // TODO: 如果此频道没有用户订阅了, 则将此频道也移除
    // TODO: 如果此用户没有任何订阅了, 则将此用户也移除
  }
  增加会话(uid, ws) {
    console.log(`用户 ${uid} 建立了新的会话连接`)
    let user = this.users.get(uid) || new Map()
    if (!user.size) this.users.set(uid, user)
    user.set(ws, true)
  }
  移除会话(uid, ws) {
    console.log(`用户 ${uid} 结束了当前会话连接`)
    let user = this.users.get(uid) || new Map()
    user.delete(ws)
    console.log(`此用户还剩 ${user.size} 个 ws 连接`)
    if (user.size < 1) {
      console.log(`由于用户 ${uid} 已经没有会话, 直接移除此用户记录`)
      this.users.delete(uid)
      // 理论上在会话结束后移除 (但为了避免反复遍历, 可以放在发送消息时)
      this.channels.forEach((channel, fm) => {
        channel.delete(uid)
      })
    }
  }
  发送消息(fm, uid, data) {
    console.log("发送消息", fm, uid, data)
    let msg = JSON.stringify({ fm, uid, data })
    let channel = this.channels.get(fm) || new Map()
    if (!channel.size) this.channels.set(fm, channel)
    channel.forEach((value, userid) => {
      //console.log(userid, value)
      let user = this.users.get(userid) || new Map()
      if (!user.size) {
        return console.log(`订阅频道的用户 ${userid} 没有会话连接, 应移除此订阅记录`);
        //console.log(user, "在这里移除可能更安全点, 因为用户可能只是断线, 立即就会重连")
        //console.log("但是, 就不能清除闲置频道占用")
        //return channel.delete(userid)
      }
      user.forEach((value, ws) => {
        ws.send(msg)
      })
    })
  }
  移除用户(uid) {
    console.log(`移除 ${uid} 的所有会话`)
    let user = this.users.get(uid)
    if (user) {
      user.forEach((value, ws) => ws.close()) // 断开用户所有会话
      this.users.delete(uid)                  // 删除用户(所有会话)
    }
    // TODO: 移除所有频道中的此用户
    // TODO: 移除此用户的记录
    // TODO: 断开所有此用户的连接
  }
}