前言
本次算是做了一個小小的專題吧，“Nodejs + WebSocket + Vue實現聊天室功能”，目前還在一步一步推進，之前已經可以一對一、一對多聊天了，今天就來創建羣聊組，加入羣聊組等，同時項目中加入了全局message提示框，有興趣的可以去看看。
如果您還沒有看過之前的文字，請點擊下方鏈接查看！
推薦文章：
《WebSocket簡介-第一章》
《Nodejs + WebSocket + Vue 實現多人聊天室- 第二章》
《Nodejs + WebSocket + Vue 一對一、一對多聊天室 - 第三章》
《Node+WebSocket+Vue 一對一、一對多聊天室消息已讀未讀 - 第四章》
WebSocket服務端
做出調整的地方有註釋，後面也會做出講解。
...
let users = [];
let conns = {};
// 羣組數組，多個羣
let groups = [];
function boardcast(obj) {
if(obj.bridge && obj.bridge.length){
obj.bridge.forEach(item=>{
conns[item].sendText(JSON.stringify(obj));
})
return;
}
// 如果是有groupId代表是羣消息
if (obj.groupId) {
// 找到對應羣
group = groups.filter(item=>{
return item.id === obj.groupId
})[0];
// 變量羣裏面的任意，發送消息
group.users.forEach(item=>{
conns[item.uid].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);
conns[''+obj.uid+''] = conn;
// 由原來的if改成switch case
switch(obj.type){
// 創建連接
case 1:
...
break;
// 創建羣
case 10:
// 向groups push數據，同時默認把創建者加入到該羣
groups.push({
id: moment().valueOf(),
name: obj.groupName,
users: [{
uid: obj.uid,
nickname: obj.nickname
}]
})
// 把創建的消息推送給所有用户
boardcast({
type: 1,
date: moment().format('YYYY-MM-DD HH:mm:ss'),
msg: obj.nickname+'創建了羣' + obj.groupName,
users: users,
groups: groups,
uid: obj.uid,
nickname: obj.nickname,
bridge: obj.bridge
});
break;
// 加入羣
case 20:
// 根據傳入的groupId找到對應的羣
let group = groups.filter(item=>{
return item.id === obj.groupId
})[0]
// 向對應的羣成員users push數據
group.users.push({
uid: obj.uid,
nickname: obj.nickname
})
boardcast({
type: 1,
date: moment().format('YYYY-MM-DD HH:mm:ss'),
msg: obj.nickname+'加入了羣' + obj.groupName,
users: users,
groups: groups,
uid: obj.uid,
nickname: obj.nickname,
bridge: obj.bridge
});
break;
// 發送消息
default:
boardcast({
type: 2,
date: moment().format('YYYY-MM-DD HH:mm:ss'),
msg: obj.msg,
uid: obj.uid,
nickname: obj.nickname,
bridge: obj.bridge,
// 添加groupId參數，有是羣發，沒有是一對一
groupId: obj.groupId,
status: 1
});
break;
}
})
...
}).listen(8001)
...ok, 通過上方的代碼，以及註釋，相信很多小夥伴應該都明白了，這裏簡單講解一下。
根據前端頁面傳入的type，來判斷是什麼操作？
1、如果是10，創建羣聊，我們就將羣名稱，以及生成的羣id，存入groups裏面，並且把創建羣聊的人默認加入到羣
2、如果是20，加入羣聊，我們根據要加入的羣id，找到對應的羣，並把需要加入的人，加入到羣
3、發送消息，判斷是否有羣id，如果沒有表示一對一，邏輯不變。如果有羣id，則去groups裏面找到對應的羣，並拿出羣下面所有的user，根據id，找到對應的conn(用户連接)，發送消息。
WebSocket客户端JS
我們也主要研究變的地方，沒有變的通過…表示。同時，如果您想看完整代碼，可以去文章最下方“瞭解更多”，來獲取源碼查看。
...
export default {
...
data(){
return {
title: '請選擇羣或者人員進行聊天',
...
groups: [], // 羣組
groupId: '' // 當前羣聊id
}
},
mounted() {
... // 不變
},
computed: {
currentMessage() {
let vm = this;
let data = vm.messageList.filter(item=>{
if(this.groupId) {
return item.groupId === this.groupId
} else if(item.bridge.length){
return item.bridge.sort().join(',') == vm.bridge.sort().join(',')
}
})
data.map(item=>{
item.status = 0
return item;
})
return data;
}
},
methods: {
addGroup(item){
this.socket.send(JSON.stringify({
uid: this.uid,
type: 20,
nickname: this.nickname,
groupId: item.id,
groupName: item.name,
bridge: []
}));
this.$message({type: 'success', message: `成功加入${item.name}羣`})
},
checkUserIsGroup (item) {
return item.users.some(item=>{
return item.uid === this.uid
})
},
createGroup(){
this.socket.send(JSON.stringify({
uid: this.uid,
type: 10,
nickname: this.nickname,
groupName: this.groupName,
bridge: []
}));
},
getGroupMsgNum(group){
return this.messageList.filter(item=>{
return item.groupId === group.id && item.status === 1
}).length
},
getUserMsgNum(user){
return this.messageList.filter(item=>{
return item.bridge.length && item.uid === user.uid && item.status === 1
}).length
},
triggerGroup(item) {
let issome = item.users.some(item=>{
return item.uid === this.uid
})
if(!issome){
this.$message({type: 'error', message: `您還不是${item.name}羣成員`})
return
}
this.bridge = [];
this.groupId = item.id;
this.title = `和${item.name}羣成員聊天`;
},
triggerPersonal(item) {
if(this.uid === item.uid){
return;
}
this.groupId = '';
this.bridge = [this.uid, item.uid];
this.title = `和${item.nickname}聊天`;
},
send(){
if(!this.msg){
return
}
if(!this.bridge.length && !this.groupId){
this.$message({type: 'error', message: '請選擇發送人或者羣'})
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: this.bridge,
groupId: this.groupId // 如果羣聊id，可能為空（一對一）
}));
this.msg = '';
},
conWebSocket(){
let vm = this;
if(window.WebSocket){
vm.socket = new WebSocket('ws://localhost:8001');
let socket = vm.socket;
...
// 接收服務器的消息
socket.onmessage = function(e){
let message = JSON.parse(e.data);
vm.messageList.push(message);
if(message.users) {
vm.users = message.users;
}
if (message.groups){
vm.groups = message.groups;
}
}
}
},
login(){
...
}
}
}WebSocket客户端HTML
...
<div class="user-list">
<div class="user create-group-btn" @click="$refs.createGroupDialog.show()">新建羣</div>
<div class="user" @click="triggerGroup(item)" v-for="item in groups">
{{item.name}}
<span class="tips-num">{{getGroupMsgNum(item)}}</span>
<span v-if="!checkUserIsGroup(item)" @click.stop="addGroup(item)" class="add-group">+</span>
</div>
<div class="user" @click="triggerPersonal(item)" v-if="item.uid!=uid" v-for="item in users">
{{item.nickname}}
<span class="tips-num">{{getUserMsgNum(item)}}</span>
</div>
</div>
...這裏增加了遍歷groups羣裏，同時判斷當前用户是否在羣裏裏面，沒有則有一個加入按鈕。
解析客户端代碼
1、socket.onmessage來判斷是否有groups羣組，有就賦值給groups
2、創建羣組，輸入名稱，確認後，發送給服務端，告訴是創建羣組，已經創建人員、羣組名稱等
3、加入羣組，發送給服務端要加入羣組id，當前用户id
4、獲取羣未讀消息數量和之前類似，只需要判斷，是羣消息，並且status為1
5、同時頁面校驗等做了一些處理，判斷用户是否在羣裏面，不在不能發送消息；發消息前需選擇用户或者羣；
快速預覽效果
源碼地址：源碼地址
本文章為轉載內容，我們尊重原作者對文章享有的著作權。如有內容錯誤或侵權問題，歡迎原作者聯繫我們進行內容更正或刪除文章。