dsh-layered-memory 0.6.1 → 0.7.1

package/dist/pipeline/runner.js

@@ -32,6 +32,8 @@ export class MemoryRunner {
     live;
     tasks = [];
     draining = false;
+    /** 停止标志（dispose 序置位）：不再取新任务；进行中任务自然收尾，其 DB 写入失败由各层兜底捕获。 */
+    stopped = false;
     pending = emptyPending();
     pendingFile;
     background = [];
@@ -104,7 +106,13 @@ export class MemoryRunner {
     runRebuildTurn(sessionId, messages) {
         return this.runTurn(sessionId, messages, 'auto', { noBufferCap: true });
     }
+    /** 停止取新任务（插件 dispose 序调用；进行中任务照常跑完但不 await——LLM 慢调用不拖住宿主卸载）。 */
+    stop() {
+        this.stopped = true;
+    }
     pushTask(task) {
+        if (this.stopped)
+            return;
         this.tasks.push(task);
         void this.drain();
     }
@@ -113,7 +121,7 @@ export class MemoryRunner {
             return;
         this.draining = true;
         try {
-            while (this.tasks.length > 0) {
+            while (!this.stopped && this.tasks.length > 0) {
                 const [task] = this.tasks.splice(pickNextTaskIndex(this.tasks), 1);
                 try {
                     await task.run();
@@ -127,9 +135,18 @@ export class MemoryRunner {
             this.draining = false;
         }
     }
-    /** 缓冲落盘（每次蒸馏尝试后调用；失败只告警不阻断管线）。 */
-    async persistPending() {
+    /** 缓冲落盘（每次蒸馏尝试后调用；失败只告警不阻断管线）。
+     *  非重建轮持久化前按桶截断到上限：重建取消后的大桶不至于在后续每次
+     *  蒸馏尝试时反复整量序列化落盘（多 MB 级 IO）；重建轮豁免维持。 */
+    async persistPending(noBufferCap = false) {
         try {
+            if (!noBufferCap) {
+                for (const key of PENDING_MODES) {
+                    const bucket = this.pending[key];
+                    if (bucket.length > PENDING_BUCKET_CAP)
+                        this.pending[key] = bucket.slice(-PENDING_BUCKET_CAP);
+                }
+            }
             await savePending(this.pendingFile, this.pending);
         }
         catch (err) {
@@ -161,13 +178,11 @@ export class MemoryRunner {
                 this.logger.info(`[memory] L1 阶段完成（${Date.now() - t}ms）`);
             }
             catch (err) {
-                // 保留 pending，下次重试；但防止无限堆积（重建轮不限，量被会话规模天然约束）
-                if (!opts?.noBufferCap && this.pending[mode].length > PENDING_BUCKET_CAP)
-                    this.pending[mode] = [];
+                // 保留 pending 下次重试（runTurn 入口已裁到 ≤200，防无限堆积；重建轮不裁，量被会话规模约束）
                 this.logger.warn(`[memory] L1 抽取失败（mode=${mode}，pending=${this.pending[mode].length}）: ${errDetail(err)}`);
             }
             // 缓冲每次尝试后立即落盘：进程中途退出不丢待重试/攒阈值状态
-            await this.persistPending();
+            await this.persistPending(opts?.noBufferCap);
             // L1 计数推进后立即落盘：L2/L3 失败或进程中途退出不得回滚阈值进度
             // （记录已入库但计数丢失会让该族 L2 永远差一截，state 与 DB 脱节）
             try {