@ray-js/robot-data-stream 0.0.13-beta-5 → 0.0.13-beta-7

package/lib/api/sweeperP2p.js

@@ -1,49 +1,32 @@
 import "core-js/modules/esnext.iterator.constructor.js";
+import "core-js/modules/esnext.iterator.filter.js";
 import "core-js/modules/esnext.iterator.for-each.js";
 import "core-js/modules/esnext.iterator.map.js";
-/* eslint-disable no-shadow */
 import Base64 from 'base64-js';
-import { join, map, padStart, once } from 'lodash-es';
+import { join, map, once, padStart } from 'lodash-es';
+import { FILE_NAME_MAP, FileNameEnum } from '../constant';
 import { trace } from '../trace';
-import { logger } from '../utils';
+import { checkIfDirIsExist, createFilePath } from '../ttt';
 import P2pApi from './p2pApi';
 /**
  * 基于P2p工具类的扫地机扩展实现
  */
 
-const FILE_NAME_MAP = {
-  // raw类型地图点数据
-  'map.bin': {
-    type: 0
-  },
-  // 结构化点数据
-  'map_structured.bin': {
-    type: 6
-  },
-  'cleanPath.bin': {
-    type: 1
-  },
-  'map.bin.stream': {
-    type: 0
-  },
-  'map_structured.bin.stream': {
-    type: 6
-  },
-  'cleanPath.bin.stream': {
-    type: 1
-  },
-  'ai.bin': {
-    type: 4
-  },
-  'ai.bin.stream': {
-    type: 4
-  },
-  'aiHD_XXXX_YYYY.bin': {
-    type: 5
-  },
-  'aiHD_XXXX_YYYY.bin.stream': {
-    type: 5
-  }
+// 需要初始化的文件名列表（不包括 wifi_map）
+const FILE_NAMES_TO_INIT = [FileNameEnum.map, FileNameEnum.mapStructured, FileNameEnum.cleanPath, FileNameEnum.ai, FileNameEnum.aiHD, FileNameEnum.mapStream, FileNameEnum.mapStructuredStream, FileNameEnum.cleanPathStream, FileNameEnum.aiStream, FileNameEnum.aiHDStream];
+
+/**
+ * 创建初始化的 Map，所有键值都设置为初始值
+ */
+const createInitializedMap = initialValue => {
+  return new Map(FILE_NAMES_TO_INIT.map(key => [key, initialValue]));
+};
+
+/**
+ * 创建初始化的 Map，值为新的 Map 实例
+ */
+const createMapOfMaps = () => {
+  return new Map(FILE_NAMES_TO_INIT.map(key => [key, new Map()]));
 };
 
 // 走p2p流传输(新) or 读取bin文件(旧)
@@ -61,10 +44,11 @@ export class SweeperP2p extends P2pApi {
     this.cacheData = {};
     this.exitFiles = [];
     this.firstPackageTime = Date.now();
-    this.packetTotalMap = new Map([['map.bin', -1], ['map_structured.bin', -1], ['cleanPath.bin', -1], ['ai.bin', -1], ['aiHD_XXXX_YYYY.bin', -1], ['map.bin.stream', -1], ['map_structured.bin.stream', -1], ['cleanPath.bin.stream', -1], ['ai.bin.stream', -1], ['aiHD_XXXX_YYYY.bin.stream', -1]]);
-    this.packetSerialNumberCacheMap = new Map([['map.bin', -1], ['map_structured.bin', -1], ['cleanPath.bin', -1], ['ai.bin', -1], ['aiHD_XXXX_YYYY.bin', -1], ['map.bin.stream', -1], ['map_structured.bin.stream', -1], ['cleanPath.bin.stream', -1], ['ai.bin.stream', -1], ['aiHD_XXXX_YYYY.bin.stream', -1]]);
-    this.fileLengthCacheMap = new Map([['map.bin', -1], ['map_structured.bin', -1], ['cleanPath.bin', -1], ['ai.bin', -1], ['aiHD_XXXX_YYYY.bin', -1], ['map.bin.stream', -1], ['map_structured.bin.stream', -1], ['cleanPath.bin.stream', -1], ['ai.bin.stream', -1], ['aiHD_XXXX_YYYY.bin.stream', -1]]);
-    this.packetDataCacheMap = new Map([['map.bin', new Map()], ['map_structured.bin', new Map()], ['cleanPath.bin', new Map()], ['ai.bin', new Map()], ['aiHD_XXXX_YYYY.bin', new Map()], ['map.bin.stream', new Map()], ['map_structured.bin.stream', new Map()], ['cleanPath.bin.stream', new Map()], ['ai.bin.stream', new Map()], ['aiHD_XXXX_YYYY.bin.stream', new Map()]]);
+    // 使用辅助函数初始化 Map，减少重复代码
+    this.packetTotalMap = createInitializedMap(-1);
+    this.packetSerialNumberCacheMap = createInitializedMap(-1);
+    this.fileLengthCacheMap = createInitializedMap(-1);
+    this.packetDataCacheMap = createMapOfMaps();
   }
   setStreamFilePath = () => {
     // this.streamFilePath = this.cacheDir + `/${this.albumName}/${devId}/stream`;
@@ -81,7 +65,7 @@ export class SweeperP2p extends P2pApi {
     if (/usr/.test(this.cacheDir)) {
       this.dataFilePath = this.cacheDir;
     } else {
-      this.streamFilePath = this.cacheDir + 'usr';
+      this.dataFilePath = this.cacheDir + 'usr';
     }
     // 检查存储文件目录是否存在
     // this.initFilePath(this.dataFilePath);
@@ -105,70 +89,17 @@ export class SweeperP2p extends P2pApi {
     return this.dataFilePath + '/' + fileName;
   };
 
-  /**
-   * 创建文件路径文件夹
-   * @param filePath
-   * @returns
-   */
-  createFilePath = filePath => {
-    try {
-      ty.getFileSystemManager().mkdirSync({
-        dirPath: filePath,
-        recursive: true
-      });
-      logger('info', {
-        msg: 'mkdirSync success: filePath ==>',
-        filePath
-      }, this.onLogger);
-      return true;
-    } catch (e) {
-      logger('error', {
-        msg: 'mkdirSync error ==>',
-        e
-      }, this.onLogger);
-      return false;
-    }
-  };
-
-  /**
-   * 检查当前文件目录是否存在
-   * @param filePath
-   * @returns
-   */
-  checkIfDirIsExist = filePath => {
-    return new Promise(resolve => {
-      ty.getFileSystemManager().access({
-        path: filePath,
-        success(params) {
-          logger('info', {
-            msg: 'file access success ==>',
-            params
-          }, this.onLogger);
-          resolve(true);
-        },
-        fail(params) {
-          logger('warn', {
-            msg: 'file access fail ==>',
-            params
-          }, this.onLogger);
-          resolve(false);
-        }
-      });
-    });
-  };
-
   /**
    * 初始化文件目录
    * @param filePath
    * @returns
    */
   initFilePath = async filePath => {
-    const isDir = await this.checkIfDirIsExist(filePath);
+    const isDir = await checkIfDirIsExist(filePath);
     if (!isDir) {
-      const result = this.createFilePath(filePath);
-      return result;
+      return await createFilePath(filePath, this.onLogger);
     }
-    return isDir;
+    return true;
   };
 
   /**
@@ -176,72 +107,186 @@ export class SweeperP2p extends P2pApi {
    * @param filename
    */
   getFileType = filename => {
-    if (filename.indexOf('map_structured') !== -1) {
-      return 6;
-    }
-    if (filename.indexOf('map.') !== -1) {
-      return 0;
+    // 首先尝试精确匹配 FILE_NAME_MAP 中的键
+    if (filename in FILE_NAME_MAP) {
+      return FILE_NAME_MAP[filename].type;
     }
-    if (filename.indexOf('cleanPath') !== -1) {
-      return 1;
-    }
-    if (filename.indexOf('ai.') !== -1) {
-      return 4;
+
+    // 处理 aiHD_XXXX_YYYY.bin 模式（XXXX_YYYY 会被替换为实际值）
+
+    if (/^aiHD.*\.bin(\.stream)?$/.test(filename)) {
+      return FILE_NAME_MAP[FileNameEnum.aiHD].type;
     }
-    if (filename.indexOf('aiHD') !== -1) {
-      return 5;
+
+    // 按优先级顺序匹配文件名（更具体的模式优先）
+    // 注意：map_structured 必须在 map 之前匹配
+    const matchOrder = [{
+      pattern: /map_structured/,
+      enumKey: FileNameEnum.mapStructured
+    }, {
+      pattern: /^map\./,
+      enumKey: FileNameEnum.map
+    }, {
+      pattern: /cleanPath/,
+      enumKey: FileNameEnum.cleanPath
+    }, {
+      pattern: /^ai\./,
+      enumKey: FileNameEnum.ai
+    }, {
+      pattern: /wifi_map/,
+      enumKey: FileNameEnum.wifiMap
+    }];
+    for (const {
+      pattern,
+      enumKey
+    } of matchOrder) {
+      if (pattern.test(filename)) {
+        return FILE_NAME_MAP[enumKey].type;
+      }
     }
+
+    // 默认返回 2
     return 2;
   };
 
+  /**
+   * 根据文件类型处理数据并调用相应的回调函数
+   */
+  handleDataByType = (() => {
+    var _this = this;
+    return function (type, hexValue) {
+      let shouldCache = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
+      if (_this.cacheData[type] === hexValue) {
+        return;
+      }
+      switch (type) {
+        case 0:
+        case 6:
+          _this.onReceiveMapData(hexValue);
+          if (shouldCache) {
+            _this.cacheData[type] = hexValue;
+          }
+          break;
+        case 1:
+          _this.log.info({
+            msg: 'push new path data'
+          });
+          _this.onReceivePathData(hexValue);
+          if (shouldCache) {
+            _this.cacheData[type] = hexValue;
+          }
+          break;
+        case 4:
+          _this.onReceiveAIPicData(hexValue);
+          break;
+        case 5:
+          _this.onReceiveAIPicHDData(hexValue);
+          break;
+        case 7:
+          _this.onReceiveWifiMapData(hexValue);
+          break;
+        default:
+          break;
+      }
+    };
+  })();
+
   /**
    * 设备连接状态发生改变
    * @param data
    */
   sessionStatusCallback = data => {
-    logger('info', {
+    this.log.info({
       msg: 'sessionStatusCallback ==>',
       data
-    }, this.onLogger);
+    });
     if (data) {
       const {
         status
       } = data;
       if (status < 0) {
-        logger('info', {
+        this.log.info({
           msg: 'receive disconnect notice ==>',
           status
-        }, this.onLogger);
+        });
         this.isConnected = false;
         if (!this.isConnecting) {
           this.reconnectP2p(() => {
             // 重连之后重新开启文件下载, 这里的成功不需要注册断开事件
-            this.startObserverSweeperDataByP2P(this.downloadType, this.devId, this.onReceiveMapData, this.onReceivePathData, this.onReceiveAIPicData, this.onReceiveAIPicHDData);
+            this.startObserverSweeperDataByP2P(this.downloadType, this.devId, this.onReceiveMapData, this.onReceivePathData, this.onReceiveAIPicData, this.onReceiveAIPicHDData, this.onReceiveWifiMapData);
           }, '');
         } else {
-          logger('warn', {
+          this.log.warn({
             msg: 'receive disconnect notice, but connectDevice is connecting'
-          }, this.onLogger);
+          });
         }
       }
     }
   };
 
+  /**
+   * 创建下载失败时的重连回调函数
+   */
+  createReconnectCallback = () => {
+    return () => {
+      this.log.error({
+        msg: `${this.downloadType} p2p downloadStream failed and try reConnect Device===>`
+      });
+      this.removeP2pDownloadEvent();
+      this.disconnectDevice().then(() => {
+        this.isConnected = false;
+        this.isConnecting = false;
+        this.connectDevice(() => {
+          // 重连之后重新开启文件下载, 这里的成功不需要注册断开事件
+          this.startObserverSweeperDataByP2P(this.downloadType, this.devId, this.onReceiveMapData, this.onReceivePathData, this.onReceiveAIPicData, this.onReceiveAIPicHDData, this.onReceiveWifiMapData);
+        });
+      });
+    };
+  };
+
+  /**
+   * 执行文件下载逻辑
+   */
+  async executeDownload(exitFiles, filePath) {
+    if (exitFiles.length === 0) {
+      return;
+    }
+    if (shouldDownloadStream) {
+      // 开启p2p流传输
+      await this.downloadStream({
+        files: exitFiles
+      }, this.albumName, () => {
+        this.log.info({
+          msg: `${this.downloadType} p2p downloadStream success===>`
+        });
+      }, this.createReconnectCallback());
+    } else if (await this.initFilePath(filePath)) {
+      // 每次要下载前都需要先检查文件目录是否存在 防止中间过程被删除掉文件目录
+      await this.downloadFile({
+        files: exitFiles
+      }, this.albumName, filePath);
+    }
+    this.firstPackageTime = Date.now();
+  }
+
   /**
    * 开始进行文件下载
    * @param downloadType
    * 0: 下载断开 1: 持续下载
    */
-  startObserverSweeperDataByP2P = async (downloadType, devId, onReceiveMapData, onReceivePathData, onReceiveAIPicData, onReceiveAIPicHDData, onDefineStructuredMode) => {
-    var _this$file$items;
-    if (![0, 1].some(item => item === downloadType)) {
-      logger('warn', {
+  startObserverSweeperDataByP2P = async (downloadType, devId, onReceiveMapData, onReceivePathData, onReceiveAIPicData, onReceiveAIPicHDData, onReceiveWifiMapData, onDefineStructuredMode) => {
+    var _this$file;
+    if (![0, 1].includes(downloadType)) {
+      this.log.warn({
         msg: 'download type must be 0 or 1'
-      }, this.onLogger);
+      });
       return;
     }
+
+    // 初始化回调函数
     this.onReceiveMapData = onReceiveMapData;
     this.onReceivePathData = onReceivePathData;
+    this.onReceiveWifiMapData = onReceiveWifiMapData;
     this.onDefineStructuredMode = onDefineStructuredMode;
     this.onReceiveAIPicData = onReceiveAIPicData || (() => {
       // do nothing
@@ -249,9 +294,9 @@ export class SweeperP2p extends P2pApi {
     this.onReceiveAIPicHDData = onReceiveAIPicHDData || (() => {
       // do nothing
     });
-    logger('info', {
+    this.log.info({
       msg: 'startObserverSweeperDataByP2P ==>'
-    }, this.onLogger);
+    });
     trace.pointFn({
       devId,
       eventName: 'startObserverSweeperDataByP2P'
@@ -259,85 +304,27 @@ export class SweeperP2p extends P2pApi {
     this.downloadType = downloadType;
     this.setDataFilePath(devId);
     this.setStreamFilePath(devId);
+
     // 先移除监听,再重新注册
     this.removeP2pDownloadEvent();
     this.registerP2pDownloadEvent();
+
+    // 查询文件列表
     if (!this.file) {
       // @ts-ignore
       this.file = await this.queryAlbumFileIndexs(this.albumName);
     }
-    if (this.file && ((_this$file$items = this.file.items) === null || _this$file$items === void 0 ? void 0 : _this$file$items.length) > 0) {
-      if (this.downloadType === 0) {
-        const exitFiles = this.queryNeedFiles(this.file.items);
-        // 赋值
-        this.exitFiles = exitFiles;
-        if (exitFiles.length > 0) {
-          if (shouldDownloadStream) {
-            // 开启p2p流传输
-            await this.downloadStream({
-              files: exitFiles
-            }, this.albumName, () => {
-              logger('info', {
-                msg: `${downloadType} p2p downloadStream success===>`
-              }, this.onLogger);
-            }, () => {
-              logger('error', {
-                msg: `${downloadType} p2p downloadStream failed and try reConnect Device===>`
-              }, this.onLogger);
-              this.removeP2pDownloadEvent();
-              this.disconnectDevice().then(() => {
-                this.isConnected = false;
-                this.isConnecting = false;
-                this.connectDevice(() => {
-                  // 重连之后重新开启文件下载, 这里的成功不需要注册断开事件
-                  this.startObserverSweeperDataByP2P(this.downloadType, this.devId, this.onReceiveMapData, this.onReceivePathData, this.onReceiveAIPicData, this.onReceiveAIPicHDData);
-                });
-              });
-            });
-          } else if (await this.initFilePath(this.dataFilePath)) {
-            await this.downloadFile({
-              files: exitFiles
-            }, this.albumName, this.dataFilePath);
-          }
-          this.firstPackageTime = Date.now();
-        }
-      } else if (this.downloadType === 1) {
-        const exitFiles = this.queryNeedFiles(this.file.items);
-        // 赋值
-        this.exitFiles = exitFiles;
-        if (exitFiles.length > 0) {
-          if (shouldDownloadStream) {
-            // 开启p2p流传输
-            await this.downloadStream({
-              files: exitFiles
-            }, this.albumName, () => {
-              logger('info', {
-                msg: `${downloadType} p2p downloadStream success===>`
-              }, this.onLogger);
-            }, () => {
-              logger('error', {
-                msg: `${downloadType} p2p downloadStream failed and try reConnect Device===>`
-              }, this.onLogger);
-              this.removeP2pDownloadEvent();
-              this.disconnectDevice().then(() => {
-                this.isConnected = false;
-                this.isConnecting = false;
-                this.connectDevice(() => {
-                  // 重连之后重新开启文件下载, 这里的成功不需要注册断开事件
-                  this.startObserverSweeperDataByP2P(this.downloadType, this.devId, this.onReceiveMapData, this.onReceivePathData, this.onReceiveAIPicData, this.onReceiveAIPicHDData);
-                });
-              });
-            });
-          } else if (await this.initFilePath(this.streamFilePath)) {
-            // 每次要下载前都需要先检查文件目录是否存在 防止中间过程被删除掉文件目录
-            await this.downloadFile({
-              files: exitFiles
-            }, this.albumName, this.streamFilePath);
-          }
-          this.firstPackageTime = Date.now();
-        }
-      }
+    if (!((_this$file = this.file) !== null && _this$file !== void 0 && (_this$file = _this$file.items) !== null && _this$file !== void 0 && _this$file.length)) {
+      return;
     }
+
+    // 根据 downloadType 选择文件路径
+    const filePath = downloadType === 0 ? this.dataFilePath : this.streamFilePath;
+    const exitFiles = this.queryNeedFiles(this.file.items);
+    this.exitFiles = exitFiles;
+
+    // 执行下载
+    await this.executeDownload(exitFiles, filePath);
   };
 
   /**
@@ -357,9 +344,9 @@ export class SweeperP2p extends P2pApi {
    * 注册下载有关的监听
    */
   registerP2pDownloadEvent = () => {
-    logger('info', {
+    this.log.info({
       msg: 'registerP2pDownloadEvent ==>'
-    }, this.onLogger);
+    });
     if (shouldDownloadStream) {
       // p2p数据流监听
       this.offP2pStreamPacketReceive = this.onP2pStreamPacketReceive(this.p2pStreamPacketReceiveCallback);
@@ -380,29 +367,25 @@ export class SweeperP2p extends P2pApi {
    * 移除下载有关的监听
    */
   removeP2pDownloadEvent = () => {
-    logger('info', {
+    var _this$offFileDownload, _this$offP2pStreamPac;
+    this.log.info({
       msg: 'removeP2pDownloadEvent ==>'
-    }, this.onLogger);
+    });
     if (shouldDownloadStream) {
-      [...this.packetSerialNumberCacheMap.keys()].forEach(key => {
+      // 重置所有 Map 的值为初始状态
+      FILE_NAMES_TO_INIT.forEach(key => {
         this.packetSerialNumberCacheMap.set(key, -1);
-      });
-      [...this.fileLengthCacheMap.keys()].forEach(key => {
         this.fileLengthCacheMap.set(key, -1);
-      });
-      [...this.packetTotalMap.keys()].forEach(key => {
         this.packetTotalMap.set(key, -1);
-      });
-      [...this.packetDataCacheMap.keys()].forEach(key => {
         this.packetDataCacheMap.set(key, new Map());
       });
     }
     this.cacheData = {};
-    this.offFileDownloadComplete && this.offFileDownloadComplete();
-    this.offP2pStreamPacketReceive && this.offP2pStreamPacketReceive();
+    (_this$offFileDownload = this.offFileDownloadComplete) === null || _this$offFileDownload === void 0 || _this$offFileDownload.call(this);
+    (_this$offP2pStreamPac = this.offP2pStreamPacketReceive) === null || _this$offP2pStreamPac === void 0 || _this$offP2pStreamPac.call(this);
 
-    // this.offDownLoadProgressUpdate && this.offDownLoadProgressUpdate();
-    // this.offTotalDownLoadProgressUpdate && this.offTotalDownLoadProgressUpdate();
+    // this.offDownLoadProgressUpdate?.();
+    // this.offTotalDownLoadProgressUpdate?.();
   };
 
   /**
@@ -451,15 +434,15 @@ export class SweeperP2p extends P2pApi {
     // 因为XXXX_YYYYY会被替换为坐标值，所以这里需要替换一下
 
     if (/^aiHD.*\.bin$/.test(fileName)) {
-      fileName = 'aiHD_XXXX_YYYY.bin';
+      fileName = FileNameEnum.aiHD;
     } else if (/^aiHD.*\.bin.stream$/.test(fileName)) {
-      fileName = 'aiHD_XXXX_YYYY.bin.stream';
+      fileName = FileNameEnum.aiHDStream;
     }
     const cachePacketMap = this.packetDataCacheMap.get(fileName);
     if (!cachePacketMap) {
-      logger('warn', {
+      this.log.warn({
         msg: `p2pStreamPacketReceiveCallback: fileName : ${fileName} is not support`
-      }, this.onLogger);
+      });
       return;
     }
     const cacheSerialNumber = this.packetSerialNumberCacheMap.get(fileName);
@@ -493,16 +476,16 @@ export class SweeperP2p extends P2pApi {
         }).join('');
         const cacheFileLength = this.fileLengthCacheMap.get(fileName);
         if (cacheFileLength > 0 && hexValue.length !== cacheFileLength * 2) {
-          logger('warn', {
+          this.log.warn({
             msg: `p2pStreamPacketReceiveCallback: fileName : ${fileName} length mismatch, expected: ${cacheFileLength * 2}, received: ${hexValue.length}`
-          }, this.onLogger);
+          });
           return;
         }
         if (this.firstPackageTime > 0 && fileName.indexOf('map') !== -1) {
           const packageTime = Date.now() - this.firstPackageTime;
-          logger('info', {
+          this.log.info({
             msg: `receive first full package, cost time: ${packageTime} ms, fileName: ${fileName}`
-          }, this.onLogger);
+          });
           trace.pointFn({
             devId: this.devId,
             eventName: 'receiveFirstFullPackage'
@@ -512,25 +495,7 @@ export class SweeperP2p extends P2pApi {
         const {
           type
         } = FILE_NAME_MAP[fileName];
-        if (this.cacheData[type] !== hexValue) {
-          if (type === 0 || type === 6) {
-            this.onReceiveMapData(hexValue);
-            this.cacheData[type] = hexValue;
-          }
-          if (type === 1) {
-            logger('info', {
-              msg: `push new path data`
-            }, this.onLogger);
-            this.onReceivePathData(hexValue);
-            this.cacheData[type] = hexValue;
-          }
-          if (type === 4) {
-            this.onReceiveAIPicData(hexValue);
-          }
-          if (type === 5) {
-            this.onReceiveAIPicHDData(hexValue);
-          }
-        }
+        this.handleDataByType(type, hexValue);
       }
     }
   };
@@ -552,83 +517,65 @@ export class SweeperP2p extends P2pApi {
           const bytes = Base64.toByteArray(params.data);
           const hexValue = join(map(bytes, d => padStart(d.toString(16), 2, '0')), '');
           const type = this.getFileType(fileName);
-          if (this.cacheData[type] !== hexValue) {
-            if (hexValue.length === 0) {
-              logger('warn', {
-                msg: 'receive empty data'
-              }, this.onLogger);
-              return;
-            }
-            if (type === 0 || type === 6) {
-              this.onReceiveMapData(hexValue);
-            }
-            if (type === 1) {
-              this.onReceivePathData(hexValue);
-            }
-            if (type === 4) {
-              this.onReceiveAIPicData(hexValue);
-            }
-            if (type === 5) {
-              this.onReceiveAIPicHDData(hexValue);
-            }
-            this.cacheData[type] = hexValue;
+          if (hexValue.length === 0) {
+            this.log.warn({
+              msg: 'receive empty data'
+            });
+            return;
           }
+          this.handleDataByType(type, hexValue);
         },
         fail: e => {
-          logger('warn', {
+          this.log.warn({
             msg: 'readFileFromPath failed ==>',
             e
-          }, this.onLogger);
+          });
           this.resetReading(fileName);
         }
       });
     } catch (e) {
       this.resetReading(fileName);
-      logger('error', {
+      this.log.error({
         msg: 'readFileFromPath ==>',
         e
-      }, this.onLogger);
+      });
     }
   };
-  setReading = fileName => {
-    if (fileName.indexOf('map_structured') !== -1) {
-      if (this.readingMap === true) return false;
-      this.readingMap = true;
-      return true;
+
+  /**
+   * 根据文件名获取对应的 reading 状态标志
+   */
+  getReadingFlag = fileName => {
+    if (fileName.includes('map_structured') || fileName.includes('map.')) {
+      return 'readingMap';
     }
-    if (fileName.indexOf('map.') !== -1) {
-      if (this.readingMap === true) return false;
-      this.readingMap = true;
-      return true;
+    if (fileName.includes('cleanPath')) {
+      return 'readingClean';
     }
-    if (fileName.indexOf('cleanPath') !== -1) {
-      if (this.readingClean === true) return false;
-      this.readingClean = true;
-      return true;
+    if (fileName.includes('ai.') && !fileName.includes('aiHD')) {
+      return 'readingAI';
     }
-    if (fileName.indexOf('ai.') !== -1) {
-      if (this.readingAI === true) return false;
-      this.readingAI = true;
-      return true;
+    if (fileName.includes('aiHD')) {
+      return 'readingHD';
     }
-    if (fileName.indexOf('aiHD') !== -1) {
-      if (this.readingHD === true) return false;
-      this.readingHD = true;
+    return null;
+  };
+  setReading = fileName => {
+    const flag = this.getReadingFlag(fileName);
+    if (!flag) {
       return true;
     }
+    const readingState = this[flag];
+    if (readingState) {
+      return false;
+    }
+    this[flag] = true;
     return true;
   };
   resetReading = fileName => {
-    if (fileName.indexOf('map_structured') !== -1) {
-      this.readingMap = false;
-    } else if (fileName.indexOf('map.') !== -1) {
-      this.readingMap = false;
-    } else if (fileName.indexOf('cleanPath') !== -1) {
-      this.readingClean = false;
-    } else if (fileName.indexOf('ai.') !== -1) {
-      this.readingAI = false;
-    } else if (fileName.indexOf('aiHD') !== -1) {
-      this.readingHD = false;
+    const flag = this.getReadingFlag(fileName);
+    if (flag) {
+      this[flag] = false;
     }
   };
 
@@ -637,21 +584,11 @@ export class SweeperP2p extends P2pApi {
    * @param fileList
    */
   queryNeedFiles = fileList => {
-    const exitFiles = [];
-    const streamPattern = /bin.stream$/;
-    const dataPattern = /bin$/;
-    fileList && fileList.forEach(item => {
-      if (this.downloadType === 0) {
-        if (dataPattern.test(item.filename)) {
-          exitFiles.push(item.filename);
-        }
-      } else if (this.downloadType === 1) {
-        if (streamPattern.test(item.filename)) {
-          exitFiles.push(item.filename);
-        }
-      }
-    });
-    return exitFiles;
+    if (!(fileList !== null && fileList !== void 0 && fileList.length)) {
+      return [];
+    }
+    const pattern = this.downloadType === 0 ? /\.bin$/ : /\.bin\.stream$/;
+    return fileList.filter(item => pattern.test(item.filename)).map(item => item.filename);
   };
 
   /**
@@ -660,13 +597,13 @@ export class SweeperP2p extends P2pApi {
   // eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types, consistent-return
   stopObserverSweeperDataByP2P = () => {
     try {
-      logger('info', {
+      this.log.info({
         msg: `isConnected: ${this.isConnected}, devId: ${this.devId}`
-      }, this.onLogger);
+      });
       if (this.isConnected) {
-        logger('info', {
+        this.log.info({
           msg: `try disconnect device, devId: ${this.devId}`
-        }, this.onLogger);
+        });
         this.disconnectDevice();
       }
       this.removeP2pDownloadEvent();
@@ -677,10 +614,10 @@ export class SweeperP2p extends P2pApi {
       typeof this.offSessionStatusChange === 'function' && this.offSessionStatusChange();
       // 销毁初始化时,先进行断连操作
     } catch (e) {
-      logger('error', {
+      this.log.error({
         msg: 'stopObserverSweeperDataByP2P occur error ==>',
         e
-      }, this.onLogger);
+      });
       return false;
     }
   };