koishi-plugin-chat-analyse 0.5.0 → 0.5.1

package/lib/Collector.d.ts

@@ -0,0 +1,88 @@
+import { Context } from 'koishi';
+import { Config } from './index';
+declare module 'koishi' {
+    interface Tables {
+        analyse_user: {
+            uid: number;
+            channelId: string;
+            userId: string;
+            channelName: string;
+            userName: string;
+        };
+        analyse_cmd: {
+            uid: number;
+            command: string;
+            count: number;
+            timestamp: Date;
+        };
+        analyse_msg: {
+            uid: number;
+            type: string;
+            count: number;
+            timestamp: Date;
+        };
+        analyse_rank: {
+            uid: number;
+            type: string;
+            count: number;
+            timestamp: Date;
+        };
+        analyse_cache: {
+            id: number;
+            uid: number;
+            content: string;
+            timestamp: Date;
+        };
+        analyse_at: {
+            id: number;
+            uid: number;
+            target: string;
+            content: string;
+            timestamp: Date;
+        };
+    }
+}
+/**
+ * @class Collector
+ * @description 核心数据收集器。根据配置，高效地监听、收集、缓冲并持久化聊天数据。
+ */
+export declare class Collector {
+    private ctx;
+    private config;
+    /** @property FLUSH_INTERVAL - 内存缓存区定时刷入数据库的间隔（毫秒）。 */
+    private static readonly FLUSH_INTERVAL;
+    /** @property BUFFER_THRESHOLD - 内存缓存区触发刷新的消息数量阈值。 */
+    private static readonly BUFFER_THRESHOLD;
+    private msgStatBuffer;
+    private rankStatBuffer;
+    private cmdStatBuffer;
+    private oriCacheBuffer;
+    private whoAtBuffer;
+    private userCache;
+    private channelCache;
+    private pendingRequests;
+    private flushInterval;
+    /**
+     * @param ctx - Koishi 的插件上下文。
+     * @param config - 插件的配置对象。
+     */
+    constructor(ctx: Context, config: Config);
+    /**
+     * @private @method onMessage
+     * @description 统一的消息事件处理器，解析消息并更新各类统计数据的缓冲区。
+     * @param session - Koishi 的会话对象。
+     */
+    private onMessage;
+    /**
+     * @private @method sanitizeContent
+     * @description 将 Koishi 消息元素数组净化为纯文本字符串。
+     * @param elements - 消息元素数组。
+     * @returns 净化后的纯文本。
+     */
+    private sanitizeContent;
+    /**
+     * @private @method flushBuffers
+     * @description 将所有内存中的数据缓冲区批量写入数据库，并清空缓冲区。
+     */
+    private flushBuffers;
+}

package/lib/index.js

@@ -68,7 +68,7 @@ var Collector = class _Collector {
   /** @property FLUSH_INTERVAL - 内存缓存区定时刷入数据库的间隔（毫秒）。 */
   static FLUSH_INTERVAL = import_koishi.Time.minute;
   /** @property BUFFER_THRESHOLD - 内存缓存区触发刷新的消息数量阈值。 */
-  static BUFFER_THRESHOLD = 100;
+  static BUFFER_THRESHOLD = 60;
   // 统一的数据缓冲区
   msgStatBuffer = /* @__PURE__ */ new Map();
   rankStatBuffer = /* @__PURE__ */ new Map();
@@ -76,7 +76,8 @@ var Collector = class _Collector {
   oriCacheBuffer = [];
   whoAtBuffer = [];
   userCache = /* @__PURE__ */ new Map();
-  pendingUserRequests = /* @__PURE__ */ new Map();
+  channelCache = /* @__PURE__ */ new Map();
+  pendingRequests = /* @__PURE__ */ new Map();
   flushInterval;
   /**
    * @private @method onMessage
@@ -90,35 +91,41 @@ var Collector = class _Collector {
     let user;
     if (this.userCache.has(cacheKey)) {
       user = this.userCache.get(cacheKey);
-    } else if (this.pendingUserRequests.has(cacheKey)) {
-      user = await this.pendingUserRequests.get(cacheKey);
+    } else if (this.pendingRequests.has(cacheKey)) {
+      user = await this.pendingRequests.get(cacheKey);
     } else {
       const promise = (async () => {
         try {
           const [dbUser] = await this.ctx.database.get("analyse_user", { channelId, userId });
           const currentUserName = session.username ?? "";
-          const guild = await bot.getGuild(channelId).catch(() => null);
-          const currentChannelName = guild?.name ?? "";
+          let currentChannelName = this.channelCache.get(channelId);
+          if (currentChannelName === void 0) {
+            const guild = await bot.getGuild(channelId).catch(() => null);
+            currentChannelName = guild?.name ?? "";
+            if (currentChannelName) this.channelCache.set(channelId, currentChannelName);
+          }
           if (dbUser) {
             if (currentUserName && dbUser.userName !== currentUserName || currentChannelName && dbUser.channelName !== currentChannelName) {
               await this.ctx.database.set("analyse_user", { uid: dbUser.uid }, { userName: currentUserName, channelName: currentChannelName });
               dbUser.userName = currentUserName;
               dbUser.channelName = currentChannelName;
             }
-            this.userCache.set(cacheKey, dbUser);
-            return dbUser;
+            const cacheData2 = { uid: dbUser.uid, userName: dbUser.userName };
+            this.userCache.set(cacheKey, cacheData2);
+            return cacheData2;
           }
           const createdUser = await this.ctx.database.create("analyse_user", { channelId, userId, userName: currentUserName, channelName: currentChannelName });
-          this.userCache.set(cacheKey, createdUser);
-          return createdUser;
+          const cacheData = { uid: createdUser.uid, userName: createdUser.userName };
+          this.userCache.set(cacheKey, cacheData);
+          return cacheData;
         } catch (error) {
           this.ctx.logger.error(`创建或获取用户(${cacheKey})失败:`, error);
           return null;
         } finally {
-          this.pendingUserRequests.delete(cacheKey);
+          this.pendingRequests.delete(cacheKey);
         }
       })();
-      this.pendingUserRequests.set(cacheKey, promise);
+      this.pendingRequests.set(cacheKey, promise);
       user = await promise;
     }
     if (!user) return;

package/package.json

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