koishi-plugin-chat-analyse 0.4.1 → 0.4.3

package/lib/Collector.d.ts

@@ -34,6 +34,7 @@ declare module 'koishi' {
             timestamp: Date;
         };
         analyse_at: {
+            id: number;
             uid: number;
             target: string;
             content: string;

package/lib/Stat.d.ts

@@ -15,6 +15,13 @@ export declare class Stat {
      * @param {Config} config - 插件的配置对象。
      */
     constructor(ctx: Context, config: Config);
+    /**
+     * @private
+     * @method setupCleanupTask
+     * @description 设置一个定时清理任务。
+     * 此任务会根据配置中的 `rankRetentionDays` 定期删除过期的发言排行数据，以防止数据库膨胀。
+     */
+    private setupCleanupTask;
     /**
      * @public
      * @method registerCommands

package/lib/WhoAt.d.ts

@@ -21,7 +21,7 @@ export declare class WhoAt {
      * @private
      * @method setupCleanupTask
      * @description 设置一个定时清理任务。
-     * 此任务会根据配置中的 `retentionDays` 定期删除过期的@记录，以防止数据库膨胀。
+     * 此任务会根据配置中的 `atRetentionDays` 定期删除过期的@记录，以防止数据库膨胀。
      */
     private setupCleanupTask;
     /**

package/lib/index.d.ts

@@ -17,7 +17,8 @@ export interface Config {
     enableRankStat: boolean;
     enableOriRecord: boolean;
     enableWhoAt: boolean;
-    retentionDays: number;
+    atRetentionDays: number;
+    rankRetentionDays: number;
 }
 /**
  * @const {Schema<Config>} Config

package/lib/index.js

@@ -55,7 +55,7 @@ var Collector = class _Collector {
   static FLUSH_INTERVAL = 60 * 1e3;
   /** @const {number} BUFFER_THRESHOLD - 内存缓存区触发自动刷新的消息数量阈值。 */
   static BUFFER_THRESHOLD = 100;
-  // 分离的缓冲区
+  // 统一的缓冲区
   msgStatBuffer = /* @__PURE__ */ new Map();
   rankStatBuffer = /* @__PURE__ */ new Map();
   cmdStatBuffer = /* @__PURE__ */ new Map();
@@ -106,11 +106,12 @@ var Collector = class _Collector {
     }
     if (this.config.enableWhoAt) {
       this.ctx.model.extend("analyse_at", {
+        id: "unsigned",
         uid: "unsigned",
         target: "string",
         content: "text",
         timestamp: "timestamp"
-      }, { indexes: ["target", "uid"] });
+      }, { primary: "id", autoInc: true, indexes: ["target", "uid"] });
     }
   }
   /**
@@ -545,11 +546,30 @@ var Stat = class {
     this.ctx = ctx;
     this.config = config;
     this.renderer = new Renderer(ctx);
+    this.setupCleanupTask();
   }
   static {
     __name(this, "Stat");
   }
   renderer;
+  /**
+   * @private
+   * @method setupCleanupTask
+   * @description 设置一个定时清理任务。
+   * 此任务会根据配置中的 `rankRetentionDays` 定期删除过期的发言排行数据，以防止数据库膨胀。
+   */
+  setupCleanupTask() {
+    if (this.config.rankRetentionDays > 0) {
+      this.ctx.cron("0 0 * * *", async () => {
+        try {
+          const cutoffDate = new Date(Date.now() - this.config.rankRetentionDays * import_koishi3.Time.day);
+          await this.ctx.database.remove("analyse_rank", { hour: { $lt: cutoffDate } });
+        } catch (error) {
+          this.ctx.logger.error("清理发言排行历史记录出错:", error);
+        }
+      });
+    }
+  }
   /**
    * @public
    * @method registerCommands
@@ -826,13 +846,13 @@ var WhoAt = class {
    * @private
    * @method setupCleanupTask
    * @description 设置一个定时清理任务。
-   * 此任务会根据配置中的 `retentionDays` 定期删除过期的@记录，以防止数据库膨胀。
+   * 此任务会根据配置中的 `atRetentionDays` 定期删除过期的@记录，以防止数据库膨胀。
    */
   setupCleanupTask() {
-    if (this.config.retentionDays > 0) {
+    if (this.config.atRetentionDays > 0) {
       this.ctx.cron("0 0 * * *", async () => {
         try {
-          const cutoffDate = new Date(Date.now() - this.config.retentionDays * import_koishi4.Time.day);
+          const cutoffDate = new Date(Date.now() - this.config.atRetentionDays * import_koishi4.Time.day);
           await this.ctx.database.remove("analyse_at", { timestamp: { $lt: cutoffDate } });
         } catch (error) {
           this.ctx.logger.error("清理 @ 历史记录出错:", error);
@@ -895,12 +915,15 @@ var Config = import_koishi5.Schema.intersect([
   }).description("监听配置"),
   import_koishi5.Schema.object({
     enableCmdStat: import_koishi5.Schema.boolean().default(true).description("启用命令统计"),
-    enableMsgStat: import_koishi5.Schema.boolean().default(true).description("启用消息统计"),
-    enableRankStat: import_koishi5.Schema.boolean().default(true).description("启用发言排行")
-  }).description("命令配置"),
+    enableMsgStat: import_koishi5.Schema.boolean().default(true).description("启用消息统计")
+  }).description("统计配置"),
+  import_koishi5.Schema.object({
+    enableRankStat: import_koishi5.Schema.boolean().default(true).description("启用发言排行"),
+    rankRetentionDays: import_koishi5.Schema.number().min(0).default(31).description("记录保留天数")
+  }).description("发言排行配置"),
   import_koishi5.Schema.object({
     enableWhoAt: import_koishi5.Schema.boolean().default(true).description("启用 @ 记录"),
-    retentionDays: import_koishi5.Schema.number().min(0).default(7).description("保留天数")
+    atRetentionDays: import_koishi5.Schema.number().min(0).default(7).description("记录保留天数")
   }).description("@ 记录配置")
 ]);
 function apply(ctx, config) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "聊天记录分析",
-  "version": "0.4.1",
+  "version": "0.4.3",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],