Nodejs + WebSocket + Vue 一对一、一对多聊天室 – 第三章

<div class="web-im">
  <div class="dis-flex">
    <div class="user-list">
      <div class="user" @click="triggerGroup">群1</div>
      <div class="user" @click="triggerPersonal(item)" v-if="item.uid!=uid" v-for="item in users">{{item.nickname}}</div>
    </div>
    <div class="msg-content">
      <div class="header im-title">{{title}}</div>
        <div class="content im-record">
          <div class="li" :class="{user: item.uid == uid}" v-for="item in currentMessage">
            <template v-if="item.type===1">
              <p class="join-tips">{{item.msg}}</p>
            </template>
            <template v-else>
              <div class="img">{{item.nickname}}</div>
              <p class="message-box">{{item.msg}}</p>
            </template>
          </div>
        </div>
        <div class="footer im-input">
          <input type="text" v-model="msg" placeholder="请输入内容">
          <button @click="send">发送</button>
        </div>
    </div>
  </div>
</div>
...

var ws = require("nodejs-websocket");
var moment = require('moment');

console.log("开始建立连接...")

let users = [];
let conns = {};

function boardcast(obj) {
  // bridge用来实现一对一的主要参数
  if(obj.bridge && obj.bridge.length){
    obj.bridge.forEach(item=>{
      conns[item].sendText(JSON.stringify(obj));
    })
    return;
  }
  server.connections.forEach((conn, index) => {
      conn.sendText(JSON.stringify(obj));
  })
}

var server = ws.createServer(function(conn){
  conn.on("text", function (obj) {
    obj = JSON.parse(obj);
    // 将所有uid对应的连接conn存到一个对象里面
    conns[''+obj.uid+''] = conn;
    if(obj.type===1){
      let isuser = users.some(item=>{
        return item.uid === obj.uid
      })
      if(!isuser){
        users.push({
          nickname: obj.nickname,
          uid: obj.uid
        });
      }
      boardcast({
        type: 1,
        date: moment().format('YYYY-MM-DD HH:mm:ss'),
        msg: obj.nickname+'加入聊天室',
        users: users,
        uid: obj.uid,
        nickname: obj.nickname,
        // 增加参数
        bridge: obj.bridge
      });
    } else {
      boardcast({
        type: 2,
        date: moment().format('YYYY-MM-DD HH:mm:ss'),
        msg: obj.msg,
        uid: obj.uid,
        nickname: obj.nickname,
        // 增加参数
        bridge: obj.bridge
      });
    }
  })
  conn.on("close", function (code, reason) {
    console.log("关闭连接")
  });
  conn.on("error", function (code, reason) {
    console.log("异常关闭")
  });
}).listen(8001)
console.log("WebSocket建立完毕")

server.connections.forEach((conn, index) => {
  conn.sendText(JSON.stringify(obj));
})

if(obj.bridge && obj.bridge.length){
  obj.bridge.forEach(item=>{
    conns[item].sendText(JSON.stringify(obj));
  })
  return;
}

export default {
  ...
  data(){
    return {
      title: '群聊',
      uid: '',
      nickname: '',
      socket: '',
      msg: '',
      // 当前用户所有消息
      messageList: [],
      users: [],
      bridge: []
    }
  },
  mounted() {
    ...
  },
  computed: {
    // 当前对话渲染的msg列表
    currentMessage() {
      let vm = this;
      // 筛选只有bridge相同的对话，展示出来
      // 数组比较，先转成字符串
      let data = vm.messageList.filter(item=>{
        return item.bridge.sort().join(',') == vm.bridge.sort().join(',')
      })
      return data;
    }
  },
  methods: {
    // 切换到群聊
    triggerGroup() {
      this.bridge = [];
      this.title = '群聊';
    },
    // 切到具体个人
    triggerPersonal(item) {
      if(this.uid === item.uid){
        return;
      }
      // 将当前用户uid，和需要对话的uid放入bridge
      this.bridge = [this.uid, item.uid];
      this.title = '和' + item.nickname + '聊天';
    },
    send(){
      if(!this.msg){
        return
      }
      this.sendMessage(2, this.msg)
    },
    sendMessage(type, msg){
      this.socket.send(JSON.stringify({
        uid: this.uid,
        type: type,
        nickname: this.nickname,
        msg: msg,
        // 增加bridge参数
        bridge: this.bridge
      }));
      this.msg = '';
    },
    conWebSocket(){
      let vm = this;
      if(window.WebSocket){
        vm.socket = new WebSocket('ws://localhost:8001');
        let socket = vm.socket;

        socket.onopen = function(e){
          console.log("连接服务器成功");
          if(!vm.uid){
            ...
          }
          // 这里将sendMessage方法if外面
          vm.sendMessage(1)
        }
        socket.onclose = function(e){
          console.log("服务器关闭");
        }
        socket.onerror = function(){
          console.log("连接出错");
        }
        // 接收服务器的消息
        socket.onmessage = function(e){
          let message = JSON.parse(e.data);
          vm.messageList.push(message);
          if(message.users) {
            vm.users = message.users;
          }
        }   
      }
    },
    login(){
      ...
    }
  }
}