添加心跳包检测

src/views/chat.vue

@@ -38,6 +38,7 @@ import {NText,NDataTable,NButton,NInput} from 'naive-ui'
 import { useSettingStore } from "@/pinia/store";
 import { generateKey,encryptString } from '@/func/upload';
 import { decryptString } from '@/func/download';
+import WebsocketHeartbeatJs from 'websocket-heartbeat-js';
 
 
 
@@ -61,18 +62,26 @@ const codeInput = ref('')
 const key = ref('')
 const msg = ref('')
 const joinedGroup = []
+const uuid = ref('')
 
 
 const gID2key = {}
+const options = {
+    url: baseUrl,
+    pingTimeout: 20000, 
+    pongTimeout: 10000, 
+    reconnectTimeout: 3000,
+    pingMsg: JSON.stringify({type:'heartbeat'})
+}
+let socket = new WebsocketHeartbeatJs(options);
+
+// const socket = new WebSocket(baseUrl);
+// socket.addEventListener("open", (event) => {
+// //   socket.send("");
+// });
 
 
-const socket = new WebSocket(baseUrl);
-socket.addEventListener("open", (event) => {
-//   socket.send("");
-});
-
-
-socket.addEventListener("message", (event) => {
+socket.onmessage= (event) => {
     const msg = JSON.parse(event.data)
     data.value.push({from:'server',msg:event.data})
     if(msg.type){
@@ -83,8 +92,8 @@ socket.addEventListener("message", (event) => {
                 gID.value=msg.gID
             }else if(type1 === 'joinVerify'){
                 let gID = msg.gID
-                let key = gID2key[gID]
-                console.log(gID2key,gID,key)
+                let key = gID2key[gID].key
+                // console.log(gID2key,gID,key)
                 let res = decryptString(key,msg.verify)
                 console.log(res,key,msg.verify)
                 if(res === 'miaoSpring'){
@@ -93,18 +102,40 @@ socket.addEventListener("message", (event) => {
                 }
             }else if(type1 === 'history'){
                 let gID = msg.gID
-                let key = gID2key[gID]
+                let key = gID2key[gID].key
                 for(let m of JSON.parse(msg.history)){
                     data.value.push({from:'decrypt',msg:decryptString(key,m.msg)})
                 }
+            }else if(type1 ===  'uuid'){
+                uuid.value = msg.uuid
             }
         }else if(type === 'gMsg'){
             let gID = msg.gID
-            let key = gID2key[gID]
+            let key = gID2key[gID].key
             data.value.push({from:'decrypt',msg:decryptString(key,msg.msg)})
         }
     }
-});
+};
+
+socket.onopen = ()=>{
+    if(uuid.value){
+        socket.send(JSON.stringify({type:'uuid',type1:'set',uuid:uuid.value}))
+    }else{
+        socket.send(JSON.stringify({type:'uuid',type1:'get'}))
+    }
+    console.log(joinedGroup)
+    let toJoin = [...joinedGroup]
+    joinedGroup.length = 0
+    for(let gID of toJoin){
+        let code = gID+gID2key[gID].key
+        console.log(code)
+        joinGroup(code)
+    }
+}
+
+socket.onreconnect = (e) => {
+    console.log('reconnecting...');
+}
 
 function createGroup(){
     let keyt = generateKey(setting)
@@ -118,7 +149,8 @@ function joinGroup(code){
     if(joinedGroup.includes(gID)){
         return
     }
-    gID2key[gID] = key
+    gID2key[gID] = {}
+    gID2key[gID].key = key
     joinGroup_Verify(gID)
 }
 
@@ -127,7 +159,7 @@ function joinGroup_Verify(gID){
 }
 
 function sendMsg(gID,msg){
-    let key = gID2key[gID]
+    let key = gID2key[gID].key
     msg = encryptString(key,msg,setting)
     socket.send(JSON.stringify({type:'message',msg,gID}))
 }