如果你看到这篇文章，还没有了解前面2篇文章的同学，可以先去了解一波，这样上手更快。
推荐文章：
《Nodejs + WebSocket简单介绍及示例 - 第一章》
《Nodejs + WebSocket + Vue 实现多人聊天室WebIM功能 - 第二章》
这篇文章都是前面文章的加强版，功能升级。给大家提供一个循序渐进的学习过程，一步一步的来。
在第二篇文章结束时，我们就已经可以一对多的聊天了，就是多人群聊。这次，我们进行扩展来实现一对一、一对多功能。
WebSocket客户端UI界面更改有了一对一，一对多，我们就需要对直接的界面做出调整了。左边显示聊天人员列表，右边是具体消息列表。
群1
{{item.nickname}}
{{title}}
{{item.msg}}
{{item.nickname}}
...
这里我们就写死了一个群，叫群1，默认是所用用户进去群聊。
WebSocket服务端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建立完毕")
主体结构还是和第二章类型，不同的是：
1、每次将uid对应的conn存储到一个对象conns上
2、根据客户端传入的参数bridge来判断，是群发还是一对一发送
3、群发还是第二章逻辑即可
server.connections.forEach((conn, index) => {
conn.sendText(JSON.stringify(obj));
})
4、一对一发送，bridge里面是一对一的两个用户uid，这样就可以在conns对象上找到uid对应的连接conn，并用conn发送信息即可
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(){
...
}
}
}
上方…的代码区域都是和第二篇文章一样的地方，所有就省略了。
1、默认是群发，即bridge是空数组，向所有用户发送消息
2、点击用户列表，赋予bridge当前用户uid，和需要对话的uid。
3、在第二篇文章中，渲染的消息列表是messageList。现在不是，是通过计算属性computed，只需要bridge相等的消息，得出currentMessage当前对话的消息列表
4、因为所有消息都是通过后台socket返回，也不需要考虑发送者/接收者是谁，判断bridge是否相等，可以用sort()方法排序并转换成字符串后进行对比。
一对一和一对多的核心，就是知道是那个用户与那个用户对话。当前目前一对多是不用考虑，因为是写死的，所有用户，后面来做不同群，随意加群聊天，就需要考虑了。也就是服务端的conn不要弄错，不然收不到消息也接收不到消息。
源码地址：源码地址
