@arcaneorion/dsh-model-channel-manager 0.3.0 → 0.3.8

package/src/health-store.js

@@ -0,0 +1,169 @@
+/**
+ * 健康数据存储：原始流水（事件/测速）搬入 storageDomain 的 per-record KV 表。
+ *
+ * 动机（审计 F11/F12/F23 的根因）：0.1 时代健康流水整字段存在 settings 命名空间里，
+ * 与配置共用同一条「HMR 独占事务 + 整字段落盘」通道，导致：
+ *  - 2s 防抖 flush 与 volatile-update 快照覆盖互相踩（刚记的账被旧快照抹掉）；
+ *  - profile patch 文件膨胀（实测 6350 行中 health 约占 3000 行）；
+ *  - 任何无关 settings 写入都会整段重写健康数据。
+ *
+ * 现在的归属：事实数据（append-only 事件）进 domain；settings 里只保留
+ * 一份给 client 渲染用的小投影（聚合摘要 + 运行态指针），由 host 定期刷新。
+ *
+ * 组合方式照抄 dsh-session-projection-cache 范本：
+ *  - 依赖方 inject ['storageDomain']（dsh-base 已装 json 后端，root=storages/）；
+ *  - Service.init 里 open(spec)，effect 持有 close；
+ *  - per-record 布局：一条事件一个文档，坏记录 backup-and-skip 不阻塞整体。
+ */
+import { z } from 'zod';
+import { defineDomain, domainTable } from '@deepseek-ai/dsh-storage-domain';
+
+const WINDOW_MS = 7 * 24 * 3600 * 1000;
+// 每渠道保留上限：30m/24h 视图 + selector 置顶绰绰有余，同时控制 storages/ 目录体量
+const PER_KEY_LIMIT = 300;
+
+/** 单条健康事件记录（zod，domain 落盘边界校验）。向后兼容：token 字段可选。 */
+const eventRecord = z.object({
+  ts: z.number().int().nonnegative(),
+  provider: z.string(),
+  model: z.string(),
+  ok: z.boolean(),
+  ttftMs: z.number().nullable().optional(),
+  latencyMs: z.number().nullable().optional(),
+  code: z.string().nullable().optional(),
+  inputTokens: z.number().optional(),
+  outputTokens: z.number().optional(),
+  cacheReadTokens: z.number().optional(),
+  cacheWriteTokens: z.number().optional(),
+  reasoningTokens: z.number().optional(),
+});
+
+/** 单渠道的事件桶记录：key = provider（与旧 state.records 的键一致，迁移零转换）。 */
+const bucketRecord = z.object({
+  events: z.array(eventRecord),
+});
+
+/** 测速结果桶：key = 组 id（与旧 state.speedResults 的键一致）。 */
+const speedRecord = z.object({
+  rows: z.array(z.object({
+    provider: z.string(),
+    model: z.string(),
+    ok: z.boolean(),
+    ttft: z.number().nullable(),
+    latency: z.number().nullable(),
+    at: z.number(),
+    failure: z.string().nullable().optional(),
+  })),
+});
+
+const healthDomainSpec = defineDomain({
+  name: 'model_channel_health',
+  version: 1,
+  layout: 'per-record',
+  invalidRecords: 'backup-and-skip',
+  tables: {
+    events: domainTable(bucketRecord),
+    speed: domainTable(speedRecord),
+  },
+});
+
+/**
+ * HealthStore：封装 domain 读写 + 保留策略 + 聚合。
+ * 非 Service（宿主插件自有生命周期），由 apply() 创建并 effect 持有 close。
+ */
+export class HealthStore {
+  constructor(ctx) {
+    this.ctx = ctx;
+    this.ready = null; // Promise<Domain>
+    this.events = null; // KvTable
+    this.speed = null; // KvTable
+    this.eventChain = Promise.resolve(); // 类内写链：appendEvent 串行化（并发不丢更新）
+  }
+
+  /** 打开 domain（幂等）。dsh-base 未装 storage 栈的 profile 会 reject——调用方降级。 */
+  open() {
+    if (this.ready === null) {
+      this.ready = this.ctx.storageDomain.open(healthDomainSpec).then((domain) => {
+        this.events = domain.table('events');
+        this.speed = domain.table('speed');
+        return domain;
+      });
+    }
+    return this.ready;
+  }
+
+  /** effect 用：domain 关闭（写入队列排干后释放）。 */
+  async close() {
+    if (this.ready === null) return;
+    const domain = await this.ready.catch(() => null);
+    this.ready = null;
+    this.events = null;
+    this.speed = null;
+    if (domain) await domain.close().catch(() => {});
+  }
+
+  /** 追加一条事件到渠道桶，同时执行窗口过期 + 条数截断。
+   * 并发安全：桶初始化（put）与追加（update）都排队到类内写链，逐个重读当前值，
+   * 并发 appendEvent 不交错、不互相覆盖（等价 KvTable.update 的原子语义；
+   * 桶不存在时 KvTable.update 报 missing-key，故初始化先行）。
+   */
+  appendEvent(provider, rec) {
+    if (this.events === null) return Promise.resolve();
+    this.eventChain = this.eventChain.then(async () => {
+      const cutoff = Date.now() - WINDOW_MS;
+      const cur = this.events.get(provider);
+      if (cur === undefined) {
+        await this.events.put(provider, { events: [rec] });
+        return;
+      }
+      await this.events.update(provider, (c) => ({
+        events: [...c.events.filter((e) => e.ts >= cutoff), rec].slice(-PER_KEY_LIMIT),
+      }));
+    }).catch(() => { }); // 链不断：失败吞掉（调用方已有 warn），后续追加继续
+    return this.eventChain;
+  }
+
+  /** 整组测速结果替换写入。 */
+  async putSpeedRows(groupId, rows) {
+    if (this.speed === null) return;
+    await this.speed.put(groupId, { rows });
+  }
+
+  /** 读全部事件桶（迁移/聚合用）。 */
+  allEventBuckets() {
+    const out = {};
+    if (this.events === null) return out;
+    for (const [provider, bucket] of this.events.entries()) out[provider] = bucket.events;
+    return out;
+  }
+
+  /** 读全部测速桶。 */
+  allSpeedBuckets() {
+    const out = {};
+    if (this.speed === null) return out;
+    for (const [groupId, bucket] of this.speed.entries()) out[groupId] = bucket.rows;
+    return out;
+  }
+
+  /**
+   * 存量迁移：settings health 子树的 records/speedResults 一次性搬入 domain。
+   * 迁移以「桶里有数据即跳过」防重放；完成后由调用方写迁移标记。
+   * 并发安全：逐桶判断改走原子链（迁移进行中 recordHealth 并发创建的新桶不会被旧数据覆盖）。
+   */
+  async migrateFrom(records, speedResults) {
+    if (this.events === null) return false;
+    const cutoff = Date.now() - WINDOW_MS;
+    for (const [provider, list] of Object.entries(records || {})) {
+      if (!Array.isArray(list) || list.length === 0) continue;
+      if (this.events.get(provider) !== undefined) continue; // 已有桶：domain 是权威，不回搬
+      const events = list.filter((e) => e && e.ts >= cutoff).slice(-PER_KEY_LIMIT);
+      if (events.length > 0) await this.events.put(provider, { events });
+    }
+    for (const [groupId, rows] of Object.entries(speedResults || {})) {
+      if (!Array.isArray(rows) || rows.length === 0) continue;
+      if (this.speed.get(groupId) !== undefined) continue;
+      await this.speed.put(groupId, { rows });
+    }
+    return true;
+  }
+}