feat: add `websocket.ts`

package.json

@@ -46,6 +46,10 @@
       "types": "./dist/plugs/http/index.d.ts",
       "default": "./dist/plugs/http/index.js"
     },
+    "./ws": {
+      "types": "./dist/plugs/websocket.d.ts",
+      "default": "./dist/plugs/websocket.js"
+    },
     "./i18n": {
       "types": "./dist/plugs/i18n/index.d.ts",
       "default": "./dist/plugs/i18n/index.js"

plugs/index.ts

@@ -4,6 +4,7 @@ export * as Store from "./store";
 export * as Http from "./http";
 export * as Lang from "./i18n";
 export * as Api from "./api";
+export * as WebSocket from "./websocket";
 export * from "./constant";
 export * from "./util/asyncUtil";
 export * from "./util/objectUtil";

plugs/websocket.ts

@@ -0,0 +1,52 @@
+let websocket: WebSocket;
+const msgHandlersMap = {};
+let lastUrl;
+let closeFlag = false;
+
+export const closeWebSocket = () => {
+  closeFlag = false;
+  websocket?.close();
+};
+
+export const sendSocketMsg = (msg) => {
+  websocket?.send(JSON.stringify(msg));
+};
+
+export const openWebSocket = (url) => {
+  if (websocket && websocket.readyState === WebSocket.OPEN) {
+    if (lastUrl == url) {
+      return;
+    }
+    closeWebSocket();
+  }
+  websocket = new WebSocket(url);
+  lastUrl = url;
+  websocket.onopen = () => {
+    console.log("websocket已连接");
+    closeFlag = true;
+  };
+  websocket.onmessage = (msg) => {
+    const event = JSON.parse(msg.data);
+    const handlers = msgHandlersMap[event.type];
+    if (!handlers) {
+      return;
+    }
+    Object.values(handlers).forEach((handler: any) => {
+      handler(event.data);
+    });
+  };
+  websocket.onclose = () => {
+    console.log("websocket已断开");
+    if (closeFlag) {
+      setTimeout(() => {
+        openWebSocket(url);
+      }, 2000);
+    }
+  };
+};
+
+export const registerHandler = (type, key, handler) => {
+  const handlers = msgHandlersMap[type] || {};
+  handlers[key] = handler;
+  msgHandlersMap[type] = handlers;
+};

tsconfig.json

@@ -42,6 +42,9 @@
       "noob-mengyxu/http": [
         "./plugs/http/index.ts"
       ],
+      "noob-mengyxu/ws": [
+        "./plugs/websockets.ts"
+      ],
       "noob-mengyxu/i18n": [
         "./plugs/i18n/index.ts"
       ],