feat(proxy-fitness): pool fitness registry + Smart rotation + pool-scoped retry

Pool/IP fitness registry (globalThis-backed, provider::model scopes, 5-min cooldown, provider::* wildcard) fed by pool-scoped failures: freebuff limited-IP / model-locked gates and opencode free per-IP limits. chatCore retries a failed pool via another pool without locking the account; new Smart rotation strategy (per-connection + no-auth providers) skips unfit pools. Proxy Fitness dashboard page: active-block table with provider/IP filters, per-record Clear and provider-scoped Clear All; egress column via pool geo enrichment. Unit tests for the registry.
2026-08-07 03:51:19 +08:00
parent 90ff8191ff
commit fd22290a34
17 changed files with 1000 additions and 35 deletions
@@ -0,0 +1,56 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import {
+  markPoolUnfit,
+  clearPoolUnfit,
+  clearAllPoolUnfit,
+  isPoolFit,
+  fitPoolIds,
+  poolFitnessSnapshot,
+  pruneExpired,
+  resetPoolFitness,
+} from "open-sse/services/proxyPoolFitness.js";
+
+describe("proxy pool fitness registry", () => {
+  beforeEach(() => resetPoolFitness());
+
+  it("marks a pool unfit for a scope and prunes on expiry", () => {
+    markPoolUnfit("p1", "freebuff::gpt-5.6-luna", Date.now() + 60_000, "limited_ip");
+    expect(isPoolFit("p1", "freebuff::gpt-5.6-luna")).toBe(false);
+    expect(isPoolFit("p1", "freebuff::other-model")).toBe(true);
+    expect(isPoolFit("p2", "freebuff::gpt-5.6-luna")).toBe(true);
+
+    markPoolUnfit("p1", "freebuff::gpt-5.6-luna", Date.now() - 1000); // expired
+    expect(isPoolFit("p1", "freebuff::gpt-5.6-luna")).toBe(true); // pruned on read
+  });
+
+  it("provider-wide mark (provider::*) covers any model lookup", () => {
+    markPoolUnfit("p1", "opencode::*", Date.now() + 60_000, "manual");
+    expect(isPoolFit("p1", "opencode::sonnet-4.6")).toBe(false);
+    expect(isPoolFit("p1", "freebuff::gpt-5.6-luna")).toBe(true);
+  });
+
+  it("fitPoolIds filters unfit pools; snapshot drops expired marks", () => {
+    markPoolUnfit("p1", "fb::m1", Date.now() + 60_000);
+    markPoolUnfit("p1", "fb::m2", Date.now() - 1000); // expired
+    expect(fitPoolIds(["p1", "p2"], "fb::m1")).toEqual(["p2"]);
+
+    const snap = poolFitnessSnapshot();
+    expect(snap.p1["fb::m1"]).toBeDefined();
+    expect(snap.p1["fb::m2"]).toBeUndefined();
+  });
+
+  it("clear per scope, clear-all per provider, clear-all global, pruneExpired", () => {
+    markPoolUnfit("p1", "freebuff::m1", Date.now() + 60_000);
+    markPoolUnfit("p1", "kiro::m3", Date.now() + 60_000);
+
+    clearPoolUnfit("p1", "freebuff::m1");
+    expect(isPoolFit("p1", "freebuff::m1")).toBe(true);
+
+    clearAllPoolUnfit("kiro");
+    expect(poolFitnessSnapshot().p1).toBeUndefined();
+
+    markPoolUnfit("p2", "x::y", Date.now() - 1000);
+    expect(pruneExpired()).toBe(1);
+    expect(poolFitnessSnapshot()).toEqual({});
+  });
+});