update middleware/socket.ts

src/middleware/socket.ts

@@ -26,6 +26,8 @@ export interface socketAction {
     player: string;
     passWd: string;
   };
+  isReplay?: boolean; // 是否是回放模式
+  replayUrl?: string; // 提供回放服务的地址，当`isReplay`为true时，必传
   // 通过长连接发送的数据
   payload?: Uint8Array;
 }
@@ -36,12 +38,18 @@ let ws: WebSocketStream | null = null;
 export default async function (action: socketAction) {
   switch (action.cmd) {
     case socketCmd.CONNECT: {
-      const info = action.initInfo;
+      const { initInfo: info, isReplay, replayUrl } = action;
       if (info) {
         ws = new WebSocketStream(info.ip, (conn, _event) =>
           handleSocketOpen(conn, info.ip, info.player, info.passWd)
         );
 
+        await ws.execute(handleSocketMessage);
+      } else if (isReplay && replayUrl) {
+        ws = new WebSocketStream(replayUrl, () =>
+          console.info("replay websocket open.")
+        );
+
         await ws.execute(handleSocketMessage);
       }