refactor: 项目重构

src/plugins/websocket.ts

@@ -1,8 +1,29 @@
 import { useDictSync } from "@/composables/useDictSync";
 import { Auth } from "@/utils/auth";
+import { useUserStore } from "@/store";
+import { watch } from "vue";
+
+// 全局 WebSocket 实例管理
+const websocketInstances = new Map<string, any>();
 
 // 用于防止重复初始化的状态标记
 let isInitialized = false;
+let dictWebSocketInstance: ReturnType<typeof useDictSync> | null = null;
+
+/**
+ * 注册 WebSocket 实例
+ */
+export function registerWebSocketInstance(key: string, instance: any) {
+  websocketInstances.set(key, instance);
+  console.log(`[WebSocketPlugin] Registered WebSocket instance: ${key}`);
+}
+
+/**
+ * 获取 WebSocket 实例
+ */
+export function getWebSocketInstance(key: string) {
+  return websocketInstances.get(key);
+}
 
 /**
  * 初始化WebSocket服务
@@ -34,19 +55,27 @@ export function setupWebSocket() {
   try {
     // 延迟初始化，确保应用完全启动
     setTimeout(() => {
-      const dictWebSocket = useDictSync();
+      // 保存实例引用
+      dictWebSocketInstance = useDictSync();
+      registerWebSocketInstance("dictSync", dictWebSocketInstance);
 
       // 初始化字典WebSocket服务
-      dictWebSocket.initWebSocket();
+      dictWebSocketInstance.initWebSocket();
       console.log("[WebSocketPlugin] 字典WebSocket初始化完成");
 
-      // 在窗口关闭前断开WebSocket连接
-      window.addEventListener("beforeunload", () => {
-        console.log("[WebSocketPlugin] 窗口即将关闭，断开WebSocket连接");
-        dictWebSocket.closeWebSocket();
-        isInitialized = false;
+      // 初始化在线用户计数WebSocket
+      import("@/composables/useOnlineCount").then(({ useOnlineCount }) => {
+        const onlineCountInstance = useOnlineCount({ autoInit: false });
+        onlineCountInstance.initWebSocket();
+        console.log("[WebSocketPlugin] 在线用户计数WebSocket初始化完成");
       });
 
+      // 在窗口关闭前断开WebSocket连接
+      window.addEventListener("beforeunload", handleWindowClose);
+
+      // 监听用户注销事件
+      watchUserLogout();
+
       console.log("[WebSocketPlugin] WebSocket服务初始化完成");
       isInitialized = true;
     }, 1000); // 延迟1秒初始化
@@ -54,3 +83,84 @@ export function setupWebSocket() {
     console.error("[WebSocketPlugin] 初始化WebSocket服务失败:", error);
   }
 }
+
+/**
+ * 处理窗口关闭
+ */
+function handleWindowClose() {
+  console.log("[WebSocketPlugin] 窗口即将关闭，断开WebSocket连接");
+  cleanupWebSocket();
+}
+
+/**
+ * 监听用户注销
+ */
+function watchUserLogout() {
+  const userStore = useUserStore();
+
+  // 监听用户信息变化，当用户信息被清空时断开连接
+  watch(
+    () => userStore.userInfo,
+    (newUserInfo, oldUserInfo) => {
+      // 从有用户信息变为无用户信息，说明用户注销了
+      if (oldUserInfo?.username && !newUserInfo?.username) {
+        console.log("[WebSocketPlugin] 检测到用户注销，断开WebSocket连接");
+        cleanupWebSocket();
+      }
+    },
+    { deep: true }
+  );
+}
+
+/**
+ * 清理WebSocket连接
+ */
+export function cleanupWebSocket() {
+  // 清理字典 WebSocket
+  if (dictWebSocketInstance) {
+    try {
+      dictWebSocketInstance.closeWebSocket();
+      console.log("[WebSocketPlugin] 字典WebSocket连接已断开");
+    } catch (error) {
+      console.error("[WebSocketPlugin] 断开字典WebSocket连接失败:", error);
+    }
+  }
+
+  // 清理所有注册的 WebSocket 实例
+  websocketInstances.forEach((instance, key) => {
+    try {
+      if (instance && typeof instance.disconnect === "function") {
+        instance.disconnect();
+        console.log(`[WebSocketPlugin] ${key} WebSocket连接已断开`);
+      } else if (instance && typeof instance.closeWebSocket === "function") {
+        instance.closeWebSocket();
+        console.log(`[WebSocketPlugin] ${key} WebSocket连接已断开`);
+      }
+    } catch (error) {
+      console.error(`[WebSocketPlugin] 断开 ${key} WebSocket连接失败:`, error);
+    }
+  });
+
+  // 清空实例映射
+  websocketInstances.clear();
+
+  // 移除事件监听器
+  window.removeEventListener("beforeunload", handleWindowClose);
+
+  // 重置状态
+  dictWebSocketInstance = null;
+  isInitialized = false;
+}
+
+/**
+ * 重新初始化WebSocket（用于登录后重连）
+ */
+export function reinitializeWebSocket() {
+  // 先清理现有连接
+  cleanupWebSocket();
+
+  // 延迟后重新初始化
+  setTimeout(() => {
+    setupWebSocket();
+  }, 500);
+}