koishi-plugin-chat-analyse 1.6.6 → 1.6.8

package/lib/index.js

@@ -2132,7 +2132,7 @@ var Data = class {
     this.ctx = ctx;
     this.config = config;
     this.dataDir = path.join(this.ctx.baseDir, "data", "chat-analyse");
-    if (this.config.enableAutoBackup) this.ctx.cron("0 0 1 * *", () => {
+    if (this.config.enableAutoBackup) this.ctx.cron("0 2 * * *", () => {
       this.backupCache();
     });
   }
@@ -2148,16 +2148,21 @@ var Data = class {
   async backupCache() {
     try {
       await fs.mkdir(this.dataDir, { recursive: true });
-      const now = /* @__PURE__ */ new Date();
-      const lastMonth = new Date(now.getFullYear(), now.getMonth() - 1, 1);
-      const year = lastMonth.getFullYear();
-      const month = (lastMonth.getMonth() + 1).toString().padStart(2, "0");
-      const filename = `analyse_cache_${year}-${month}.json`;
-      const filepath = path.join(this.dataDir, filename);
       const allUsers = await this.ctx.database.get("analyse_user", {});
       if (allUsers.length === 0) return;
       const uidToUserInfoMap = new Map(allUsers.map((u) => [u.uid, u]));
-      const records = await this.ctx.database.get("analyse_cache", {});
+      const targetDate = /* @__PURE__ */ new Date();
+      targetDate.setMonth(targetDate.getMonth() - 1);
+      const year = targetDate.getFullYear();
+      const monthIndex = targetDate.getMonth();
+      const day = targetDate.getDate();
+      const monthString = (monthIndex + 1).toString().padStart(2, "0");
+      const dayString = day.toString().padStart(2, "0");
+      const filename = `analyse_cache_${year}-${monthString}-${dayString}.json`;
+      const filepath = path.join(this.dataDir, filename);
+      const startDate = new Date(year, monthIndex, day);
+      const endDate = new Date(year, monthIndex, day + 1);
+      const records = await this.ctx.database.get("analyse_cache", { timestamp: { $gte: startDate, $lt: endDate } });
       if (records.length === 0) return;
       const dataToExport = records.map((record) => {
         const userInfo = uidToUserInfoMap.get(record.uid);
@@ -2165,7 +2170,7 @@ var Data = class {
         const { id, uid, ...restOfRecord } = record;
         return { userId: userInfo.userId, channelId: userInfo.channelId, ...restOfRecord };
       }).filter(Boolean);
-      await fs.writeFile(filepath, JSON.stringify(dataToExport, null, 2));
+      if (dataToExport.length > 0) await fs.writeFile(filepath, JSON.stringify(dataToExport, null, 2));
     } catch (error) {
       this.ctx.logger.error("原始记录备份失败:", error);
     }
@@ -2185,19 +2190,29 @@ var Data = class {
         const uidToUserInfoMap = new Map(allUsers.map((u) => [u.uid, u]));
         for (const tableName of tablesToProcess) {
           const filepath = path.join(this.dataDir, `${tableName}.json`);
-          let dataToExport;
-          if (tableName === "analyse_user") {
-            dataToExport = allUsers.map(({ uid, ...rest }) => rest);
-          } else {
-            const records = await this.ctx.database.get(tableName, {});
-            dataToExport = records.map((record) => {
-              const userInfo = uidToUserInfoMap.get(record.uid);
-              if (!userInfo) return null;
-              const { uid, ...restOfRecord } = record;
-              return { userId: userInfo.userId, channelId: userInfo.channelId, ...restOfRecord };
-            }).filter(Boolean);
+          const allDataToExport = [];
+          let offset = 0;
+          while (true) {
+            const batchRecords = await this.ctx.database.get(tableName, {}, {
+              limit: BATCH_SIZE,
+              offset
+            });
+            if (batchRecords.length === 0) break;
+            let processedBatch;
+            if (tableName === "analyse_user") {
+              processedBatch = batchRecords.map(({ uid, ...rest }) => rest);
+            } else {
+              processedBatch = batchRecords.map((record) => {
+                const userInfo = uidToUserInfoMap.get(record.uid);
+                if (!userInfo) return null;
+                const { uid, ...restOfRecord } = record;
+                return { userId: userInfo.userId, channelId: userInfo.channelId, ...restOfRecord };
+              }).filter(Boolean);
+            }
+            allDataToExport.push(...processedBatch);
+            offset += BATCH_SIZE;
           }
-          await fs.writeFile(filepath, JSON.stringify(dataToExport, null, 2));
+          await fs.writeFile(filepath, JSON.stringify(allDataToExport, null, 2));
         }
         return "数据备份成功";
       } catch (error) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "强大而全面的聊天数据分析插件。支持多维度统计（命令、发言、消息类型、活跃度），可生成发言排行、词云图，并提供完善的数据管理。",
-  "version": "1.6.6",
+  "version": "1.6.8",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],