Merge branch 'cache2'

This commit is contained in:
root committed 2026-09-29 17:42:24 +08:00
commit eb9cfaa63d
32 files changed
+1927 -87

No files matched your search

@@ -1,5 +1,11 @@
import { createCipheriv, createDecipheriv, createHmac, randomBytes, randomUUID } from "node:crypto"; import { createCipheriv, createDecipheriv, createHmac, randomBytes, randomUUID } from "node:crypto";
import { DEFAULT_SETTINGS, GlobalSettings, type SecretPurpose, type SettingsKey } from "@umt/contracts"; import {
DEFAULT_SETTINGS,
GlobalSettings,
type SecretPurpose,
type SettingsKey,
withStorageDefaults,
} from "@umt/contracts";
import type { DB, Kysely } from "@umt/db"; import type { DB, Kysely } from "@umt/db";
import type { Logger } from "pino"; import type { Logger } from "pino";
import type { EventBusPort } from "../../http/sse.ts"; import type { EventBusPort } from "../../http/sse.ts";
@@ -335,7 +341,13 @@ export function createSettingsReader(deps: {
for (const key of Object.keys(DEFAULT_SETTINGS) as SettingsKey[]) { for (const key of Object.keys(DEFAULT_SETTINGS) as SettingsKey[]) {
const def = DEFAULT_SETTINGS[key]; const def = DEFAULT_SETTINGS[key];
const v = stored.get(key); const v = stored.get(key);
const merged = v && typeof v === "object" && !Array.isArray(v) ? { ...def, ...(v as object) } : def; // storage 是两层结构:逐层补缺省,旧库里缺子字段也能整体通过校验
const merged =
key === "storage"
? withStorageDefaults(v)
: v && typeof v === "object" && !Array.isArray(v)
? { ...def, ...(v as object) }
: def;
const parsed = GlobalSettings.shape[key].safeParse(merged); const parsed = GlobalSettings.shape[key].safeParse(merged);
if (!parsed.success) deps.logger.warn({ key }, "settings 值不合法,回退缺省"); if (!parsed.success) deps.logger.warn({ key }, "settings 值不合法,回退缺省");
out[key] = parsed.success ? parsed.data : def; out[key] = parsed.success ? parsed.data : def;
+51 -4
View File
@@ -9,6 +9,8 @@ import { TMDB_IMAGE_SIZES } from "@umt/contracts";
// - 读缓存只在 use_cache(GLOBAL_IMAGE_CACHE)开启时;拉到的合法图片总是写盘(与 MP 一致); // - 读缓存只在 use_cache(GLOBAL_IMAGE_CACHE)开启时;拉到的合法图片总是写盘(与 MP 一致);
// - 合法图片按内容识别 MIME,拒绝非图片与 SVG;ETag = md5(内容),Cache-Control: public, max-age=7 天,If-None-Match 相同回 304; // - 合法图片按内容识别 MIME,拒绝非图片与 SVG;ETag = md5(内容),Cache-Control: public, max-age=7 天,If-None-Match 相同回 304;
// - 清理:删除 mtime 早于 GLOBAL_IMAGE_CACHE_DAYS 天的文件与空目录(MP clear_temp → SystemUtils.clear)。 // - 清理:删除 mtime 早于 GLOBAL_IMAGE_CACHE_DAYS 天的文件与空目录(MP clear_temp → SystemUtils.clear)。
// - CACHE-2 容量 LRU(我们加的):读盘命中时把 atime 显式写成当前时间(不依赖挂载的 relatime),
// 超过 storage.budgets.imagesGb 时按 atime 从旧到新删到预算的 90%;按天清理保留作兜底。
// 我们只代理 TMDB 图片域(MP SECURITY_IMAGE_DOMAINS 的 TMDB 项),上游失败回 502。 // 我们只代理 TMDB 图片域(MP SECURITY_IMAGE_DOMAINS 的 TMDB 项),上游失败回 502。
export const TMDB_IMAGE_HOST = "image.tmdb.org"; export const TMDB_IMAGE_HOST = "image.tmdb.org";
@@ -51,6 +53,13 @@ export function imageMimeType(b: Uint8Array): string | null {
return null; return null;
} }
interface ImageFile {
path: string;
size: number;
mtimeMs: number;
atimeMs: number;
}
export interface ImageCacheOptions { export interface ImageCacheOptions {
/** DATA_DIR/cache/images */ /** DATA_DIR/cache/images */
dir: string; dir: string;
@@ -104,7 +113,10 @@ export class ImageCache {
if (useCache) { if (useCache) {
const cached = await fs.readFile(path).catch(() => null); const cached = await fs.readFile(path).catch(() => null);
const mime = cached?.length ? imageMimeType(cached) : null; const mime = cached?.length ? imageMimeType(cached) : null;
if (cached && mime) return { content: cached, mime }; if (cached && mime) {
void this.touch(path);
return { content: cached, mime };
}
} }
let res: Response; let res: Response;
try { try {
@@ -130,8 +142,18 @@ export class ImageCache {
return { content, mime }; return { content, mime };
} }
private async files(): Promise<{ path: string; size: number; mtimeMs: number }[]> { /** 记一次访问:atime = 现在,mtime 不变(按天清理仍按写入时间) */
const out: { path: string; size: number; mtimeMs: number }[] = []; private async touch(path: string): Promise<void> {
try {
const st = await fs.stat(path);
await fs.utimes(path, new Date(), st.mtime);
} catch {
// 文件刚被清理等:忽略
}
}
private async files(): Promise<ImageFile[]> {
const out: ImageFile[] = [];
const walk = async (d: string) => { const walk = async (d: string) => {
let names: string[]; let names: string[];
try { try {
@@ -144,7 +166,7 @@ export class ImageCache {
const st = await fs.stat(p).catch(() => null); const st = await fs.stat(p).catch(() => null);
if (!st) continue; if (!st) continue;
if (st.isDirectory()) await walk(p); if (st.isDirectory()) await walk(p);
else out.push({ path: p, size: st.size, mtimeMs: st.mtimeMs }); else if (!n.endsWith(".tmp")) out.push({ path: p, size: st.size, mtimeMs: st.mtimeMs, atimeMs: st.atimeMs });
} }
}; };
await walk(this.opts.dir); await walk(this.opts.dir);
@@ -156,6 +178,31 @@ export class ImageCache {
return { bytes: f.reduce((a, x) => a + x.size, 0), files: f.length }; return { bytes: f.reduce((a, x) => a + x.size, 0), files: f.length };
} }
/**
* 容量 LRU:总量超过 maxBytes 时按最近访问(atime)从旧到新删除,直到不超过 maxBytes × lowWater;
* 没超不动。返回删除数、释放字节与清理后用量。
*/
async enforceBudget(
maxBytes: number,
lowWater = 0.9,
): Promise<{ deleted: number; freedBytes: number; usedBytes: number }> {
const files = await this.files();
let used = files.reduce((a, f) => a + f.size, 0);
if (used <= maxBytes) return { deleted: 0, freedBytes: 0, usedBytes: used };
const target = Math.floor(maxBytes * lowWater);
files.sort((a, b) => a.atimeMs - b.atimeMs || a.mtimeMs - b.mtimeMs);
let deleted = 0;
let freed = 0;
for (const f of files) {
if (used <= target) break;
await fs.rm(f.path, { force: true });
used -= f.size;
freed += f.size;
deleted++;
}
return { deleted, freedBytes: freed, usedBytes: used };
}
/** SystemUtils.clear(images, days):删 mtime 早于 days 天的文件,再删空目录;返回删除数 */ /** SystemUtils.clear(images, days):删 mtime 早于 days 天的文件,再删空目录;返回删除数 */
async clean(days: number, now = Date.now()): Promise<number> { async clean(days: number, now = Date.now()): Promise<number> {
let deleted = 0; let deleted = 0;
+23 -2
View File
@@ -12,7 +12,7 @@ import type { z } from "zod";
import { dataPaths } from "../../config.ts"; import { dataPaths } from "../../config.ts";
import { requireScope } from "../../http/auth.ts"; import { requireScope } from "../../http/auth.ts";
import { errorBody } from "../../http/errors.ts"; import { errorBody } from "../../http/errors.ts";
import { CACHE_REGIONS, sharedCache, tmdbCachePolicy } from "../../shared-cache.ts"; import { CACHE_REGIONS, sharedCache, storageSettings, tmdbCachePolicy } from "../../shared-cache.ts";
import { registerScheduledJob } from "../schedules/registry.ts"; import { registerScheduledJob } from "../schedules/registry.ts";
import type { ModuleApp, ServerContext, ServerModule } from "../types.ts"; import type { ModuleApp, ServerContext, ServerModule } from "../types.ts";
import { ImageCache, ImageProxyError } from "./images.ts"; import { ImageCache, ImageProxyError } from "./images.ts";
@@ -22,7 +22,8 @@ import { ImageCache, ImageProxyError } from "./images.ts";
* - PG 缓存 kv_cache(UNLOGGED,移植 MP TTLCache)的过期行物理清理任务 cache.sweep(每 10 分钟); * - PG 缓存 kv_cache(UNLOGGED,移植 MP TTLCache)的过期行物理清理任务 cache.sweep(每 10 分钟);
* - TMDB 缓存管理(对照 MP tmdb_cache_items / tmdb_cache_clear):GET /cache/tmdb、POST /cache/tmdb/clear; * - TMDB 缓存管理(对照 MP tmdb_cache_items / tmdb_cache_clear):GET /cache/tmdb、POST /cache/tmdb/clear;
* - 图片代理 GET /img/tmdb/{size}/{file}(会话或 API Key read,移植 MP ImageHelper / fetch_image)+ DATA_DIR/cache/images 落盘, * - 图片代理 GET /img/tmdb/{size}/{file}(会话或 API Key read,移植 MP ImageHelper / fetch_image)+ DATA_DIR/cache/images 落盘,
* 用量 / 清空 GET /cache/images、POST /cache/images/clear,按保留天数清理的任务 cache.images_clean(每天)。 * 用量 / 清空 GET /cache/images、POST /cache/images/clear,按保留天数清理的任务 cache.images_clean(每天);
* - CACHE-2:图片缓存容量 LRU 任务 cache.images_lru(30 分钟,预算 storage.budgets.imagesGb,超出删最久未访问)。
*/ */
export interface CacheModuleOptions { export interface CacheModuleOptions {
/** 图片上游 fetch(测试注入) */ /** 图片上游 fetch(测试注入) */
@@ -203,6 +204,26 @@ export function createCacheModule(opts: CacheModuleOptions = {}): ServerModule &
return { summary: `删除 ${n} 张过期图片` }; return { summary: `删除 ${n} 张过期图片` };
}, },
}); });
registerScheduledJob(ctx, {
key: "cache.images_lru",
defaults: {
name: "图片缓存容量控制",
description: "图片代理缓存超过容量预算(存储设置,缺省 20 GB)时,按最久未访问删除到预算的 90%",
group: "maintain",
schedule: { type: "interval", everyMs: 30 * 60_000 },
firstRun: "after_interval",
},
run: async () => {
const { imagesGb } = (await storageSettings(ctx)).budgets;
const r = await images(ctx).enforceBudget(Math.round(imagesGb * 1024 ** 3));
const gb = (b: number) => `${Math.round((b / 1024 ** 3) * 100) / 100} GB`;
return {
summary: r.deleted
? `删除 ${r.deleted} 张最久未访问的图片,释放 ${gb(r.freedBytes)},现用 ${gb(r.usedBytes)}`
: `现用 ${gb(r.usedBytes)},未超过预算 ${imagesGb} GB`,
};
},
});
}, },
}; };
} }
@@ -15,7 +15,14 @@ import {
} from "@umt/adapters"; } from "@umt/adapters";
import { DEFAULT_SETTINGS, type SitesSettings } from "@umt/contracts"; import { DEFAULT_SETTINGS, type SitesSettings } from "@umt/contracts";
import { pgTokenBucketFor, type SiteRow, sql } from "@umt/db"; import { pgTokenBucketFor, type SiteRow, sql } from "@umt/db";
import { CACHE_REGIONS, sharedCache, type TmdbCachePolicy, tmdbCachePolicy } from "../../shared-cache.ts"; import {
CACHE_REGIONS,
renewTrackedTmdbDetail,
sharedCache,
type TmdbCachePolicy,
tmdbCachePolicy,
tmdbDetailRef,
} from "../../shared-cache.ts";
import type { ServerContext } from "../types.ts"; import type { ServerContext } from "../types.ts";
// 模块运行时依赖:站点适配器与限流器(每站一个实例,令牌在 PG 跨进程共享)、RSS 抓取器、TMDB 客户端、时钟。 // 模块运行时依赖:站点适配器与限流器(每站一个实例,令牌在 PG 跨进程共享)、RSS 抓取器、TMDB 客户端、时钟。
@@ -108,7 +115,11 @@ export class SourcesRuntime {
...(baseUrl ? { baseUrl } : {}), ...(baseUrl ? { baseUrl } : {}),
...(this.overrides.fetch ? { fetch: this.overrides.fetch } : {}), ...(this.overrides.fetch ? { fetch: this.overrides.fetch } : {}),
// 请求响应进 PG 缓存(region tmdb.request),三进程共享;TTL / 条数上限 / 空结果短 TTL 取设置(对照 MP) // 请求响应进 PG 缓存(region tmdb.request),三进程共享;TTL / 条数上限 / 空结果短 TTL 取设置(对照 MP)
cache: { ttlMs: policy.ttl * 1000, emptyTtlMs: policy.emptyTtl * 1000 }, cache: {
ttlMs: policy.ttl * 1000,
searchTtlMs: policy.searchTtl * 1000,
emptyTtlMs: policy.emptyTtl * 1000,
},
sharedCache: tmdbSharedCache(this.ctx, policy), sharedCache: tmdbSharedCache(this.ctx, policy),
}) })
: null; : null;
@@ -276,17 +287,27 @@ export class SourcesRuntime {
} }
} }
/** TMDB 请求响应缓存:PG kv_cache region=tmdb.request;PG 出错按未命中 / 不写处理,不让 TMDB 请求失败 */ /**
* TMDB 请求响应缓存:PG kv_cache region=tmdb.request;PG 出错按未命中 / 不写处理,不让 TMDB 请求失败。
* 详情命中且属于追踪中的作品时续期(CACHE-2),续期失败只记 debug。
*/
function tmdbSharedCache(ctx: ServerContext, policy: TmdbCachePolicy): TmdbSharedCache { function tmdbSharedCache(ctx: ServerContext, policy: TmdbCachePolicy): TmdbSharedCache {
const cache = sharedCache(ctx); const cache = sharedCache(ctx);
const warn = (err: unknown) => const warn = (err: unknown) =>
ctx.logger.warn({ err: { message: (err as Error).message } }, "TMDB 缓存读写失败,按未命中处理"); ctx.logger.warn({ err: { message: (err as Error).message } }, "TMDB 缓存读写失败,按未命中处理");
return { return {
get: (key) => get: async (key) => {
cache.get(CACHE_REGIONS.tmdbRequest, key).catch((e) => { const v = await cache.get(CACHE_REGIONS.tmdbRequest, key).catch((e) => {
warn(e); warn(e);
return null; return null;
}), });
if (v !== null && tmdbDetailRef(key)) {
await renewTrackedTmdbDetail(ctx, key, policy.ttl).catch((e) =>
ctx.logger.debug({ err: { message: (e as Error).message } }, "TMDB 详情缓存续期失败"),
);
}
return v;
},
set: (key, value, ttlMs) => set: (key, value, ttlMs) =>
cache.set(CACHE_REGIONS.tmdbRequest, key, value, { ttl: ttlMs / 1000, maxsize: policy.maxsize }).catch(warn), cache.set(CACHE_REGIONS.tmdbRequest, key, value, { ttl: ttlMs / 1000, maxsize: policy.maxsize }).catch(warn),
}; };
@@ -1,18 +1,18 @@
import type { SiteDetailPage } from "@umt/adapters"; import type { SiteDetailPage } from "@umt/adapters";
import type { SiteReleasePage } from "@umt/contracts"; import type { SiteReleasePage } from "@umt/contracts";
import { sql } from "@umt/db"; import { sql } from "@umt/db";
import { sharedCache } from "../../../shared-cache.ts"; import { sharedCache, storageSettings } from "../../../shared-cache.ts";
import type { SourcesRuntime } from "../runtime.ts"; import type { SourcesRuntime } from "../runtime.ts";
import type { Exec } from "./common.ts"; import type { Exec } from "./common.ts";
// 站点详情页扩展内容(简介 / MediaInfo / 截图 / 外部 ID / 文件 / 标签 / 上传者): // 站点详情页扩展内容(简介 / MediaInfo / 截图 / 外部 ID / 文件 / 标签 / 上传者):
// - 只在按需抓详情时得到(工作台打开详情、release.inspect 复核),放 PG kv_cache region site.detail, // - 只在按需抓详情时得到(工作台打开详情、release.inspect 复核),放 PG kv_cache region site.detail,
// key = source_release.id,TTL = 站点详情缓存期(至少 24 小时);不进 source_release 列,过期即丢。 // key = source_release.id,TTL = storage.cacheTtl.siteDetailDays(CACHE-2,缺省 7 天;不短于站点详情缓存期);
// 不进 source_release 列,过期即丢。促销 / H&R 不靠它:下载前复核走 refreshReleaseDetail(按站点详情缓存期判断是否重抓)。
// - 外部 ID(IMDb / 豆瓣 / TMDB)另写 source_release.raw._detail(列表快照 upsert 不覆盖它),供身份解析用。 // - 外部 ID(IMDb / 豆瓣 / TMDB)另写 source_release.raw._detail(列表快照 upsert 不覆盖它),供身份解析用。
// - 截图对外只给代理路径 /api/v1/img/site-detail/{releaseId}/{index},原始地址留在服务端。 // - 截图对外只给代理路径 /api/v1/img/site-detail/{releaseId}/{index},原始地址留在服务端。
export const DETAIL_PAGE_REGION = "site.detail"; export const DETAIL_PAGE_REGION = "site.detail";
const MIN_TTL_SEC = 86_400;
const MAX_ENTRIES = 5_000; const MAX_ENTRIES = 5_000;
interface StoredPage extends SiteDetailPage { interface StoredPage extends SiteDetailPage {
@@ -33,8 +33,9 @@ export async function saveDetailPage(
cacheSec: number, cacheSec: number,
): Promise<void> { ): Promise<void> {
const stored: StoredPage = { ...page, fetchedAt: rt.now().toISOString() }; const stored: StoredPage = { ...page, fetchedAt: rt.now().toISOString() };
const { siteDetailDays } = (await storageSettings(rt.ctx)).cacheTtl;
await sharedCache(rt.ctx) await sharedCache(rt.ctx)
.set(DETAIL_PAGE_REGION, srId, stored, { ttl: Math.max(cacheSec, MIN_TTL_SEC), maxsize: MAX_ENTRIES }) .set(DETAIL_PAGE_REGION, srId, stored, { ttl: Math.max(cacheSec, siteDetailDays * 86_400), maxsize: MAX_ENTRIES })
.catch((e) => rt.ctx.logger.warn({ err: { message: (e as Error).message } }, "详情页缓存写入失败")); .catch((e) => rt.ctx.logger.warn({ err: { message: (e as Error).message } }, "详情页缓存写入失败"));
const ids: DetailIds = { const ids: DetailIds = {
...(page.imdbId ? { imdbId: page.imdbId } : {}), ...(page.imdbId ? { imdbId: page.imdbId } : {}),
@@ -1,5 +1,6 @@
import type { DecisionOption, PoolCandidate } from "@umt/contracts"; import type { DecisionOption, PoolCandidate } from "@umt/contracts";
import { sql } from "@umt/db"; import { sql } from "@umt/db";
import { storageSettings } from "../../../shared-cache.ts";
import type { ServerContext } from "../../types.ts"; import type { ServerContext } from "../../types.ts";
import { jsonb } from "../rows.ts"; import { jsonb } from "../rows.ts";
import type { SourcesRuntime } from "../runtime.ts"; import type { SourcesRuntime } from "../runtime.ts";
@@ -10,7 +11,7 @@ import { closeDecisions, type Exec, openDecision } from "./common.ts";
// - 同一来源 + 同一归一化标题合并为一条事项(dedupeKey identity-group:<siteId>:<normalized>),成员 = 池里 // - 同一来源 + 同一归一化标题合并为一条事项(dedupeKey identity-group:<siteId>:<normalized>),成员 = 池里
// decision_id 指向它且仍 pending 的条目;事项的 subject 与候选选项取自代表资源(最近一次进来的那条); // decision_id 指向它且仍 pending 的条目;事项的 subject 与候选选项取自代表资源(最近一次进来的那条);
// - 成员被识别 / 指定 / 忽略时从组里摘除,组空了才关闭事项; // - 成员被识别 / 指定 / 忽略时从组里摘除,组空了才关闭事项;
// - unmatched 池条目超过 identity.poolArchiveDays 没有新证据就归档。 // - unmatched 池条目超过 storage.retentionDays.identityPoolArchive 天没有新证据就归档。
const OPEN = ["open", "investigating", "awaiting_user"] as const; const OPEN = ["open", "investigating", "awaiting_user"] as const;
/** 事项证据里最多列多少条成员资源(完整列表在待识别池按标题检索) */ /** 事项证据里最多列多少条成员资源(完整列表在待识别池按标题检索) */
@@ -208,14 +209,15 @@ const ARCHIVE_EVERY_MS = 3600_000;
let lastArchiveAt = 0; let lastArchiveAt = 0;
/** /**
* 过期归档:unmatched 池条目超过 identity.poolArchiveDays 没有新证据 → archived(仍可按状态筛出、可手动指定; * 过期归档:unmatched 池条目超过 storage.retentionDays.identityPoolArchive 天没有新证据 → archived(仍可按状态筛出、可手动指定;
* 同一资源再被观察到或新增相近别名时回到 pending)。每进程每小时最多执行一次。 * 同一资源再被观察到或新增相近别名时回到 pending)。每进程每小时最多执行一次。
*/ */
export async function archiveStalePool(rt: SourcesRuntime, opts: { force?: boolean } = {}): Promise<number> { export async function archiveStalePool(rt: SourcesRuntime, opts: { force?: boolean } = {}): Promise<number> {
const now = rt.now(); const now = rt.now();
if (!opts.force && now.getTime() - lastArchiveAt < ARCHIVE_EVERY_MS) return 0; if (!opts.force && now.getTime() - lastArchiveAt < ARCHIVE_EVERY_MS) return 0;
lastArchiveAt = now.getTime(); lastArchiveAt = now.getTime();
const { poolArchiveDays } = await rt.ctx.settings.get("identity"); // 与存储保留任务 storage.retention.identity_pool 同一条件(那边在 system 模块里按表直接执行)
const poolArchiveDays = (await storageSettings(rt.ctx)).retentionDays.identityPoolArchive;
const cutoff = new Date(now.getTime() - poolArchiveDays * 86_400_000); const cutoff = new Date(now.getTime() - poolArchiveDays * 86_400_000);
const r = await rt.ctx.db const r = await rt.ctx.db
.updateTable("identity_candidate") .updateTable("identity_candidate")
@@ -110,7 +110,8 @@ export async function saveRecognizeCache(
poster_path: info.posterPath ?? null, poster_path: info.posterPath ?? null,
backdrop_path: info.backdropPath ?? null, backdrop_path: info.backdropPath ?? null,
}; };
await cache.set(CACHE_REGIONS.tmdbRecognize, key, entry, { ttl: policy.ttl, maxsize: policy.maxsize }); // 识别结果 = 名称 → TMDB 条目,按搜索类 TTL(storage.cacheTtl.tmdbSearchDays)
await cache.set(CACHE_REGIONS.tmdbRecognize, key, entry, { ttl: policy.searchTtl, maxsize: policy.maxsize });
return; return;
} }
// 负识别缓存用独立短 TTL(EMPTY_RESULT_CACHE_TTL):故障期间的空结果不被固化整个有效期 // 负识别缓存用独立短 TTL(EMPTY_RESULT_CACHE_TTL):故障期间的空结果不被固化整个有效期
+9 -1
View File
@@ -28,6 +28,8 @@ export interface HealthDeps {
checkRoot?: typeof checkRoot; checkRoot?: typeof checkRoot;
/** 单个路径 IO 检查超时(ms),缺省 3000 */ /** 单个路径 IO 检查超时(ms),缺省 3000 */
ioTimeoutMs?: number; ioTimeoutMs?: number;
/** 存储段(CACHE-2);不给时该段为 unknown */
storage?: () => Promise<{ level: HealthLevel; alerts: string[] }>;
} }
const LEVEL_ORDER: HealthLevel[] = ["ok", "warn", "critical"]; const LEVEL_ORDER: HealthLevel[] = ["ok", "warn", "critical"];
@@ -384,7 +386,7 @@ export async function loadSchedules(db: Kysely<DB>, now: Date): Promise<SystemHe
export async function buildHealth(d: HealthDeps): Promise<SystemHealth> { export async function buildHealth(d: HealthDeps): Promise<SystemHealth> {
const now = d.now?.() ?? new Date(); const now = d.now?.() ?? new Date();
const [{ processes, retiredProcesses }, db, paths, engine, schedules] = await Promise.all([ const [{ processes, retiredProcesses }, db, paths, engine, schedules, storage] = await Promise.all([
loadProcesses(d.db, now).catch(() => ({ loadProcesses(d.db, now).catch(() => ({
processes: [] as SystemProcess[], processes: [] as SystemProcess[],
retiredProcesses: [] as SystemProcess[], retiredProcesses: [] as SystemProcess[],
@@ -408,6 +410,10 @@ export async function buildHealth(d: HealthDeps): Promise<SystemHealth> {
failing: [], failing: [],
failedRuns24h: 0, failedRuns24h: 0,
})), })),
(d.storage?.() ?? Promise.resolve({ level: "unknown" as HealthLevel, alerts: [] })).catch(() => ({
level: "unknown" as HealthLevel,
alerts: [] as string[],
})),
]); ]);
// SSE:存活的 api/all 进程上报的连接数合计;本进程若还没写过心跳,用实时值 // SSE:存活的 api/all 进程上报的连接数合计;本进程若还没写过心跳,用实时值
const selfCounted = processes.some((p) => !p.stale && p.instanceId === d.self.instanceId && p.sseConnections != null); const selfCounted = processes.some((p) => !p.stale && p.instanceId === d.self.instanceId && p.sseConnections != null);
@@ -423,6 +429,7 @@ export async function buildHealth(d: HealthDeps): Promise<SystemHealth> {
mountLevel, mountLevel,
engine.level, engine.level,
schedules.level, schedules.level,
storage.level,
]), ]),
checkedAt: now.toISOString(), checkedAt: now.toISOString(),
api: { api: {
@@ -439,6 +446,7 @@ export async function buildHealth(d: HealthDeps): Promise<SystemHealth> {
mounts: paths.mounts, mounts: paths.mounts,
engine, engine,
schedules, schedules,
storage,
}; };
} }
+4 -2
View File
@@ -130,11 +130,13 @@ export async function markHeartbeatStopped(db: Kysely<DB>, instanceId: string, a
/** 裁剪:超过 24 小时没心跳的进程行、超过 7 天的采样;返回删除行数 */ /** 裁剪:超过 24 小时没心跳的进程行、超过 7 天的采样;返回删除行数 */
export async function pruneHeartbeats( export async function pruneHeartbeats(
db: Kysely<DB>, db: Kysely<DB>,
o: { now?: Date; rowRetentionMs?: number; sampleDays?: number } = {}, o: { now?: Date; rowRetentionMs?: number; sampleDays?: number; sampleHours?: number } = {},
): Promise<{ rows: number; samples: number }> { ): Promise<{ rows: number; samples: number }> {
const now = o.now ?? new Date(); const now = o.now ?? new Date();
const rowCutoff = new Date(now.getTime() - (o.rowRetentionMs ?? HEARTBEAT_ROW_RETENTION_MS)); const rowCutoff = new Date(now.getTime() - (o.rowRetentionMs ?? HEARTBEAT_ROW_RETENTION_MS));
const sampleCutoff = new Date(now.getTime() - (o.sampleDays ?? SAMPLE_RETENTION_DAYS) * 24 * 3600_000); // CACHE-2:采样保留取 storage.retentionDays.heartbeatHours(调用方传 sampleHours);sampleDays 为旧参数
const sampleHours = o.sampleHours ?? (o.sampleDays ?? SAMPLE_RETENTION_DAYS) * 24;
const sampleCutoff = new Date(now.getTime() - sampleHours * 3600_000);
const rows = await db.deleteFrom("process_heartbeat").where("last_beat_at", "<", rowCutoff).executeTakeFirst(); const rows = await db.deleteFrom("process_heartbeat").where("last_beat_at", "<", rowCutoff).executeTakeFirst();
const samples = await db.deleteFrom("process_heartbeat_sample").where("at", "<", sampleCutoff).executeTakeFirst(); const samples = await db.deleteFrom("process_heartbeat_sample").where("at", "<", sampleCutoff).executeTakeFirst();
return { rows: Number(rows.numDeletedRows), samples: Number(samples.numDeletedRows) }; return { rows: Number(rows.numDeletedRows), samples: Number(samples.numDeletedRows) };
+106 -6
View File
@@ -1,9 +1,19 @@
import { createRoute, type RouteConfig } from "@hono/zod-openapi"; import { createRoute, type RouteConfig } from "@hono/zod-openapi";
import { Api, apiResponse, DEFAULT_SETTINGS, type EndpointDef } from "@umt/contracts"; import {
type Actor,
Api,
apiResponse,
DEFAULT_SETTINGS,
type EndpointDef,
type StorageCleanupTarget,
} from "@umt/contracts";
import { sql } from "@umt/db";
import type { z } from "zod"; import type { z } from "zod";
import { dataPaths } from "../../config.ts"; import { dataPaths } from "../../config.ts";
import { requireScope } from "../../http/auth.ts"; import { requireScope } from "../../http/auth.ts";
import { ApiException } from "../../http/errors.ts";
import { sseConnectionCount } from "../../http/sse.ts"; import { sseConnectionCount } from "../../http/sse.ts";
import { storageSettings } from "../../shared-cache.ts";
import { registerScheduledJob } from "../schedules/registry.ts"; import { registerScheduledJob } from "../schedules/registry.ts";
import type { ModuleApp, ServerContext, ServerModule } from "../types.ts"; import type { ModuleApp, ServerContext, ServerModule } from "../types.ts";
import { buildHealth, type HealthDeps, loadHistory } from "./health.ts"; import { buildHealth, type HealthDeps, loadHistory } from "./health.ts";
@@ -15,6 +25,8 @@ import {
safeHostname, safeHostname,
startHeartbeatWriter, startHeartbeatWriter,
} from "./heartbeat.ts"; } from "./heartbeat.ts";
import { type BatchOptions, CLEANUP_JOB_IDS, RETENTION_JOBS } from "./retention.ts";
import { buildStorage, type StoragePaths, storageHealth } from "./storage.ts";
export { HEARTBEAT_STALE_MS, HEARTBEAT_WRITE_MS, startHeartbeatWriter } from "./heartbeat.ts"; export { HEARTBEAT_STALE_MS, HEARTBEAT_WRITE_MS, startHeartbeatWriter } from "./heartbeat.ts";
@@ -23,7 +35,13 @@ export { HEARTBEAT_STALE_MS, HEARTBEAT_WRITE_MS, startHeartbeatWriter } from "./
* - GET /system/health:进程心跳 / 事件循环延迟 / SSE 连接 / DB / 磁盘 / 挂载根 / 引擎 / 定时任务,仅 admin; * - GET /system/health:进程心跳 / 事件循环延迟 / SSE 连接 / DB / 磁盘 / 挂载根 / 引擎 / 定时任务,仅 admin;
* - GET /system/health/history?hours=:心跳采样时间序列。 * - GET /system/health/history?hours=:心跳采样时间序列。
* 心跳写入:worker 进程在 worker/index.ts 里起写入器;纯 api 进程(role=api)在注册路由时起(role=all 由 worker 写 kind=all)。 * 心跳写入:worker 进程在 worker/index.ts 里起写入器;纯 api 进程(role=api)在注册路由时起(role=all 由 worker 写 kind=all)。
* 清理:定时任务 system.heartbeat_cleanup(维护组,1 小时)删 24 小时没心跳的进程行与 7 天前的采样。 * 清理:定时任务 system.heartbeat_cleanup(维护组,1 小时)删 24 小时没心跳的进程行与超过
* storage.retentionDays.heartbeatHours 的采样。
*
* CACHE-2 存储(docs/runbook/storage.md):
* - GET /system/storage:表体积 top 15、库体积、kv_cache 各 region、图片 / 备份 / 会话与证据对照预算、DATA_DIR 分目录、清理任务;
* - POST /system/storage/cleanup/{target}:把对应的维护任务排入立即执行(同 schedules run_now 语义);
* - 保留清理任务 storage.retention.* 与每周 storage.analyze(维护组,各自开关与周期在定时任务看板改)。
*/ */
export interface SystemModuleOptions { export interface SystemModuleOptions {
/** 引擎 schema(测试覆盖);缺省 openworkflow */ /** 引擎 schema(测试覆盖);缺省 openworkflow */
@@ -34,6 +52,12 @@ export interface SystemModuleOptions {
apiHeartbeat?: boolean; apiHeartbeat?: boolean;
/** 定时任务 system.heartbeat_cleanup 的默认周期,默认 1 小时;<=0 不登记 */ /** 定时任务 system.heartbeat_cleanup 的默认周期,默认 1 小时;<=0 不登记 */
cleanupIntervalMs?: number; cleanupIntervalMs?: number;
/** 存储面板路径(测试覆盖);缺省按 DATA_DIR / AGENT_SESSION_DIR */
storagePaths?: StoragePaths;
/** false 时不登记保留清理任务(测试);缺省 true */
retentionJobs?: boolean;
/** 保留清理的分批参数(测试) */
retentionBatch?: BatchOptions;
} }
type Infer<T> = T extends z.ZodType ? z.infer<T> : Record<string, never>; type Infer<T> = T extends z.ZodType ? z.infer<T> : Record<string, never>;
@@ -41,10 +65,11 @@ type Infer<T> = T extends z.ZodType ? z.infer<T> : Record<string, never>;
function route<D extends EndpointDef>( function route<D extends EndpointDef>(
app: ModuleApp, app: ModuleApp,
def: D, def: D,
handler: (i: { query: Infer<D["query"]> }) => Promise<unknown>, handler: (i: { query: Infer<D["query"]>; params: Infer<D["params"]>; actor: Actor | undefined }) => Promise<unknown>,
) { ) {
const request: RouteConfig["request"] = {}; const request: RouteConfig["request"] = {};
if (def.query) request.query = def.query as never; if (def.query) request.query = def.query as never;
if (def.params) request.params = def.params as never;
const r = createRoute({ const r = createRoute({
method: def.method, method: def.method,
path: def.path, path: def.path,
@@ -58,7 +83,13 @@ function route<D extends EndpointDef>(
}); });
// biome-ignore lint/suspicious/noExplicitAny: 路由泛型在封装里擦除,入参已由 zod-openapi 校验 // biome-ignore lint/suspicious/noExplicitAny: 路由泛型在封装里擦除,入参已由 zod-openapi 校验
app.openapi(r, (async (c: any) => { app.openapi(r, (async (c: any) => {
const data = def.response.parse(await handler({ query: def.query ? c.req.valid("query") : {} } as never)); const data = def.response.parse(
await handler({
query: def.query ? c.req.valid("query") : {},
params: def.params ? c.req.valid("param") : {},
actor: c.get("actor"),
} as never),
);
return c.json({ ok: true, data }, 200); return c.json({ ok: true, data }, 200);
// biome-ignore lint/suspicious/noExplicitAny: 同上 // biome-ignore lint/suspicious/noExplicitAny: 同上
}) as any); }) as any);
@@ -84,6 +115,11 @@ export function createSystemModule(opts: SystemModuleOptions = {}): ServerModule
} }
const selfKind = ctx.role === "all" ? "all" : ctx.role === "api" ? "api" : null; const selfKind = ctx.role === "all" ? "all" : ctx.role === "api" ? "api" : null;
const selfId = selfKind ? instanceIdOf(selfKind, safeHostname(), Deno.pid) : undefined; const selfId = selfKind ? instanceIdOf(selfKind, safeHostname(), Deno.pid) : undefined;
const paths = (): StoragePaths => {
if (opts.storagePaths) return opts.storagePaths;
const p = dataPaths(ctx.config);
return { dataDir: opts.healthDeps?.dataDir ?? p.dataDir, sessionDir: p.sessionDir };
};
route(app, Api.system.healthDetail, async () => { route(app, Api.system.healthDetail, async () => {
const cache = await ctx.settings.get("cache").catch(() => DEFAULT_SETTINGS.cache); const cache = await ctx.settings.get("cache").catch(() => DEFAULT_SETTINGS.cache);
@@ -98,25 +134,89 @@ export function createSystemModule(opts: SystemModuleOptions = {}): ServerModule
...(selfId ? { instanceId: selfId } : {}), ...(selfId ? { instanceId: selfId } : {}),
}, },
thresholds: { warnPercent: cache.resumePercent, criticalPercent: cache.hardThresholdPercent }, thresholds: { warnPercent: cache.resumePercent, criticalPercent: cache.hardThresholdPercent },
storage: async () =>
await storageHealth({ db: ctx.db, paths: paths(), settings: await storageSettings(ctx) }),
...(opts.healthDeps ?? {}), ...(opts.healthDeps ?? {}),
}); });
}); });
route(app, Api.system.healthHistory, async ({ query }) => await loadHistory(ctx.db, query.hours)); route(app, Api.system.healthHistory, async ({ query }) => await loadHistory(ctx.db, query.hours));
route(app, Api.system.storage, async () =>
buildStorage({
db: ctx.db,
paths: paths(),
settings: await storageSettings(ctx),
engineSchema: opts.engineSchema ?? "openworkflow",
}),
);
route(app, Api.system.storageCleanup, async ({ params, actor }) => {
const target = params.target as StorageCleanupTarget;
const jobId = CLEANUP_JOB_IDS[target];
const job = await ctx.db
.selectFrom("scheduled_job")
.select(["id", "available"])
.where("id", "=", jobId)
.executeTakeFirst();
if (!job) throw new ApiException("not_found", `清理任务 ${jobId} 还没登记(调度器未同步)`);
if (!job.available) throw new ApiException("conflict", "任务未配置,不能执行");
// 同 schedules 的立即执行:只置 run_requested_at,已在排队或运行中则合并
const r = await sql`update scheduled_job set run_requested_at = now()
where id = ${jobId} and run_requested_at is null and running_since is null`.execute(ctx.db);
const queued = Number(r.numAffectedRows ?? 0) > 0;
if (queued) {
await ctx.db
.insertInto("audit_event")
.values({
actor_type: actor?.type === "api_key" ? "api_key" : actor ? "user" : "system",
actor_id: actor?.id ?? null,
action: "storage.cleanup",
target_type: "scheduled_job",
target_id: jobId,
payload: JSON.stringify({ target }),
})
.execute()
.catch((err) => ctx.logger.warn({ err }, "审计写入失败"));
}
return { target, jobId, queued };
});
}, },
registerTasks() {}, registerTasks() {},
start(ctx) { start(ctx) {
if (opts.retentionJobs !== false) {
for (const j of RETENTION_JOBS) {
registerScheduledJob(ctx, {
key: j.jobId,
defaults: {
name: j.name,
description: j.description,
group: "maintain",
schedule: { type: "interval", everyMs: j.everyMs },
firstRun: "after_interval",
},
run: async ({ now }) => ({
summary: await j.run({
db: ctx.db,
engineSchema: opts.engineSchema ?? "openworkflow",
settings: await storageSettings(ctx),
now,
...(opts.retentionBatch ? { batch: opts.retentionBatch } : {}),
}),
}),
});
}
}
const everyMs = opts.cleanupIntervalMs ?? 3600_000; const everyMs = opts.cleanupIntervalMs ?? 3600_000;
if (everyMs <= 0) return; if (everyMs <= 0) return;
registerScheduledJob(ctx, { registerScheduledJob(ctx, {
key: "system.heartbeat_cleanup", key: "system.heartbeat_cleanup",
defaults: { defaults: {
name: "进程心跳清理", name: "进程心跳清理",
description: "删除 24 小时没有心跳的进程行和 7 天前的心跳采样", description: "删除 24 小时没有心跳的进程行和超过保留期(存储设置,缺省 24 小时)的心跳采样",
group: "maintain", group: "maintain",
schedule: { type: "interval", everyMs }, schedule: { type: "interval", everyMs },
}, },
run: async ({ now }) => { run: async ({ now }) => {
const r = await pruneHeartbeats(ctx.db, { now }); const sampleHours = (await storageSettings(ctx)).retentionDays.heartbeatHours;
const r = await pruneHeartbeats(ctx.db, { now, sampleHours });
return { summary: `删除 ${r.rows} 个进程行、${r.samples} 条采样` }; return { summary: `删除 ${r.rows} 个进程行、${r.samples} 条采样` };
}, },
}); });
+273
View File
@@ -0,0 +1,273 @@
import type { StorageCleanupTarget, StorageSettings } from "@umt/contracts";
import { type DB, type Kysely, type RawBuilder, sql } from "@umt/db";
// CACHE-2 保留策略清理(docs/runbook/storage.md):按表按时间列分批删除流水数据。
// - 每批:单独事务,`set local statement_timeout / lock_timeout` 限时,按 ctid 取一批再删,不做整表长事务;
// - 每次运行有总时限,到点就停,剩下的留给下次;批与批之间让出一小段时间给业务写入;
// - 只删已结束的数据(任务 / Agent 会话 / 能力调用看状态),进行中的不动。
// 待识别池不删:unmatched 条目到期归档(与 sources-identity archiveStalePool 同一条件)。
export interface BatchOptions {
/** 每批行数,缺省 5000 */
batchSize?: number;
/** 每批语句超时(ms),缺省 15000 */
statementTimeoutMs?: number;
/** 等锁超时(ms),缺省 3000 */
lockTimeoutMs?: number;
/** 本次运行总时限(ms),缺省 5 分钟 */
budgetMs?: number;
/** 批间停顿(ms),缺省 50 */
pauseMs?: number;
/** 测试注入 */
now?: () => number;
}
export interface BatchResult {
deleted: number;
batches: number;
/** 到总时限还没删完 */
truncated: boolean;
}
const sleep = (ms: number) => new Promise((r) => setTimeout(r, ms));
/**
* 分批删除 `table` 中满足 `where` 的行:每批 `delete … where ctid = any(array(select ctid … limit n))`,
* 单独事务 + 语句 / 锁超时。某批超时(57014)或等锁超时(55P03)时停止本次运行,不抛错。
*/
export async function deleteInBatches(
db: Kysely<DB>,
table: RawBuilder<unknown>,
where: RawBuilder<unknown>,
o: BatchOptions = {},
): Promise<BatchResult> {
const size = o.batchSize ?? 5000;
const now = o.now ?? Date.now;
const deadline = now() + (o.budgetMs ?? 5 * 60_000);
let deleted = 0;
let batches = 0;
for (;;) {
let n: number;
try {
n = await db.transaction().execute(async (trx) => {
await sql`select set_config('statement_timeout', ${String(o.statementTimeoutMs ?? 15_000)}, true),
set_config('lock_timeout', ${String(o.lockTimeoutMs ?? 3000)}, true)`.execute(trx);
const r = await sql`delete from ${table} where ctid = any(array(
select ctid from ${table} where ${where} limit ${size}))`.execute(trx);
return Number(r.numAffectedRows ?? 0);
});
} catch (e) {
const code = (e as { code?: string }).code;
if (code === "57014" || code === "55P03") return { deleted, batches, truncated: true };
throw e;
}
batches++;
deleted += n;
if (n < size) return { deleted, batches, truncated: false };
if (now() >= deadline) return { deleted, batches, truncated: true };
if ((o.pauseMs ?? 50) > 0) await sleep(o.pauseMs ?? 50);
}
}
export interface RetentionJob {
target: StorageCleanupTarget;
/** scheduled_job key */
jobId: string;
name: string;
description: string;
everyMs: number;
/** 保留期说明(存储面板) */
retention(s: StorageSettings): string | null;
run(c: RetentionRunContext): Promise<string>;
}
export interface RetentionRunContext {
db: Kysely<DB>;
engineSchema: string;
settings: StorageSettings;
now: Date;
batch?: BatchOptions;
}
const DAY_MS = 86_400_000;
const cutoffOf = (now: Date, days: number) => new Date(now.getTime() - days * DAY_MS);
const t = (name: string) => sql.raw(`"${name}"`);
const summaryOf = (label: string, r: BatchResult) =>
`${label}删除 ${r.deleted} 行(${r.batches} 批)${r.truncated ? ",未删完,下次继续" : ""}`;
const HOUR = 3600_000;
/**
* 保留清理任务(维护组;heartbeat 目标由 system.heartbeat_cleanup 承担,images / kv_cache 由 cache 模块承担)。
* 顺序即存储面板的展示顺序。
*/
export const RETENTION_JOBS: RetentionJob[] = [
{
target: "observation",
jobId: "storage.retention.observation",
name: "观察记录清理",
description: "删除超过保留期的 Feed 观察记录(observation,按 created_at)",
everyMs: 6 * HOUR,
retention: (s) => `${s.retentionDays.observation} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.observation);
const r = await deleteInBatches(c.db, t("observation"), sql`created_at < ${cutoff}`, c.batch);
return summaryOf("观察记录", r);
},
},
{
target: "sse_event",
jobId: "storage.retention.sse_event",
name: "SSE 事件清理",
description: "删除超过保留期的 SSE 补读事件(sse_event,按 at;发布方另按条数裁剪)",
everyMs: HOUR,
retention: (s) => `${s.retentionDays.sseEvent} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.sseEvent);
const r = await deleteInBatches(c.db, t("sse_event"), sql`at < ${cutoff}`, c.batch);
return summaryOf("SSE 事件", r);
},
},
{
target: "task_run",
jobId: "storage.retention.task_run",
name: "任务运行清理",
description: "删除超过保留期、已结束的业务任务(task)与引擎 run(含步骤、运行元数据)",
everyMs: 6 * HOUR,
retention: (s) => `${s.retentionDays.taskRun} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.taskRun);
const e = (n: string) => sql.raw(`"${c.engineSchema}"."${n}"`);
const task = await deleteInBatches(
c.db,
t("task"),
sql`created_at < ${cutoff} and status in ('succeeded', 'failed', 'cancelled', 'dead')`,
c.batch,
);
// 引擎 run:step_attempts 随 run 级联删除
const runs = await deleteInBatches(
c.db,
e("workflow_runs"),
sql`created_at < ${cutoff} and status in ('completed', 'succeeded', 'failed', 'canceled')`,
c.batch,
);
// 运行元数据与 run 同时写入:早于保留期的 run 已结束(上一步删掉了),元数据按时间直接删
const meta = await deleteInBatches(c.db, t("tasks_run_meta"), sql`created_at < ${cutoff}`, c.batch);
const facts = await deleteInBatches(c.db, t("tasks_fact"), sql`created_at < ${cutoff}`, c.batch);
const ledger = await deleteInBatches(
c.db,
t("tasks_op_ledger"),
sql`status = 'succeeded' and started_at < ${cutoff}`,
c.batch,
);
const truncated = [task, runs, meta, facts, ledger].some((x) => x.truncated);
return `业务任务 ${task.deleted}、引擎 run ${runs.deleted}、运行元数据 ${meta.deleted + facts.deleted + ledger.deleted} 行已删除${truncated ? ",未删完,下次继续" : ""}`;
},
},
{
target: "agent_event",
jobId: "storage.retention.agent_event",
name: "Agent 事件清理",
description: "删除超过保留期、会话已结束的 Agent 事件流与消息(agent_run_event / agent_run_message;会话本身保留)",
everyMs: 6 * HOUR,
retention: (s) => `${s.retentionDays.agentEvent} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.agentEvent);
const ended = sql`exists (select 1 from agent_run r where r.id = run_id
and r.status in ('succeeded', 'failed', 'cancelled', 'budget_exceeded'))`;
const ev = await deleteInBatches(c.db, t("agent_run_event"), sql`at < ${cutoff} and ${ended}`, c.batch);
const msg = await deleteInBatches(
c.db,
t("agent_run_message"),
sql`created_at < ${cutoff} and status = 'consumed' and ${ended}`,
c.batch,
);
return `Agent 事件 ${ev.deleted}、消息 ${msg.deleted} 行已删除${ev.truncated || msg.truncated ? ",未删完,下次继续" : ""}`;
},
},
{
target: "audit",
jobId: "storage.retention.audit",
name: "审计清理",
description: "删除超过保留期的审计事件(audit_event,能力调用审计按能力统计的保留期另算)",
everyMs: 24 * HOUR,
retention: (s) => `${s.retentionDays.audit} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.audit);
const r = await deleteInBatches(
c.db,
t("audit_event"),
sql`at < ${cutoff} and target_type is distinct from 'capability'`,
c.batch,
);
return summaryOf("审计事件", r);
},
},
{
target: "capability_stats",
jobId: "storage.retention.capability_stats",
name: "能力统计清理",
description: "删除超过保留期的能力调用审计(audit_event target_type=capability)与已结束的能力调用记录",
everyMs: 24 * HOUR,
retention: (s) => `${s.retentionDays.capabilityStats} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.capabilityStats);
const a = await deleteInBatches(
c.db,
t("audit_event"),
sql`at < ${cutoff} and target_type = 'capability'`,
c.batch,
);
const inv = await deleteInBatches(
c.db,
t("capability_invocation"),
sql`created_at < ${cutoff} and status <> 'running'`,
c.batch,
);
return `能力审计 ${a.deleted}、调用记录 ${inv.deleted} 行已删除${a.truncated || inv.truncated ? ",未删完,下次继续" : ""}`;
},
},
{
target: "identity_pool",
jobId: "storage.retention.identity_pool",
name: "待识别池归档",
description: "待识别池里 unmatched 条目超过保留期没有新证据就归档(不删除,仍可按状态筛出)",
everyMs: 6 * HOUR,
retention: (s) => `${s.retentionDays.identityPoolArchive} 天`,
async run(c) {
const cutoff = cutoffOf(c.now, c.settings.retentionDays.identityPoolArchive);
let archived = 0;
const size = c.batch?.batchSize ?? 5000;
for (;;) {
const r = await sql`update identity_candidate set status = 'archived', next_review_at = null
where id in (select id from identity_candidate where status = 'pending' and reason = 'unmatched'
and coalesce(last_attempt_at, created_at) < ${cutoff} limit ${size})`.execute(c.db);
const n = Number(r.numAffectedRows ?? 0);
archived += n;
if (n < size) break;
}
return `归档 ${archived} 条待识别条目`;
},
},
{
target: "analyze",
jobId: "storage.analyze",
name: "统计信息更新",
description: "对全库执行 ANALYZE,刷新查询规划用的统计信息(每周)",
everyMs: 7 * 24 * HOUR,
retention: () => null,
async run(c) {
const started = Date.now();
await sql`analyze`.execute(c.db);
return `ANALYZE 完成,用时 ${Math.round((Date.now() - started) / 100) / 10} 秒`;
},
},
];
/** 清理目标 → 定时任务 key(heartbeat / images / kv_cache 由其它任务承担) */
export const CLEANUP_JOB_IDS: Record<StorageCleanupTarget, string> = {
...(Object.fromEntries(RETENTION_JOBS.map((j) => [j.target, j.jobId])) as Record<string, string>),
heartbeat: "system.heartbeat_cleanup",
images: "cache.images_lru",
kv_cache: "cache.sweep",
} as Record<StorageCleanupTarget, string>;
+347
View File
@@ -0,0 +1,347 @@
import { promises as fs } from "node:fs";
import { join, relative } from "node:path";
import type { HealthLevel, StorageBudgetUsage, StorageSettings, SystemStorage } from "@umt/contracts";
import { type DB, type Kysely, sql } from "@umt/db";
import { worst } from "./health.ts";
import { CLEANUP_JOB_IDS, RETENTION_JOBS } from "./retention.ts";
// CACHE-2 存储面板(GET /system/storage)与 health 的存储档位。
// 目录遍历可能很慢(图片缓存几十万个文件),结果按目录集合缓存 5 分钟;health 只用缓存值,过期时后台刷新,不等。
const GB = 1024 ** 3;
const DIR_TTL_MS = 5 * 60_000;
export interface StoragePaths {
dataDir: string;
/** Agent 会话目录(可能在 DATA_DIR 之外) */
sessionDir: string;
/** 图片缓存目录;缺省 DATA_DIR/cache/images */
imagesDir?: string;
}
export interface DirUsage {
name: string;
path: string;
usedBytes: number;
files: number;
error: string | null;
}
interface DirScan {
at: Date;
dirs: DirUsage[];
images: DirUsage;
sessions: DirUsage;
evidence: DirUsage;
backups: DirUsage & { count: number };
}
async function walk(path: string): Promise<{ bytes: number; files: number; error: string | null }> {
let bytes = 0;
let files = 0;
const go = async (d: string, top: boolean): Promise<string | null> => {
let names: string[];
try {
names = await fs.readdir(d);
} catch (e) {
const code = (e as { code?: string }).code;
// 目录不存在按 0 计
return top && code !== "ENOENT" ? (code ?? (e as Error).message) : null;
}
for (const n of names) {
const p = join(d, n);
const st = await fs.lstat(p).catch(() => null);
if (!st) continue;
if (st.isDirectory()) await go(p, false);
else {
bytes += st.size;
files++;
}
}
return null;
};
const error = await go(path, true);
return { bytes, files, error };
}
const usageOf = async (name: string, path: string): Promise<DirUsage> => {
const w = await walk(path);
return { name, path, usedBytes: w.bytes, files: w.files, error: w.error };
};
/** DATA_DIR 一级目录(cache / tmp 再细分一层)+ 会话目录 + 备份份数 */
export async function scanDirs(p: StoragePaths): Promise<DirScan> {
const dirs: DirUsage[] = [];
const top = await fs.readdir(p.dataDir, { withFileTypes: true }).catch(() => []);
for (const e of top.sort((a, b) => a.name.localeCompare(b.name))) {
if (!e.isDirectory()) continue;
const path = join(p.dataDir, e.name);
if (e.name === "cache" || e.name === "tmp") {
const sub = await fs.readdir(path, { withFileTypes: true }).catch(() => []);
for (const s of sub.sort((a, b) => a.name.localeCompare(b.name))) {
if (s.isDirectory()) dirs.push(await usageOf(`${e.name}/${s.name}`, join(path, s.name)));
}
continue;
}
dirs.push(await usageOf(e.name, path));
}
const find = async (path: string, name: string) => dirs.find((d) => d.path === path) ?? (await usageOf(name, path));
const imagesDir = p.imagesDir ?? join(p.dataDir, "cache", "images");
const backupsDir = join(p.dataDir, "backups");
const backupNames = await fs.readdir(backupsDir, { withFileTypes: true }).catch(() => []);
const sessionsInData = !relative(p.dataDir, p.sessionDir).startsWith("..");
const sessions = await find(p.sessionDir, "agent-sessions");
if (!sessionsInData) dirs.push(sessions);
return {
at: new Date(),
dirs,
images: await find(imagesDir, "cache/images"),
sessions,
evidence: await find(join(p.dataDir, "tmp", "evidence"), "tmp/evidence"),
backups: {
...(await find(backupsDir, "backups")),
count: backupNames.filter((b) => b.isDirectory() && !b.name.startsWith(".")).length,
},
};
}
const scans = new Map<string, { at: number; value: Promise<DirScan>; last: DirScan | null }>();
/**
* 目录用量(缓存 5 分钟)。wait=false 时不等遍历:有旧值返回旧值(过期则后台刷新),没有返回 null。
*/
export async function dirScan(p: StoragePaths, o: { wait: boolean; force?: boolean }): Promise<DirScan | null> {
const key = `${p.dataDir}\u0000${p.sessionDir}\u0000${p.imagesDir ?? ""}`;
const cur = scans.get(key);
if (!cur || o.force || Date.now() - cur.at > DIR_TTL_MS) {
const value = scanDirs(p);
const entry = { at: Date.now(), value, last: cur?.last ?? null };
scans.set(key, entry);
value.then(
(v) => {
entry.last = v;
},
() => {
if (scans.get(key) === entry) scans.delete(key);
},
);
if (o.wait) return await value;
return entry.last;
}
if (o.wait) return await cur.value;
return cur.last;
}
/** 测试用 */
export function resetDirScans(): void {
scans.clear();
}
/** 用量对照预算:≥ budget critical,≥ warn warn */
export function budgetLevel(used: number | null, budget: number | null, warn: number | null): HealthLevel {
if (used === null) return "unknown";
if (budget !== null && used >= budget) return "critical";
if (warn !== null && used >= warn) return "warn";
return "ok";
}
const gb = (v: number) => Math.round(v * GB);
function usage(used: number | null, budgetGb: number, warnGb: number, error: string | null = null): StorageBudgetUsage {
const budgetBytes = gb(budgetGb);
const warnBytes = gb(warnGb);
return {
usedBytes: used,
budgetBytes,
warnBytes,
level: error ? "unknown" : budgetLevel(used, budgetBytes, warnBytes),
error,
};
}
/**
* 图片缓存有容量 LRU(到预算就按最久未访问删),接近预算是常态:超过预算 10% 才告警(说明 LRU 没跟上),不给 critical。
*/
function imagesUsage(used: number | null, budgetGb: number, error: string | null): StorageBudgetUsage {
const budgetBytes = gb(budgetGb);
const warnBytes = Math.round(budgetBytes * 1.1);
return {
usedBytes: used,
budgetBytes,
warnBytes,
level: error ? "unknown" : used === null ? "unknown" : used >= warnBytes ? "warn" : "ok",
error,
};
}
const fmtGb = (b: number | null) => (b === null ? "?" : `${Math.round((b / GB) * 10) / 10} GB`);
export interface StorageDeps {
db: Kysely<DB>;
paths: StoragePaths;
settings: StorageSettings;
}
async function dbSize(db: Kysely<DB>): Promise<number> {
const r = await sql<{ n: string }>`select pg_database_size(current_database())::text as n`.execute(db);
return Number(r.rows[0]?.n ?? 0);
}
/** 汇总档位 + 告警文案(health 与存储面板共用) */
function levelsOf(
s: StorageSettings,
dbBytes: number | null,
scan: DirScan | null,
): {
db: StorageBudgetUsage;
images: StorageBudgetUsage;
backups: StorageBudgetUsage;
sessionsEvidence: StorageBudgetUsage;
level: HealthLevel;
alerts: string[];
} {
const b = s.budgets;
const db = usage(dbBytes, b.pgDataGb, b.pgDataWarnGb, dbBytes === null ? "db_unavailable" : null);
const images = imagesUsage(scan ? scan.images.usedBytes : null, b.imagesGb, scan?.images.error ?? null);
const backups = usage(
scan ? scan.backups.usedBytes : null,
b.backupsGb,
b.backupsGb * 0.8,
scan?.backups.error ?? null,
);
const se = scan ? scan.sessions.usedBytes + scan.evidence.usedBytes : null;
const sessionsEvidence = usage(
se,
b.sessionsEvidenceGb,
b.sessionsEvidenceGb * 0.8,
scan?.sessions.error ?? scan?.evidence.error ?? null,
);
const alerts: string[] = [];
const say = (label: string, u: StorageBudgetUsage) => {
if (u.level === "warn" || u.level === "critical") {
alerts.push(
`${label} ${fmtGb(u.usedBytes)},${u.level === "critical" ? "已超过预算" : "超过告警线"} ${fmtGb(u.level === "critical" ? u.budgetBytes : u.warnBytes)}`,
);
}
};
say("数据库", db);
say("图片缓存", images);
say("备份", backups);
say("会话与证据", sessionsEvidence);
return {
db,
images,
backups,
sessionsEvidence,
level: worst([db.level, images.level, backups.level, sessionsEvidence.level]),
alerts,
};
}
/** health 的存储段:库体积实时取,目录用量只用缓存(不等遍历) */
export async function storageHealth(d: StorageDeps): Promise<{ level: HealthLevel; alerts: string[] }> {
const [bytes, scan] = await Promise.all([
dbSize(d.db).catch(() => null),
dirScan(d.paths, { wait: false }).catch(() => null),
]);
const l = levelsOf(d.settings, bytes, scan);
return { level: l.level, alerts: l.alerts };
}
export async function buildStorage(
d: StorageDeps & { engineSchema: string; refresh?: boolean },
): Promise<SystemStorage> {
const [bytes, scan, tables, kv, jobs] = await Promise.all([
dbSize(d.db).catch(() => null),
dirScan(d.paths, { wait: true, force: d.refresh ?? false }).catch(() => null),
sql<{
schema: string;
name: string;
total: string;
tbl: string;
idx: string;
rows: number | null;
dead: number | null;
vac: Date | null;
ana: Date | null;
}>`select n.nspname as schema, c.relname as name,
pg_total_relation_size(c.oid)::text as total, pg_relation_size(c.oid)::text as tbl,
pg_indexes_size(c.oid)::text as idx,
case when c.reltuples < 0 then null else c.reltuples::bigint end as rows,
s.n_dead_tup as dead, s.last_autovacuum as vac,
greatest(s.last_analyze, s.last_autoanalyze) as ana
from pg_class c join pg_namespace n on n.oid = c.relnamespace
left join pg_stat_all_tables s on s.relid = c.oid
where c.relkind in ('r', 'm', 'p') and n.nspname not in ('pg_catalog', 'information_schema')
and n.nspname not like 'pg_toast%'
order by pg_total_relation_size(c.oid) desc limit 15`.execute(d.db),
sql<{ region: string; entries: number; expired: number; bytes: string }>`
select region, count(*)::int as entries, count(*) filter (where expires_at <= now())::int as expired,
coalesce(sum(pg_column_size(value) + pg_column_size(key)), 0)::text as bytes
from kv_cache group by region order by region`
.execute(d.db)
.catch(() => ({ rows: [] })),
d.db
.selectFrom("scheduled_job")
.select(["id", "enabled", "last_run_at", "last_status", "next_run_at"])
.where("id", "in", Object.values(CLEANUP_JOB_IDS))
.execute()
.catch(() => []),
]);
const lastSummaries = new Map<string, string | null>();
if (jobs.length) {
const r = await sql<{ job_id: string; summary: string | null; error: string | null }>`
select distinct on (job_id) job_id, summary, error from scheduled_job_run
where job_id in (${sql.join(jobs.map((j) => j.id))}) and status <> 'running'
order by job_id, started_at desc`
.execute(d.db)
.catch(() => ({ rows: [] as { job_id: string; summary: string | null; error: string | null }[] }));
for (const x of r.rows) lastSummaries.set(x.job_id, x.summary ?? x.error);
}
const l = levelsOf(d.settings, bytes, scan);
const s = d.settings;
const retentionOf: Record<string, string | null> = {
...Object.fromEntries(RETENTION_JOBS.map((j) => [j.target, j.retention(s)])),
heartbeat: `${s.retentionDays.heartbeatHours} 小时`,
images: `容量 ${s.budgets.imagesGb} GB`,
kv_cache: null,
};
const iso = (v: Date | string | null) => (v ? new Date(v).toISOString() : null);
return {
level: l.level,
checkedAt: new Date().toISOString(),
dirsScannedAt: scan ? scan.at.toISOString() : null,
db: {
...l.db,
tables: tables.rows.map((t) => ({
schema: t.schema,
name: t.name,
totalBytes: Number(t.total),
tableBytes: Number(t.tbl),
indexBytes: Number(t.idx),
rowsEstimate: t.rows === null ? null : Number(t.rows),
deadRows: t.dead === null ? null : Number(t.dead),
lastAutovacuumAt: iso(t.vac),
lastAnalyzeAt: iso(t.ana),
})),
},
kvCache: kv.rows.map((r) => ({ region: r.region, entries: r.entries, expired: r.expired, bytes: Number(r.bytes) })),
images: { ...l.images, files: scan ? scan.images.files : null },
backups: { ...l.backups, count: scan ? scan.backups.count : null },
sessionsEvidence: l.sessionsEvidence,
dataDir: scan?.dirs ?? [],
cleanup: (Object.entries(CLEANUP_JOB_IDS) as [keyof typeof CLEANUP_JOB_IDS, string][]).map(([target, jobId]) => {
const j = jobs.find((x) => x.id === jobId);
return {
target,
jobId,
retention: retentionOf[target] ?? null,
enabled: j ? j.enabled : null,
lastRunAt: iso(j?.last_run_at ?? null),
lastStatus: j?.last_status ?? null,
lastSummary: lastSummaries.get(jobId) ?? null,
nextRunAt: iso(j?.next_run_at ?? null),
};
}),
};
}
+3 -4
View File
@@ -17,10 +17,11 @@ import {
type WebSearchTestResult, type WebSearchTestResult,
} from "@umt/contracts"; } from "@umt/contracts";
import { cachedCall, pgCacheFor, pgTokenBucketFor } from "@umt/db"; import { cachedCall, pgCacheFor, pgTokenBucketFor } from "@umt/db";
import { storageSettings } from "../../shared-cache.ts";
import type { ServerContext } from "../types.ts"; import type { ServerContext } from "../types.ts";
/** /**
* WEB-1 网页搜索服务:Brave 客户端之上的 Key 读取、每分钟上限(跨进程令牌桶)、10 分钟 PG 缓存(kv_cache region websearch)、 * WEB-1 网页搜索服务:Brave 客户端之上的 Key 读取、每分钟上限(跨进程令牌桶)、PG 缓存(kv_cache region websearch,TTL = storage.cacheTtl.webSearchDays)、
* 以及 web.fetch 的白名单——按调用方(Agent 会话 / 其它调用者)记录最近一次 web.search 返回的 URL,只允许读取这些 URL。 * 以及 web.fetch 的白名单——按调用方(Agent 会话 / 其它调用者)记录最近一次 web.search 返回的 URL,只允许读取这些 URL。
*/ */
@@ -36,8 +37,6 @@ export interface WebSearchModuleOptions {
export const WEBSEARCH_CACHE_REGION = "websearch"; export const WEBSEARCH_CACHE_REGION = "websearch";
/** 白名单:key = 调用方,value = 最近一次搜索的 URL */ /** 白名单:key = 调用方,value = 最近一次搜索的 URL */
export const WEBSEARCH_ALLOW_REGION = "websearch.allow"; export const WEBSEARCH_ALLOW_REGION = "websearch.allow";
/** 搜索结果缓存 10 分钟 */
export const SEARCH_CACHE_TTL_SEC = 600;
const SEARCH_CACHE_MAX = 1000; const SEARCH_CACHE_MAX = 1000;
/** 白名单保留 6 小时(会话时长上限远小于此) */ /** 白名单保留 6 小时(会话时长上限远小于此) */
const ALLOW_TTL_SEC = 6 * 3600; const ALLOW_TTL_SEC = 6 * 3600;
@@ -142,7 +141,7 @@ export class WebSearchService {
key, key,
{ {
region: WEBSEARCH_CACHE_REGION, region: WEBSEARCH_CACHE_REGION,
ttl: SEARCH_CACHE_TTL_SEC, ttl: (await storageSettings(this.ctx)).cacheTtl.webSearchDays * 86_400,
maxsize: SEARCH_CACHE_MAX, maxsize: SEARCH_CACHE_MAX,
...(o.fresh ? { fresh: true } : {}), ...(o.fresh ? { fresh: true } : {}),
}, },
+74 -9
View File
@@ -1,9 +1,10 @@
import { DEFAULT_SETTINGS } from "@umt/contracts"; import { DEFAULT_SETTINGS, type StorageSettings, withStorageDefaults } from "@umt/contracts";
import { type PgCache, pgCacheFor } from "@umt/db"; import { type PgCache, pgCacheFor, sql } from "@umt/db";
import type { ServerContext } from "./modules/types.ts"; import type { ServerContext } from "./modules/types.ts";
// 进程级共享缓存入口:PG kv_cache(移植 MP app/runtime/cache.py 的 TTLCache,存储换成 PG,三进程共享)。 // 进程级共享缓存入口:PG kv_cache(移植 MP app/runtime/cache.py 的 TTLCache,存储换成 PG,三进程共享)。
// region 与 TTL 规则见 docs/runbook/cache.md 与 docs/porting/moviepilot-tmdb-cache.md。 // region 与 TTL 规则见 docs/runbook/cache.md 与 docs/porting/moviepilot-tmdb-cache.md;
// CACHE-2 起各 region 的 TTL 取值来源是 settings.storage.cacheTtl(docs/runbook/storage.md),语义不变。
export const CACHE_REGIONS = { export const CACHE_REGIONS = {
/** TMDB 请求响应(MP TMDb.request 的 @cached):key = path?query */ /** TMDB 请求响应(MP TMDb.request 的 @cached):key = path?query */
@@ -12,14 +13,27 @@ export const CACHE_REGIONS = {
tmdbRecognize: "tmdb.recognize", tmdbRecognize: "tmdb.recognize",
} as const; } as const;
const DAY = 86_400;
export function sharedCache(ctx: Pick<ServerContext, "db">): PgCache { export function sharedCache(ctx: Pick<ServerContext, "db">): PgCache {
return pgCacheFor(ctx.db); return pgCacheFor(ctx.db);
} }
/** TMDB 缓存策略(MP:CONF.meta = META_CACHE_EXPIRE 小时、CONF.tmdb 条数、EMPTY_RESULT_CACHE_TTL 秒) */ /** settings.storage(读失败用缺省) */
export async function storageSettings(ctx: Pick<ServerContext, "settings">): Promise<StorageSettings> {
return withStorageDefaults(await ctx.settings.get("storage").catch(() => undefined));
}
/**
* TMDB 缓存策略(MP:CONF.meta = META_CACHE_EXPIRE、CONF.tmdb 条数、EMPTY_RESULT_CACHE_TTL)。
* CACHE-2:MP 只有一个 CONF.meta,这里按请求类别取 storage.cacheTtl:搜索 / find 与识别结果用 searchTtl,
* 详情(/movie/{id}、/tv/{id}、季)用 ttl(= tmdbDetailDays),空结果与未识别负缓存用 emptyTtl。
*/
export interface TmdbCachePolicy { export interface TmdbCachePolicy {
/** 秒 */ /** 详情 TTL(秒) */
ttl: number; ttl: number;
/** 搜索 / find / 识别结果 TTL(秒) */
searchTtl: number;
maxsize: number; maxsize: number;
/** 秒 */ /** 秒 */
emptyTtl: number; emptyTtl: number;
@@ -27,10 +41,61 @@ export interface TmdbCachePolicy {
export async function tmdbCachePolicy(ctx: Pick<ServerContext, "settings">): Promise<TmdbCachePolicy> { export async function tmdbCachePolicy(ctx: Pick<ServerContext, "settings">): Promise<TmdbCachePolicy> {
const c = await ctx.settings.get("cache").catch(() => DEFAULT_SETTINGS.cache); const c = await ctx.settings.get("cache").catch(() => DEFAULT_SETTINGS.cache);
const d = DEFAULT_SETTINGS.cache; const t = (await storageSettings(ctx)).cacheTtl;
return { return {
ttl: (c.tmdbTtlHours ?? d.tmdbTtlHours) * 3600, ttl: t.tmdbDetailDays * DAY,
maxsize: c.tmdbMaxSize ?? d.tmdbMaxSize, searchTtl: t.tmdbSearchDays * DAY,
emptyTtl: c.emptyResultTtlSec ?? d.emptyResultTtlSec, maxsize: c.tmdbMaxSize ?? DEFAULT_SETTINGS.cache.tmdbMaxSize,
emptyTtl: t.tmdbMissDays * DAY,
}; };
} }
/** TMDB 请求键是否是搜索类(按 searchTtl 缓存) */
export function isTmdbSearchPath(key: string): boolean {
return /^\/(search|find)\//.test(key);
}
/** 详情键 → `movie:<id>` / `tv:<id>`;不是详情返回 null */
export function tmdbDetailRef(key: string): string | null {
const m = /^\/(movie|tv)\/(\d+)(?:[/?]|$)/.exec(key);
return m ? `${m[1]}:${m[2]}` : null;
}
// ── 追踪中的作品:详情缓存命中时续期 ──
// 追踪 = 作品有未结束的订阅(active / paused / pending)且有已接受的 TMDB 身份。集合按进程缓存 5 分钟。
const TRACKED_TTL_MS = 5 * 60_000;
const trackedCache = new WeakMap<object, { at: number; set: Promise<Set<string>> }>();
export async function trackedTmdbRefs(ctx: Pick<ServerContext, "db">, now = Date.now()): Promise<Set<string>> {
const hit = trackedCache.get(ctx.db);
if (hit && now - hit.at < TRACKED_TTL_MS) return hit.set;
const set = sql<{ ref: string }>`select distinct x.entity_type || ':' || x.external_id as ref
from subscription s join external_identity x on x.media_entity_id = s.media_entity_id
where s.status in ('active', 'paused', 'pending') and x.provider = 'tmdb' and x.status = 'accepted'
and x.entity_type in ('movie', 'tv')`
.execute(ctx.db)
.then((r) => new Set(r.rows.map((x) => x.ref)));
trackedCache.set(ctx.db, { at: now, set });
set.catch(() => trackedCache.delete(ctx.db));
return set;
}
/** 测试用:清掉追踪集合缓存 */
export function resetTrackedTmdbRefs(ctx: Pick<ServerContext, "db">): void {
trackedCache.delete(ctx.db);
}
/**
* TMDB 详情缓存命中后的续期:键属于追踪中的作品时把过期时间推到 now + 详情 TTL(只延长不缩短)。
* 返回是否续期。失败不影响命中结果。
*/
export async function renewTrackedTmdbDetail(
ctx: Pick<ServerContext, "db">,
key: string,
detailTtlSec: number,
): Promise<boolean> {
const ref = tmdbDetailRef(key);
if (!ref || !(await trackedTmdbRefs(ctx)).has(ref)) return false;
return await sharedCache(ctx).renew(CACHE_REGIONS.tmdbRequest, key, detailTtlSec);
}
+112 -16
View File
@@ -1,14 +1,14 @@
import { mkdtempSync, rmSync, utimesSync } from "node:fs"; import { existsSync, mkdtempSync, rmSync, statSync, utimesSync } from "node:fs";
import { tmpdir } from "node:os"; import { tmpdir } from "node:os";
import { join } from "node:path"; import { join } from "node:path";
import { PgCache, sql } from "@umt/db"; import { PgCache, sql } from "@umt/db";
import { createMockTmdb, type MockTmdb } from "@umt/testkit"; import { createMockTmdb, type MockTmdb } from "@umt/testkit";
import { afterAll, beforeAll, describe, expect, it } from "vitest"; import { afterAll, beforeAll, describe, expect, it } from "vitest";
import { imageCachePath } from "../../src/modules/cache/images.ts"; import { ImageCache, imageCachePath } from "../../src/modules/cache/images.ts";
import { createCacheModule } from "../../src/modules/cache/index.ts"; import { createCacheModule } from "../../src/modules/cache/index.ts";
import { scheduleRegistry } from "../../src/modules/schedules/registry.ts"; import { scheduleRegistry } from "../../src/modules/schedules/registry.ts";
import { SourcesRuntime } from "../../src/modules/sources-identity/runtime.ts"; import { SourcesRuntime } from "../../src/modules/sources-identity/runtime.ts";
import { sharedCache } from "../../src/shared-cache.ts"; import { resetTrackedTmdbRefs, sharedCache } from "../../src/shared-cache.ts";
import { bearer, createHarness, type Harness, pgAvailable } from "../auth-settings/harness.ts"; import { bearer, createHarness, type Harness, pgAvailable } from "../auth-settings/harness.ts";
import { DEV_KEY, json } from "../helpers.ts"; import { DEV_KEY, json } from "../helpers.ts";
@@ -53,31 +53,77 @@ describe.skipIf(!hasPg)("C2-A 缓存(真实 PG)", () => {
rmSync(dir, { recursive: true, force: true }); rmSync(dir, { recursive: true, force: true });
}); });
it("TMDB 请求响应进 kv_cache(tmdb.request,TTL = cache.tmdbTtlHours,条数上限 = cache.tmdbMaxSize),另一个进程直接命中", async () => { it("TMDB 请求响应进 kv_cache(tmdb.request,TTL 取 storage.cacheTtl:详情 / 搜索 / 未命中,条数上限 = cache.tmdbMaxSize),另一个进程直接命中", async () => {
const r = await admin("PATCH", "/settings", { const r = await admin("PATCH", "/settings", {
metadata: { tmdbApiKey: "0123456789abcdef0123456789abcdef", tmdbBaseUrl: `${tmdb.url}/3` }, metadata: { tmdbApiKey: "0123456789abcdef0123456789abcdef", tmdbBaseUrl: `${tmdb.url}/3` },
cache: { tmdbTtlHours: 6, tmdbMaxSize: 16 }, cache: { tmdbMaxSize: 16 },
storage: { cacheTtl: { tmdbDetailDays: 20, tmdbSearchDays: 3, tmdbMissDays: 2 } },
}); });
expect(r.status).toBe(200); expect(r.status).toBe(200);
// 部分更新只改给出的子字段,其余保持缺省
const st = (await json(await admin("GET", "/settings"))).data.storage;
expect(st.cacheTtl).toMatchObject({ tmdbDetailDays: 20, tmdbSearchDays: 3, tmdbMissDays: 2, siteDetailDays: 7 });
expect(st.budgets.imagesGb).toBe(20);
await (await new SourcesRuntime(h.ctx).tmdb())!.details("movie", 872585); await (await new SourcesRuntime(h.ctx).tmdb())!.details("movie", 872585);
const calls = tmdb.calls.length; const calls = tmdb.calls.length;
const rows = ( const rows = (
await sql<{ key: string; hours: number }>`select key, extract(epoch from expires_at - now()) / 3600 as hours await sql<{ key: string; days: number }>`select key, extract(epoch from expires_at - now()) / 86400 as days
from kv_cache where region = 'tmdb.request'`.execute(h.db) from kv_cache where region = 'tmdb.request'`.execute(h.db)
).rows; ).rows;
const key = rows.find((x) => x.key.startsWith("/movie/872585?"))!.key; const detail = rows.find((x) => x.key.startsWith("/movie/872585?"))!;
expect(rows.every((x) => !x.key.includes("0123456789abcdef"))).toBe(true); expect(rows.every((x) => !x.key.includes("0123456789abcdef"))).toBe(true);
expect(Number(rows[0]!.hours)).toBeGreaterThan(5.9); expect(Number(detail.days)).toBeGreaterThan(19.9);
expect(await new PgCache(h.db).get("tmdb.request", key)).toMatchObject({ id: 872585 }); expect(Number(detail.days)).toBeLessThanOrEqual(20);
expect(await new PgCache(h.db).get("tmdb.request", detail.key)).toMatchObject({ id: 872585 });
await (await new SourcesRuntime(h.ctx).tmdb())!.details("movie", 872585); await (await new SourcesRuntime(h.ctx).tmdb())!.details("movie", 872585);
expect(tmdb.calls.length).toBe(calls); expect(tmdb.calls.length).toBe(calls);
// 空搜索结果按 EMPTY_RESULT_CACHE_TTL(30 分钟) const days = async (like: string) =>
Number(
(
await sql<{ d: number }>`select extract(epoch from expires_at - now()) / 86400 as d
from kv_cache where region = 'tmdb.request' and key like ${like}`.execute(h.db)
).rows[0]!.d,
);
// 有结果的搜索按 tmdbSearchDays
await (await new SourcesRuntime(h.ctx).tmdb())!.search("Oppenheimer", { type: "movie" });
const hit = await days("/search/movie?%Oppenheimer%");
expect(hit).toBeGreaterThan(2.9);
expect(hit).toBeLessThanOrEqual(3);
// 空搜索结果按 EMPTY_RESULT_CACHE_TTL 语义,取 tmdbMissDays
await (await new SourcesRuntime(h.ctx).tmdb())!.search("完全不存在的片名c2a", { type: "movie" }); await (await new SourcesRuntime(h.ctx).tmdb())!.search("完全不存在的片名c2a", { type: "movie" });
const empty = ( const empty = await days("/search/movie?%c2a%");
await sql<{ m: number }>`select extract(epoch from expires_at - now()) / 60 as m expect(empty).toBeGreaterThan(1.9);
from kv_cache where region = 'tmdb.request' and key like '/search/movie?%'`.execute(h.db) expect(empty).toBeLessThanOrEqual(2);
).rows[0]; });
expect(Number(empty!.m)).toBeLessThanOrEqual(30);
it("追踪中的作品:详情缓存命中时续期到 now + 详情 TTL;未追踪的不动", async () => {
const ent = (
await sql<{
id: string;
}>`insert into media_entity (kind, title) values ('movie', '奥本海默') returning id`.execute(h.db)
).rows[0]!.id;
await sql`insert into external_identity (media_entity_id, provider, entity_type, external_id, status)
values (${ent}, 'tmdb', 'movie', '872585', 'accepted')`.execute(h.db);
await sql`insert into subscription (media_entity_id, media_kind, status) values (${ent}, 'movie', 'active')`.execute(
h.db,
);
resetTrackedTmdbRefs(h.ctx);
const client = (await new SourcesRuntime(h.ctx).tmdb())!;
await client.details("movie", 872585);
await client.details("movie", 693134);
await sql`update kv_cache set expires_at = now() + interval '1 day'
where region = 'tmdb.request' and (key like '/movie/872585?%' or key like '/movie/693134?%')`.execute(h.db);
await client.details("movie", 872585);
await client.details("movie", 693134);
const left = async (like: string) =>
Number(
(
await sql<{ d: number }>`select extract(epoch from expires_at - now()) / 86400 as d
from kv_cache where region = 'tmdb.request' and key like ${like}`.execute(h.db)
).rows[0]!.d,
);
expect(await left("/movie/872585?%")).toBeGreaterThan(19.9);
expect(await left("/movie/693134?%")).toBeLessThanOrEqual(1);
}); });
it("GET /cache/tmdb 列出条目(键 / 过期 / 大小)与上限;POST /cache/tmdb/clear 清空", async () => { it("GET /cache/tmdb 列出条目(键 / 过期 / 大小)与上限;POST /cache/tmdb/clear 清空", async () => {
@@ -88,7 +134,7 @@ describe.skipIf(!hasPg)("C2-A 缓存(真实 PG)", () => {
{ ttl: 3600 }, { ttl: 3600 },
); );
const list = await json(await admin("GET", "/cache/tmdb")); const list = await json(await admin("GET", "/cache/tmdb"));
expect(list.data).toMatchObject({ maxSize: 16, ttlHours: 6, emptyResultTtlSec: 1800 }); expect(list.data).toMatchObject({ maxSize: 16, ttlHours: 20 * 24, emptyResultTtlSec: 2 * 86400 });
expect(list.data.counts.request).toBeGreaterThan(0); expect(list.data.counts.request).toBeGreaterThan(0);
const rec = list.data.items.find((i: { kind: string }) => i.kind === "recognize"); const rec = list.data.items.find((i: { kind: string }) => i.kind === "recognize");
expect(rec).toMatchObject({ tmdbId: 872585, title: "奥本海默", mediaType: "movie", posterPath: "/p.jpg" }); expect(rec).toMatchObject({ tmdbId: 872585, title: "奥本海默", mediaType: "movie", posterPath: "/p.jpg" });
@@ -143,4 +189,54 @@ describe.skipIf(!hasPg)("C2-A 缓存(真实 PG)", () => {
expect((await json(await admin("GET", "/cache/images"))).data.files).toBe(0); expect((await json(await admin("GET", "/cache/images"))).data.files).toBe(0);
expect((await json(await admin("POST", "/cache/images/clear"))).data.deleted).toBe(0); expect((await json(await admin("POST", "/cache/images/clear"))).data.deleted).toBe(0);
}); });
it("图片容量 LRU:超过预算按最久未访问(atime)删到预算的 90%;读盘命中刷新 atime、不动 mtime", async () => {
const lruDir = mkdtempSync(join(tmpdir(), "umt-c2-lru-"));
try {
const cache = new ImageCache({
dir: lruDir,
upstreamBase: "https://image.tmdb.org/t/p",
fetch: async () => new Response(new Uint8Array([...JPEG, ...new Uint8Array(92)])),
});
// 4 张 100 字节的图,访问时间从旧到新 a < b < c < d
for (const f of ["a", "b", "c", "d"]) await cache.serve("w92", `${f}.jpg`, false, null);
const p = (f: string) => join(lruDir, "t/p/w92", `${f}.jpg`);
const base = Date.now() - 10 * 86_400_000;
["a", "b", "c", "d"].forEach((f, i) => {
utimesSync(p(f), new Date(base + i * 3600_000), new Date(base));
});
// 读盘命中 a:atime 变成现在(最近使用),mtime 仍是写入时间
await cache.serve("w92", "a.jpg", true, null);
await new Promise((r) => setTimeout(r, 50));
const st = statSync(p("a"));
expect(st.atimeMs).toBeGreaterThan(Date.now() - 60_000);
expect(Math.abs(st.mtimeMs - base)).toBeLessThan(1000);
// 没超预算不动
expect(await cache.enforceBudget(1000)).toMatchObject({ deleted: 0, usedBytes: 400 });
// 预算 300 → 删到 ≤ 270:最久未访问的 b、c 被删,a(刚读过)与 d 保留
const r = await cache.enforceBudget(300);
expect(r).toMatchObject({ deleted: 2, freedBytes: 200, usedBytes: 200 });
expect(existsSync(p("a"))).toBe(true);
expect(existsSync(p("b"))).toBe(false);
expect(existsSync(p("c"))).toBe(false);
expect(existsSync(p("d"))).toBe(true);
} finally {
rmSync(lruDir, { recursive: true, force: true });
}
});
it("cache.images_lru 登记为维护任务,预算取 storage.budgets.imagesGb", async () => {
await mod.start?.(h.ctx);
const job = scheduleRegistry(h.ctx).get("cache.images_lru");
expect(job?.defaults).toMatchObject({ group: "maintain", schedule: { everyMs: 30 * 60_000 } });
const out = await job!.run({
jobId: "cache.images_lru",
runId: "r",
trigger: "manual",
now: new Date(),
config: {},
slot: "1",
});
expect(out?.summary).toMatch(/未超过预算 20 GB/);
});
}); });
@@ -468,18 +468,21 @@ describe.skipIf(!up)("resolveObservation / 身份与元数据", () => {
).rows; ).rows;
const opp = (await rows()).find((r) => r.key === "[未知][zh-CN]Oppenheimer-2023-None"); const opp = (await rows()).find((r) => r.key === "[未知][zh-CN]Oppenheimer-2023-None");
expect(opp?.value).toMatchObject({ id: 872585, type: "movie", title: "奥本海默" }); expect(opp?.value).toMatchObject({ id: 872585, type: "movie", title: "奥本海默" });
expect(Number(opp!.minutes)).toBeGreaterThan(23 * 60); // 正常条目按搜索类 TTL(storage.cacheTtl.tmdbSearchDays,缺省 7 天)
expect(Number(opp!.minutes)).toBeGreaterThan(7 * 24 * 60 - 5);
expect(Number(opp!.minutes)).toBeLessThanOrEqual(7 * 24 * 60);
// 同名同年的另一条资源:命中识别缓存,不再调 TMDB 搜索 // 同名同年的另一条资源:命中识别缓存,不再调 TMDB 搜索
const searches = tmdb.callsTo(/\/3\/search\//).length; const searches = tmdb.callsTo(/\/3\/search\//).length;
const again = await release("Oppenheimer.2023.1080p.BluRay.x264-OTHER"); const again = await release("Oppenheimer.2023.1080p.BluRay.x264-OTHER");
await resolveRelease(h.rt, { sourceReleaseId: again.id }); await resolveRelease(h.rt, { sourceReleaseId: again.id });
expect(tmdb.callsTo(/\/3\/search\//).length).toBe(searches); expect(tmdb.callsTo(/\/3\/search\//).length).toBe(searches);
// 查无此项 → 负缓存 {id: 0},TTL = 空结果短 TTL(30 分钟) // 查无此项 → 负缓存 {id: 0},TTL = 未命中 TTL(storage.cacheTtl.tmdbMissDays,缺省 1 天)
const none = await release("Zzqxv.Nonexistent.Title.2019.1080p.WEB-DL-NOGRP"); const none = await release("Zzqxv.Nonexistent.Title.2019.1080p.WEB-DL-NOGRP");
await resolveRelease(h.rt, { sourceReleaseId: none.id }); await resolveRelease(h.rt, { sourceReleaseId: none.id });
const neg = (await rows()).find((r) => r.key === "[未知][zh-CN]Zzqxv Nonexistent Title-2019-None"); const neg = (await rows()).find((r) => r.key === "[未知][zh-CN]Zzqxv Nonexistent Title-2019-None");
expect(neg?.value).toEqual({ id: 0 }); expect(neg?.value).toEqual({ id: 0 });
expect(Number(neg!.minutes)).toBeLessThanOrEqual(30); expect(Number(neg!.minutes)).toBeLessThanOrEqual(24 * 60);
expect(Number(neg!.minutes)).toBeGreaterThan(24 * 60 - 5);
const s2 = tmdb.callsTo(/\/3\/search\//).length; const s2 = tmdb.callsTo(/\/3\/search\//).length;
await resolveRelease(h.rt, { sourceReleaseId: none.id }); await resolveRelease(h.rt, { sourceReleaseId: none.id });
expect(tmdb.callsTo(/\/3\/search\//).length).toBe(s2); expect(tmdb.callsTo(/\/3\/search\//).length).toBe(s2);
+320
View File
@@ -0,0 +1,320 @@
import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { DEFAULT_STORAGE_SETTINGS, SystemHealth, SystemStorage } from "@umt/contracts";
import { type RawBuilder, sql } from "@umt/db";
import { afterAll, beforeAll, describe, expect, it } from "vitest";
import { scheduleRegistry } from "../../src/modules/schedules/registry.ts";
import { createSystemModule } from "../../src/modules/system/index.ts";
import { CLEANUP_JOB_IDS, deleteInBatches, RETENTION_JOBS } from "../../src/modules/system/retention.ts";
import { budgetLevel, resetDirScans } from "../../src/modules/system/storage.ts";
import { bearer, createHarness, type Harness, pgAvailable } from "../auth-settings/harness.ts";
import { DEV_KEY, json } from "../helpers.ts";
// CACHE-2 存储预算与保留:保留清理分批与边界、存储面板形状、health 存储档位、手动触发清理。
const hasPg = await pgAvailable();
const DAY = 86_400_000;
describe("预算档位", () => {
it("≥ 预算 critical,≥ 告警线 warn,取不到 unknown", () => {
expect(budgetLevel(null, 100, 80)).toBe("unknown");
expect(budgetLevel(79, 100, 80)).toBe("ok");
expect(budgetLevel(80, 100, 80)).toBe("warn");
expect(budgetLevel(100, 100, 80)).toBe("critical");
});
it("每个清理目标都对应一个定时任务", () => {
expect(Object.keys(CLEANUP_JOB_IDS).sort()).toEqual(
[
"observation",
"sse_event",
"heartbeat",
"task_run",
"agent_event",
"audit",
"capability_stats",
"identity_pool",
"analyze",
"images",
"kv_cache",
].sort(),
);
});
});
describe.skipIf(!hasPg)("CACHE-2 存储(真实 PG)", () => {
let h: Harness;
let admin: ReturnType<typeof bearer>;
const dir = mkdtempSync(join(tmpdir(), "umt-cache2-"));
beforeAll(async () => {
h = await createHarness({ engine: true });
const { createApp } = await import("../../src/http/app.ts");
const { createAuthSettingsModule } = await import("../../src/modules/auth-settings/index.ts");
const sys = createSystemModule({
engineSchema: h.engineSchema,
apiHeartbeat: false,
storagePaths: { dataDir: dir, sessionDir: join(dir, "agent-sessions") },
retentionBatch: { batchSize: 3, pauseMs: 0 },
healthDeps: {
dataDir: dir,
diskUsage: async (path) => ({
status: "ok",
path,
totalBytes: 1000,
freeBytes: 500,
availableBytes: 500,
usedBytes: 500,
usedRatio: 0.5,
}),
},
});
h.app = createApp({
config: h.ctx.config,
logger: h.ctx.logger,
bus: h.bus,
db: h.db,
role: "all",
version: "0.0.0-test",
modules: [createAuthSettingsModule({ setupTokenOnStart: false, engineSchema: h.engineSchema }), sys],
ctx: h.ctx,
}) as Harness["app"];
admin = bearer(h.app, DEV_KEY);
await sys.start?.(h.ctx);
// 目录:图片 3 KB、备份 2 份、会话 1 KB、证据 1 KB
mkdirSync(join(dir, "cache", "images", "t", "p", "w92"), { recursive: true });
writeFileSync(join(dir, "cache", "images", "t", "p", "w92", "a.jpg"), new Uint8Array(3072));
for (const b of ["20260901T000000Z", "20260902T000000Z", ".20260903T000000Z.partial"]) {
mkdirSync(join(dir, "backups", b), { recursive: true });
writeFileSync(join(dir, "backups", b, "db.dump"), new Uint8Array(1024));
}
mkdirSync(join(dir, "agent-sessions"), { recursive: true });
writeFileSync(join(dir, "agent-sessions", "s.jsonl"), new Uint8Array(1024));
mkdirSync(join(dir, "tmp", "evidence"), { recursive: true });
writeFileSync(join(dir, "tmp", "evidence", "e.jpg"), new Uint8Array(1024));
resetDirScans();
}, 60_000);
afterAll(async () => {
await h?.close();
rmSync(dir, { recursive: true, force: true });
});
const job = (target: string) => {
const j = RETENTION_JOBS.find((x) => x.target === target)!;
return (now = new Date(), settings = DEFAULT_STORAGE_SETTINGS) =>
j.run({ db: h.db, engineSchema: h.engineSchema, settings, now, batch: { batchSize: 3, pauseMs: 0 } });
};
const count = async (q: RawBuilder<{ n: number }>) => Number((await q.execute(h.db)).rows[0]!.n);
it("保留任务登记在维护组,各自一个 key", () => {
const reg = scheduleRegistry(h.ctx);
for (const j of RETENTION_JOBS) {
expect(reg.get(j.jobId)?.defaults, j.jobId).toMatchObject({ group: "maintain" });
}
expect(reg.get("storage.analyze")?.defaults.schedule).toMatchObject({ everyMs: 7 * DAY });
expect(reg.get("system.heartbeat_cleanup")?.defaults.group).toBe("maintain");
});
it("deleteInBatches:按批删除,只删满足条件的行;总时限到了停在中途,下次接着删", async () => {
await sql`insert into sse_event (type, at) select 'x', now() - interval '30 days' from generate_series(1, 10)`.execute(
h.db,
);
await sql`insert into sse_event (type, at) values ('keep', now())`.execute(h.db);
// 时钟每次调用前进 1 秒,预算 1.5 秒:删完 2 批就停
let t = 0;
const r1 = await deleteInBatches(h.db, sql.raw("sse_event"), sql`at < now() - interval '7 days'`, {
batchSize: 3,
pauseMs: 0,
budgetMs: 1500,
now: () => (t += 1000),
});
expect(r1).toEqual({ deleted: 6, batches: 2, truncated: true });
const r2 = await deleteInBatches(h.db, sql.raw("sse_event"), sql`at < now() - interval '7 days'`, {
batchSize: 3,
pauseMs: 0,
});
expect(r2).toEqual({ deleted: 4, batches: 2, truncated: false });
expect(await count(sql<{ n: number }>`select count(*)::int as n from sse_event`)).toBe(1);
});
it("每批语句超时:停止本次运行,不抛错", async () => {
await sql`insert into sse_event (type, at) values ('slow', now() - interval '30 days')`.execute(h.db);
const r = await deleteInBatches(
h.db,
sql.raw("sse_event"),
sql`at < now() - interval '7 days' and pg_sleep(0.3) is not null`,
{ batchSize: 3, statementTimeoutMs: 50 },
);
expect(r).toEqual({ deleted: 0, batches: 0, truncated: true });
await sql`delete from sse_event where type = 'slow'`.execute(h.db);
});
it("观察记录:保留边界(90 天前一刻删,之后保留)", async () => {
const feed = (
await sql<{
id: string;
}>`insert into feed_source (name, url_display) values ('rss', 'https://x/rss') returning id`.execute(h.db)
).rows[0]!.id;
const now = new Date();
const ins = (guid: string, at: Date) =>
sql`insert into observation (feed_id, guid, content_digest, title, created_at)
values (${feed}, ${guid}, ${guid}, ${guid}, ${at})`.execute(h.db);
for (let i = 0; i < 5; i++) await ins(`old${i}`, new Date(now.getTime() - 90 * DAY - 60_000 - i));
await ins("edge", new Date(now.getTime() - 90 * DAY + 60_000));
await ins("new", now);
expect(await job("observation")(now)).toMatch(/删除 5 行(2 批)/);
const left = await sql<{ guid: string }>`select guid from observation order by guid`.execute(h.db);
expect(left.rows.map((r) => r.guid)).toEqual(["edge", "new"]);
});
it("任务运行:只删过期且已结束的业务任务与引擎 run", async () => {
const now = new Date();
const old = new Date(now.getTime() - 61 * DAY);
const ins = (key: string, status: string, at: Date) =>
sql`insert into task (kind, idempotency_key, status, created_at) values ('k', ${key}, ${status}, ${at})`.execute(
h.db,
);
await ins("t1", "succeeded", old);
await ins("t2", "failed", old);
await ins("t3", "running", old);
await ins("t4", "succeeded", now);
const t = (n: string) => sql.raw(`"${h.engineSchema}".${n}`);
const run = (id: string, status: string, at: Date) =>
sql`insert into ${t("workflow_runs")} (namespace_id, id, workflow_name, version, status, config, attempts, created_at, updated_at)
values ('default', ${id}, 'w', 'v1', ${status}, '{}'::jsonb, 1, ${at}, ${at})`.execute(h.db);
await run("r-old-done", "completed", old);
await run("r-old-running", "running", old);
await run("r-new", "completed", now);
expect(await job("task_run")(now)).toMatch(/业务任务 2、引擎 run 1/);
expect(
(await sql<{ idempotency_key: string }>`select idempotency_key from task order by 1`.execute(h.db)).rows.map(
(r) => r.idempotency_key,
),
).toEqual(["t3", "t4"]);
expect(
(await sql<{ id: string }>`select id from ${t("workflow_runs")} order by id`.execute(h.db)).rows.map((r) => r.id),
).toEqual(expect.arrayContaining(["r-old-running", "r-new"]));
expect(
await count(sql<{ n: number }>`select count(*)::int as n from ${t("workflow_runs")} where id = 'r-old-done'`),
).toBe(0);
});
it("Agent 事件:只删已结束会话里过期的事件;审计与能力统计分开保留", async () => {
const now = new Date();
const runs = await sql<{ id: string; status: string }>`insert into agent_run (trigger, goal, status)
values ('user', 'g', 'succeeded'), ('user', 'g', 'running') returning id, status`.execute(h.db);
const done = runs.rows.find((r) => r.status === "succeeded")!.id;
const live = runs.rows.find((r) => r.status === "running")!.id;
const old = new Date(now.getTime() - 91 * DAY);
await sql`insert into agent_run_event (run_id, seq, type, at) values
(${done}, 1, 'run_started', ${old}), (${done}, 2, 'run_started', ${now}), (${live}, 1, 'run_started', ${old})`.execute(
h.db,
);
expect(await job("agent_event")(now)).toMatch(/Agent 事件 1、/);
expect(await count(sql<{ n: number }>`select count(*)::int as n from agent_run_event`)).toBe(2);
await sql`delete from audit_event`.execute(h.db);
const a = (days: number, targetType: string | null) =>
sql`insert into audit_event (actor_type, action, target_type, at)
values ('system', 'x', ${targetType}, ${new Date(now.getTime() - days * DAY)})`.execute(h.db);
await a(181, null);
await a(179, null);
await a(100, "capability");
await a(80, "capability");
await a(181, "capability");
expect(await job("audit")(now)).toMatch(/审计事件删除 1 行/);
expect(await job("capability_stats")(now)).toMatch(/能力审计 2、/);
const left = await sql<{ t: string | null; d: number }>`select target_type as t,
round(extract(epoch from now() - at) / 86400)::int as d from audit_event order by d`.execute(h.db);
expect(left.rows).toEqual([
{ t: "capability", d: 80 },
{ t: null, d: 179 },
]);
});
it("待识别池:unmatched 超过保留期没有新证据的归档,其余不动", async () => {
const site = (
await sql<{ id: string }>`insert into site (key, name, adapter, base_url)
values ('s-cache2', 'S', 'nexusphp', 'https://s.example') returning id`.execute(h.db)
).rows[0]!.id;
const now = new Date();
const cand = async (rid: string, reason: string, lastAttemptDays: number) => {
const sr = (
await sql<{ id: string }>`insert into source_release (site_id, site_release_id, title)
values (${site}, ${rid}, ${rid}) returning id`.execute(h.db)
).rows[0]!.id;
await sql`insert into identity_candidate (source_release_id, reason, evidence_version, last_attempt_at)
values (${sr}, ${reason}, 'v1', ${new Date(now.getTime() - lastAttemptDays * DAY)})`.execute(h.db);
};
await cand("r1", "unmatched", 31);
await cand("r2", "unmatched", 29);
await cand("r3", "ambiguous", 31);
expect(await job("identity_pool")(now)).toBe("归档 1 条待识别条目");
const st = await sql<{ t: string; s: string }>`select sr.title as t, c.status as s from identity_candidate c
join source_release sr on sr.id = c.source_release_id order by sr.title`.execute(h.db);
expect(st.rows).toEqual([
{ t: "r1", s: "archived" },
{ t: "r2", s: "pending" },
{ t: "r3", s: "pending" },
]);
});
it("ANALYZE 任务", async () => {
expect(await job("analyze")()).toMatch(/ANALYZE 完成/);
});
it("GET /system/storage:表体积 top 15、库体积对照预算、kv_cache、目录用量、清理任务", async () => {
await sql`insert into kv_cache (region, key, value, expires_at) values
('tmdb.request', 'a', '{}'::jsonb, now() + interval '1 day'),
('tmdb.request', 'b', '{}'::jsonb, now() - interval '1 day')`.execute(h.db);
const res = await admin("GET", "/system/storage");
expect(res.status).toBe(200);
const d = SystemStorage.parse((await json(res)).data);
expect(d.db.tables.length).toBeGreaterThan(0);
expect(d.db.tables.length).toBeLessThanOrEqual(15);
expect(d.db.usedBytes).toBeGreaterThan(0);
expect(d.db.budgetBytes).toBe(25 * 1024 ** 3);
expect(d.db.warnBytes).toBe(20 * 1024 ** 3);
expect(d.kvCache.find((r) => r.region === "tmdb.request")).toMatchObject({ entries: 2, expired: 1 });
expect(d.images).toMatchObject({ usedBytes: 3072, files: 1, level: "ok", budgetBytes: 20 * 1024 ** 3 });
expect(d.backups).toMatchObject({ usedBytes: 3072, count: 2, level: "ok" });
expect(d.sessionsEvidence).toMatchObject({ usedBytes: 2048, level: "ok" });
expect(d.dataDir.map((x) => x.name)).toEqual(
expect.arrayContaining(["agent-sessions", "backups", "cache/images", "tmp/evidence"]),
);
const obs = d.cleanup.find((c) => c.target === "observation")!;
expect(obs).toMatchObject({ jobId: "storage.retention.observation", retention: "90 天" });
expect(d.cleanup.find((c) => c.target === "heartbeat")?.retention).toBe("24 小时");
expect(d.level).toBe("ok");
});
it("预算超了:存储面板与 health 都报出来", async () => {
const r = await admin("PATCH", "/settings", {
storage: { budgets: { pgDataWarnGb: 0.01, pgDataGb: 50 } },
});
expect(r.status).toBe(200);
// 把 PG 告警线压到库体积之下验证 db warn
const d = SystemStorage.parse((await json(await admin("GET", "/system/storage"))).data);
expect(d.db.level).toBe("warn");
expect(d.level).toBe("warn");
const hl = SystemHealth.parse((await json(await admin("GET", "/system/health"))).data);
expect(hl.storage.level).toBe("warn");
expect(hl.storage.alerts.join("")).toMatch(/数据库/);
expect(["warn", "critical"]).toContain(hl.level);
await admin("PATCH", "/settings", { storage: { budgets: { pgDataWarnGb: 20, pgDataGb: 25 } } });
});
it("POST /system/storage/cleanup/{target}:排入对应定时任务;重复请求合并;未知目标 422", async () => {
await sql`insert into scheduled_job (id, name, board_group, kind, enabled, default_enabled, schedule, default_schedule)
values ('storage.retention.audit', '审计清理', 'maintain', 'task', true, true,
'{"type":"interval","everyMs":86400000}'::jsonb, '{"type":"interval","everyMs":86400000}'::jsonb)
on conflict (id) do nothing`.execute(h.db);
const r1 = await json(await admin("POST", "/system/storage/cleanup/audit"));
expect(r1.data).toEqual({ target: "audit", jobId: "storage.retention.audit", queued: true });
const r2 = await json(await admin("POST", "/system/storage/cleanup/audit"));
expect(r2.data.queued).toBe(false);
expect((await admin("POST", "/system/storage/cleanup/nope")).status).toBe(422);
expect((await admin("POST", "/system/storage/cleanup/observation")).status).toBe(404);
});
});
+21
View File
@@ -19,6 +19,27 @@ services:
postgres: postgres:
image: postgres:16-alpine image: postgres:16-alpine
restart: unless-stopped restart: unless-stopped
# 内存参数按 4C/8G 机器给 PG 3G 设定(CACHE-2);机器不同按 docs/runbook/storage.md「PG 参数」调整,
# 改 .env 里的 PG_* 后 docker compose up -d postgres 重建生效
command:
- postgres
- -c
- shared_buffers=${PG_SHARED_BUFFERS:-2GB}
- -c
- effective_cache_size=${PG_EFFECTIVE_CACHE_SIZE:-5GB}
- -c
- work_mem=${PG_WORK_MEM:-32MB}
- -c
- maintenance_work_mem=${PG_MAINTENANCE_WORK_MEM:-256MB}
- -c
- max_wal_size=${PG_MAX_WAL_SIZE:-2GB}
mem_limit: ${PG_MEM_LIMIT:-3g}
deploy:
resources:
limits:
memory: ${PG_MEM_LIMIT:-3g}
# 并行查询的动态共享内存走 /dev/shm,Docker 缺省 64MB 不够
shm_size: 512mb
environment: environment:
POSTGRES_USER: umt POSTGRES_USER: umt
POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:?请设置 POSTGRES_PASSWORD} POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:?请设置 POSTGRES_PASSWORD}
+8
View File
@@ -17,6 +17,14 @@ PGID=10001
# 备份保留份数(0 = 不轮转) # 备份保留份数(0 = 不轮转)
BACKUP_KEEP=14 BACKUP_KEEP=14
# PG 内存(缺省按 4C/8G 给 PG 3G;调整方法见 docs/runbook/storage.md「PG 参数」)
# PG_MEM_LIMIT=3g
# PG_SHARED_BUFFERS=2GB
# PG_EFFECTIVE_CACHE_SIZE=5GB
# PG_WORK_MEM=32MB
# PG_MAINTENANCE_WORK_MEM=256MB
# PG_MAX_WAL_SIZE=2GB
LOG_LEVEL=info LOG_LEVEL=info
# APP_VERSION=v0.1.0 # APP_VERSION=v0.1.0
# 国内网络构建换源(可选):APT_MIRROR=mirrors.ustc.edu.cn # 国内网络构建换源(可选):APT_MIRROR=mirrors.ustc.edu.cn
+7 -5
View File
@@ -15,7 +15,8 @@
| `@cached(region, maxsize, ttl, skip_none, skip_empty, skip_if, empty_ttl, empty_if)` | `cachedCall(cache, key, opts, fn)` | should_cache / get_cache_ttl 逻辑逐条对应 | | `@cached(region, maxsize, ttl, skip_none, skip_empty, skip_if, empty_ttl, empty_if)` | `cachedCall(cache, key, opts, fn)` | should_cache / get_cache_ttl 逻辑逐条对应 |
| `fresh()` 上下文 | `opts.fresh` | 跳过读、照常写 | | `fresh()` 上下文 | `opts.fresh` | 跳过读、照常写 |
偏离(仅存储介质):值用 JSON(jsonb)代替 pickle;过期行需要物理删除,加定时任务 `cache.sweep`(每 10 分钟); 偏离(仅存储介质与 TTL 取值):值用 JSON(jsonb)代替 pickle;TTL 值来自 `settings.storage.cacheTtl`(CACHE-2,见 runbook/storage.md),
追踪中的作品详情命中时续期(`PgCache.renew`,只延长);过期行需要物理删除,加定时任务 `cache.sweep`(每 10 分钟);
PG 读写失败时 TMDB 路径按未命中处理(内存缓存不会失败)。 PG 读写失败时 TMDB 路径按未命中处理(内存缓存不会失败)。
## TMDB 请求缓存:`tmdbv3api/tmdb.py TMDb.request` 的 `@cached` → `packages/adapters/src/tmdb/client.ts` ## TMDB 请求缓存:`tmdbv3api/tmdb.py TMDb.request` 的 `@cached` → `packages/adapters/src/tmdb/client.ts`
@@ -23,8 +24,8 @@ PG 读写失败时 TMDB 路径按未命中处理(内存缓存不会失败)。
| 上游 | 我们 | | 上游 | 我们 |
|---|---| |---|---|
| `maxsize = CONF.tmdb`(256;大内存 1024) | 设置 `cache.tmdbMaxSize`(缺省 256) | | `maxsize = CONF.tmdb`(256;大内存 1024) | 设置 `cache.tmdbMaxSize`(缺省 256) |
| `ttl = CONF.meta = META_CACHE_EXPIRE 小时`(缺省 24) | 设置 `cache.tmdbTtlHours`(缺省 24) | | `ttl = CONF.meta = META_CACHE_EXPIRE 小时`(缺省 24) | CACHE-2 起按请求类别取设置:`/search/`、`/find/` 用 `storage.cacheTtl.tmdbSearchDays`(7 天),其余用 `tmdbDetailDays`(30 天) |
| `empty_ttl = EMPTY_RESULT_CACHE_TTL`(1800 秒),`empty_if = results 为空列表` | 设置 `cache.emptyResultTtlSec`(缺省 1800);`isEmptyResult` 同判定 | | `empty_ttl = EMPTY_RESULT_CACHE_TTL`(1800 秒),`empty_if = results 为空列表` | `storage.cacheTtl.tmdbMissDays`(1 天);`isEmptyResult` 同判定 |
| `skip_none`(连接失败不缓存)、`skip_if = success:false`(业务失败不缓存) | 这两类在 `fetchWithRetry` 抛错,不会写缓存 | | `skip_none`(连接失败不缓存)、`skip_if = success:false`(业务失败不缓存) | 这两类在 `fetchWithRetry` 抛错,不会写缓存 |
| 键 = 函数名 + (method, url, params) | 键 = `path?query`(参数排序,不含 api_key / token) | | 键 = 函数名 + (method, url, params) | 键 = `path?query`(参数排序,不含 api_key / token) |
| region 自动生成 | region `tmdb.request` | | region 自动生成 | region `tmdb.request` |
@@ -39,7 +40,7 @@ PG 读写失败时 TMDB 路径按未命中处理(内存缓存不会失败)。
| `update(meta, info)`:info 非空写正常条目(默认 ttl);`{}` 写负缓存 `{"id": 0}`(`EMPTY_RESULT_CACHE_TTL`);None 不写 | `saveRecognizeCache(meta, info)`:info / null / undefined 三态一一对应 | | `update(meta, info)`:info 非空写正常条目(默认 ttl);`{}` 写负缓存 `{"id": 0}`(`EMPTY_RESULT_CACHE_TTL`);None 不写 | `saveRecognizeCache(meta, info)`:info / null / undefined 三态一一对应 |
| `get(meta)` 读;类型冲突(元数据电视剧、缓存电影)删除后按未命中 | `loadRecognizeCache` 同 | | `get(meta)` 读;类型冲突(元数据电视剧、缓存电影)删除后按未命中 | `loadRecognizeCache` 同 |
| `update` 时类型冲突拒绝写入 | 同 | | `update` 时类型冲突拒绝写入 | 同 |
| `maxsize = CONF.tmdb`、`ttl = CONF.meta`,每条独立 `expires_at` | 同设置;`kv_cache.expires_at` 每条独立 | | `maxsize = CONF.tmdb`、`ttl = CONF.meta`,每条独立 `expires_at` | 条数同设置;正常条目 TTL 取 `tmdbSearchDays`,负缓存取 `tmdbMissDays`;`kv_cache.expires_at` 每条独立 |
| `_restore / save`:pickle 文件持久化与恢复 | 不需要:表本身就是持久化(UNLOGGED 在 PG 崩溃时清空,等价于缓存丢失) | | `_restore / save`:pickle 文件持久化与恢复 | 不需要:表本身就是持久化(UNLOGGED 在 PG 崩溃时清空,等价于缓存丢失) |
| `list_items / delete / clear` | `GET /cache/tmdb`、`POST /cache/tmdb/clear`(清空时同时清请求缓存,对应 MP `clear_cache()`) | | `list_items / delete / clear` | `GET /cache/tmdb`、`POST /cache/tmdb/clear`(清空时同时清请求缓存,对应 MP `clear_cache()`) |
| `is_redis` 分支 | 忽略 | | `is_redis` 分支 | 忽略 |
@@ -68,5 +69,6 @@ PG 读写失败时 TMDB 路径按未命中处理(内存缓存不会失败)。
| `SECURITY_IMAGE_DOMAINS` 白名单 | 只代理 `image.tmdb.org`,尺寸与文件名走白名单 | | `SECURITY_IMAGE_DOMAINS` 白名单 | 只代理 `image.tmdb.org`,尺寸与文件名走白名单 |
| `clear_temp`:删除 `CACHE_PATH/images` 下 mtime 超过 `GLOBAL_IMAGE_CACHE_DAYS`(7)天的文件与空目录 | 设置 `cache.globalImageCacheDays`(缺省 7);定时任务 `cache.images_clean` 每天执行 | | `clear_temp`:删除 `CACHE_PATH/images` 下 mtime 超过 `GLOBAL_IMAGE_CACHE_DAYS`(7)天的文件与空目录 | 设置 `cache.globalImageCacheDays`(缺省 7);定时任务 `cache.images_clean` 每天执行 |
偏离:MP 失败时返回空响应,我们回 502(前端回退占位);MP 按配置走代理服务器,我们直连(TMDB API 也未走代理); 偏离:MP 失败时返回空响应,我们回 502(前端回退占位);CACHE-2 加了容量 LRU(`cache.images_lru`,按 atime,预算 `storage.budgets.imagesGb`),
按天清理保留作兜底;MP 按配置走代理服务器,我们直连(TMDB API 也未走代理);
MP 只在启动时清理,我们改为每天一次的定时任务。 MP 只在启动时清理,我们改为每天一次的定时任务。
+10 -4
View File
@@ -24,11 +24,14 @@
| region | 内容 | 键 | TTL | 条数上限 | | region | 内容 | 键 | TTL | 条数上限 |
|---|---|---|---|---| |---|---|---|---|---|
| `tmdb.request` | TMDB 请求响应(MP `TMDb.request`) | `path?query`(不含凭证) | `cache.tmdbTtlHours`;`results` 为空按 `cache.emptyResultTtlSec` | `cache.tmdbMaxSize` | | `tmdb.request` | TMDB 请求响应(MP `TMDb.request`) | `path?query`(不含凭证) | 搜索 / find:`storage.cacheTtl.tmdbSearchDays`(7 天);详情 / 季:`tmdbDetailDays`(30 天,追踪中的作品命中续期);`results` 为空:`tmdbMissDays`(1 天) | `cache.tmdbMaxSize` |
| `tmdb.recognize` | TMDB 识别结果(MP `TmdbCache`) | `[类型][语言]{tmdbid 或名称}-{年份}-{季}` | 同上;未识别 `{id:0}` 按空结果 TTL | `cache.tmdbMaxSize` | | `tmdb.recognize` | TMDB 识别结果(MP `TmdbCache`) | `[类型][语言]{tmdbid 或名称}-{年份}-{季}` | `tmdbSearchDays`;未识别 `{id:0}` 按 `tmdbMissDays` | `cache.tmdbMaxSize` |
| `site.detail` | 站点详情页扩展内容 | `source_release.id` | `max(站点详情缓存期, storage.cacheTtl.siteDetailDays)`(7 天) | 5000 |
| `websearch` | 网页搜索结果 | 参数 sha256 | `storage.cacheTtl.webSearchDays`(1 天) | 1000 |
设置(设置 → 元数据 → TMDB 缓存):`cache.tmdbTtlHours`(缺省 24,MP `META_CACHE_EXPIRE`)、`cache.tmdbMaxSize`(缺省 256,MP `CONF.tmdb`)、 TTL 设置(CACHE-2 起):`settings.storage.cacheTtl`,见 [storage.md](storage.md)。MP 的 `META_CACHE_EXPIRE` 只有一个值,
`cache.emptyResultTtlSec`(缺省 1800,MP `EMPTY_RESULT_CACHE_TTL`)。 我们按请求类别拆成搜索 / 详情两档,`EMPTY_RESULT_CACHE_TTL` 对应 `tmdbMissDays`;缓存语义不变。
条数上限仍是 `cache.tmdbMaxSize`(缺省 256,MP `CONF.tmdb`)。旧字段 `cache.tmdbTtlHours`、`cache.emptyResultTtlSec` 不再读取。
API:`GET /api/v1/cache/tmdb?kind=recognize|request&limit=` 列出条目(键 / 过期 / 大小,识别条目带 tmdbId / 标题 / 海报); API:`GET /api/v1/cache/tmdb?kind=recognize|request&limit=` 列出条目(键 / 过期 / 大小,识别条目带 tmdbId / 标题 / 海报);
`POST /api/v1/cache/tmdb/clear` 清空两类(仅 admin)。 `POST /api/v1/cache/tmdb/clear` 清空两类(仅 admin)。
@@ -48,6 +51,8 @@ API:`GET /api/v1/cache/tmdb?kind=recognize|request&limit=` 列出条目(键 /
- 按文件头识别图片类型,非图片 / SVG 拒绝;ETag = md5(内容),`Cache-Control: public, max-age=604800`,If-None-Match 相同回 304。 - 按文件头识别图片类型,非图片 / SVG 拒绝;ETag = md5(内容),`Cache-Control: public, max-age=604800`,If-None-Match 相同回 304。
- 上游超时(20s)/ 非 200 / 非图片 → 502,前端回退无彩占位。 - 上游超时(20s)/ 非 200 / 非图片 → 502,前端回退无彩占位。
- 清理:定时任务 `cache.images_clean`(每天)删除 mtime 早于 `cache.globalImageCacheDays`(缺省 7)天的文件。 - 清理:定时任务 `cache.images_clean`(每天)删除 mtime 早于 `cache.globalImageCacheDays`(缺省 7)天的文件。
- 容量 LRU(CACHE-2):`cache.images_lru`(30 分钟)在总量超过 `storage.budgets.imagesGb`(缺省 20 GB)时按最久未访问删到 90%;
读盘命中会刷新文件 atime。见 [storage.md](storage.md)。
- 用量 / 清空:`GET /api/v1/cache/images`、`POST /api/v1/cache/images/clear`(仅 admin);设置 → 元数据 → 图片代理与缓存。 - 用量 / 清空:`GET /api/v1/cache/images`、`POST /api/v1/cache/images/clear`(仅 admin);设置 → 元数据 → 图片代理与缓存。
前端:作品库列表、作品详情、搜索面板的 TMDB 候选在有 `posterPath`(metadata.posterPath / TMDB 候选)时显示海报, 前端:作品库列表、作品详情、搜索面板的 TMDB 候选在有 `posterPath`(metadata.posterPath / TMDB 候选)时显示海报,
@@ -58,4 +63,5 @@ API:`GET /api/v1/cache/tmdb?kind=recognize|request&limit=` 列出条目(键 /
- TMDB 缓存不生效:确认 `select region, count(*) from kv_cache group by 1` 有 `tmdb.request`;PG 读写失败时按未命中处理并记 - TMDB 缓存不生效:确认 `select region, count(*) from kv_cache group by 1` 有 `tmdb.request`;PG 读写失败时按未命中处理并记
`TMDB 缓存读写失败,按未命中处理`。 `TMDB 缓存读写失败,按未命中处理`。
- 改了 TMDB 缓存设置:新写入的条目按新 TTL / 上限;已有条目保持原过期时间,必要时清空。 - 改了 TMDB 缓存设置:新写入的条目按新 TTL / 上限;已有条目保持原过期时间,必要时清空。
- TTL 拉长后条目更早被条数上限挤掉:`cache.tmdbMaxSize` 缺省只有 256,追踪作品多时调大。
- 数据库重启后缓存为空:UNLOGGED 表的正常行为。 - 数据库重启后缓存为空:UNLOGGED 表的正常行为。
+6 -2
View File
@@ -14,7 +14,8 @@
- 每 15 秒 upsert `process_heartbeat` 一行,同时往 `process_heartbeat_sample` 追加一条采样。 - 每 15 秒 upsert `process_heartbeat` 一行,同时往 `process_heartbeat_sample` 追加一条采样。
- 行主键 `instance_id = <kind>:<host>:<pid>`,容器重启覆盖同一行。 - 行主键 `instance_id = <kind>:<host>:<pid>`,容器重启覆盖同一行。
- 正常退出(SIGTERM / SIGINT)时写入器把本行标 `stopped_at`,读取方立即归入已下线实例;重启后同一行的 upsert 把它清回 null。 - 正常退出(SIGTERM / SIGINT)时写入器把本行标 `stopped_at`,读取方立即归入已下线实例;重启后同一行的 upsert 把它清回 null。
- 裁剪由定时任务 `system.heartbeat_cleanup`(维护组,1 小时)负责:24 小时没心跳的进程行删除,采样保留 7 天。 - 裁剪由定时任务 `system.heartbeat_cleanup`(维护组,1 小时)负责:24 小时没心跳的进程行删除,采样保留
`storage.retentionDays.heartbeatHours`(缺省 24 小时,CACHE-2 前是 7 天)。
- 写库失败只记 debug 日志,不影响业务。 - 写库失败只记 debug 日志,不影响业务。
## GET /system/health 字段 ## GET /system/health 字段
@@ -40,12 +41,13 @@
| `mounts[]` | 库根做哨兵检查(`checkRoot`),下载目录只查目录可读。`status`:`online` / `offline` / `timeout` | | `mounts[]` | 库根做哨兵检查(`checkRoot`),下载目录只查目录可读。`status`:`online` / `offline` / `timeout` |
| `engine` | 待处理 run 数、执行中 run 数、最老待处理时长、24 小时结束数 / 失败数 / 失败率 | | `engine` | 待处理 run 数、执行中 run 数、最老待处理时长、24 小时结束数 / 失败数 / 失败率 |
| `schedules` | 定时任务总数、启用数、执行中数、最近一次失败的启用任务、24 小时失败运行数 | | `schedules` | 定时任务总数、启用数、执行中数、最近一次失败的启用任务、24 小时失败运行数 |
| `storage` | 存储预算档位与告警文案(PG 库体积、图片缓存、备份、会话与证据对照 `settings.storage.budgets`),明细见 `GET /system/storage`([storage.md](storage.md)) |
单个路径的 statfs 或哨兵检查超过 3 秒记 `timeout`:磁盘段标 critical,挂载段标 `timeout`。请求本身不会被卡住。 单个路径的 statfs 或哨兵检查超过 3 秒记 `timeout`:磁盘段标 critical,挂载段标 `timeout`。请求本身不会被卡住。
## GET /system/health/history?hours= ## GET /system/health/history?hours=
- `hours` 取 1–168,缺省 6。 - `hours` 取 1–168,缺省 6。采样只保留 `heartbeatHours`(缺省 24 小时),更早的部分为空。
- 按桶聚合,每条序列最多约 360 个点。桶宽最小 15 秒,168 小时时是 1680 秒。 - 按桶聚合,每条序列最多约 360 个点。桶宽最小 15 秒,168 小时时是 1680 秒。
- 每个进程一条序列,每个点有 `lagMs`(桶内平均)、`lagMaxMs`(桶内最大)、`runningRuns`、`sseConnections`。 - 每个进程一条序列,每个点有 `lagMs`(桶内平均)、`lagMaxMs`(桶内最大)、`runningRuns`、`sseConnections`。
@@ -61,6 +63,8 @@
- 最老待处理 run 等待 ≥ 5 分钟记 warn,≥ 30 分钟记 critical。通常是 worker 没起来或并发被占满。 - 最老待处理 run 等待 ≥ 5 分钟记 warn,≥ 30 分钟记 critical。通常是 worker 没起来或并发被占满。
- 24 小时失败率 ≥ 10% 记 warn;失败率 ≥ 30% 且结束数 ≥ 10 记 critical。 - 24 小时失败率 ≥ 10% 记 warn;失败率 ≥ 30% 且结束数 ≥ 10 记 critical。
- 有启用的定时任务最近一次运行失败:warn。 - 有启用的定时任务最近一次运行失败:warn。
- PG 库体积 ≥ `pgDataWarnGb`(20 GB)记 warn,≥ `pgDataGb`(25 GB)记 critical;备份、会话与证据 ≥ 预算 80% 记 warn、≥ 预算记 critical;
图片缓存超预算 10% 记 warn。
## 引擎 runs 的业务字段 ## 引擎 runs 的业务字段
+6 -1
View File
@@ -30,7 +30,12 @@
| `trash.purge` | 回收站清理 | 维护 | 24 小时 | 入队 `purgeTrash` | | `trash.purge` | 回收站清理 | 维护 | 24 小时 | 入队 `purgeTrash` |
| `pi.token_refresh` | 上游登录刷新 | 维护 | 10 分钟 | 入队 `refreshPiTokens`,幂等键 `refreshPiTokens:<时间槽>`;对 30 分钟内过期的订阅登录上游刷新 token | | `pi.token_refresh` | 上游登录刷新 | 维护 | 10 分钟 | 入队 `refreshPiTokens`,幂等键 `refreshPiTokens:<时间槽>`;对 30 分钟内过期的订阅登录上游刷新 token |
| `auth.cleanup` | 登录会话清理 | 维护 | 1 小时 | 直接执行 `cleanupAuthTables`:删 7 天前的登录尝试、30 天前过期或吊销的会话 | | `auth.cleanup` | 登录会话清理 | 维护 | 1 小时 | 直接执行 `cleanupAuthTables`:删 7 天前的登录尝试、30 天前过期或吊销的会话 |
| `system.heartbeat_cleanup` | 进程心跳清理 | 维护 | 1 小时 | 删 24 小时没有心跳的进程行、7 天前的心跳采样 | | `system.heartbeat_cleanup` | 进程心跳清理 | 维护 | 1 小时 | 删 24 小时没有心跳的进程行、超过 `storage.retentionDays.heartbeatHours`(24 小时)的心跳采样 |
| `cache.sweep` | 缓存清理 | 维护 | 10 分钟 | 删 kv_cache 过期行 |
| `cache.images_clean` | 图片缓存清理 | 维护 | 24 小时 | 删超过保留天数的图片 |
| `cache.images_lru` | 图片缓存容量控制 | 维护 | 30 分钟 | 超过 `storage.budgets.imagesGb` 按最久未访问删到 90% |
| `storage.retention.*` | 保留清理(观察 / SSE / 任务运行 / Agent 事件 / 审计 / 能力统计 / 待识别池归档) | 维护 | 1–24 小时 | 分批删除超过保留期的流水数据,见 [storage.md](storage.md) |
| `storage.analyze` | 统计信息更新 | 维护 | 7 天 | 全库 `ANALYZE` |
| `maintain.backup` | 备份 | 维护 | 每天 03:30(占位) | 未配置。备份目前由宿主机上的 `scripts/backup.sh` 执行 | | `maintain.backup` | 备份 | 维护 | 每天 03:30(占位) | 未配置。备份目前由宿主机上的 `scripts/backup.sh` 执行 |
| `check.pending` | 待处理巡检 | Agent | 6 小时,默认关闭 | 只读。有未关闭事项时才会运行 | | `check.pending` | 待处理巡检 | Agent | 6 小时,默认关闭 | 只读。有未关闭事项时才会运行 |
| `check.downloads` | 下载诊断 | Agent | 2 小时,默认关闭 | 默认只读。打开 `allowRecover` 后,在授权范围内重试或重新校验 | | `check.downloads` | 下载诊断 | Agent | 2 小时,默认关闭 | 默认只读。打开 `allowRecover` 后,在授权范围内重试或重新校验 |
+149
View File
@@ -0,0 +1,149 @@
# 存储预算与保留(CACHE-2)
按 4C / 8G / 100G 的机器分配:PG ≤ 25 GB(20 GB 告警)、图片缓存 20 GB、备份 20 GB(14 份)、会话与证据 10 GB。
流水数据按保留期定时分批删除,缓存有效期延长,容量不够时按 LRU 淘汰。本轮只有后端与配置,设置页 UI 后续接。
## 设置 `settings.storage`
三组,都在「设置 → 自动化 → 存储与保留」(分区锚点 `storage`,UI 待接)。PATCH 可以只给子字段:
```sh
PATCH /api/v1/settings {"storage":{"retentionDays":{"observation":60}}}
```
| 字段 | 缺省 | 说明 |
|---|---|---|
| `budgets.pgDataGb` / `pgDataWarnGb` | 25 / 20 | PG 库体积(`pg_database_size`,不含 WAL)。≥ 告警线 warn,≥ 预算 critical |
| `budgets.imagesGb` | 20 | 图片代理缓存容量。超出由 `cache.images_lru` 按最久未访问删到 90% |
| `budgets.backupsGb` | 20 | `DATA_DIR/backups`。≥ 80% warn,≥ 预算 critical。份数仍由 `BACKUP_KEEP` 控制 |
| `budgets.sessionsEvidenceGb` | 10 | Agent 会话目录 + `DATA_DIR/tmp/evidence`。≥ 80% warn,≥ 预算 critical |
| `retentionDays.observation` | 90 | Feed 观察记录 |
| `retentionDays.sseEvent` | 7 | SSE 补读事件(发布方另按条数裁剪 10000 条) |
| `retentionDays.heartbeatHours` | 24 | 进程心跳采样(小时)。历史曲线最多只能看这么久 |
| `retentionDays.taskRun` | 60 | 已结束的业务任务 `task`、引擎 run(步骤级联)、运行元数据 |
| `retentionDays.agentEvent` | 90 | 已结束会话的事件流与已消费消息。`agent_run` 本身保留 |
| `retentionDays.audit` | 180 | 审计事件(能力调用审计除外) |
| `retentionDays.capabilityStats` | 90 | 能力调用审计(`target_type=capability`)与已结束的 `capability_invocation` |
| `retentionDays.identityPoolArchive` | 30 | 待识别池 unmatched 条目无新证据多少天后归档(不删) |
| `cacheTtl.tmdbSearchDays` | 7 | TMDB `/search/*`、`/find/*` 响应与识别缓存 |
| `cacheTtl.tmdbDetailDays` | 30 | TMDB 详情 / 季。追踪中的作品命中时续期 |
| `cacheTtl.tmdbMissDays` | 1 | TMDB 空结果与未识别负缓存 |
| `cacheTtl.siteDetailDays` | 7 | 站点详情页扩展内容(`kv_cache` `site.detail`) |
| `cacheTtl.webSearchDays` | 1 | 网页搜索结果 |
被取代的旧字段(保留在 schema 里兼容已存 JSON,服务端不再读取,已列入 `DEPRECATED_SETTINGS_FIELDS`):
`cache.tmdbTtlHours` → `cacheTtl.tmdbSearchDays / tmdbDetailDays`,`cache.emptyResultTtlSec` → `cacheTtl.tmdbMissDays`,
`identity.poolArchiveDays` → `retentionDays.identityPoolArchive`。设置页「元数据」里这三项接 UI 时要换成新字段。
`cache.tmdbMaxSize`(条数上限,缺省 256)仍然有效。TTL 拉长后条数上限会先起作用,需要时调大。
## 缓存有效期
只改 TTL 的取值来源,缓存语义还是对照 MP(见 [porting/moviepilot-tmdb-cache.md](../porting/moviepilot-tmdb-cache.md))。
- TMDB 请求缓存:键以 `/search/` 或 `/find/` 开头的用搜索 TTL,其余用详情 TTL。`results` 为空的用未命中 TTL。
- TMDB 识别缓存:正常条目用搜索 TTL(识别结果就是名称到条目的映射),负缓存 `{id:0}` 用未命中 TTL。
- 追踪中的作品续期:作品有未结束的订阅(active / paused / pending)且有已接受的 TMDB 身份,
它的详情键(`/movie/<id>?…`、`/tv/<id>…`)命中时把过期时间推到「现在 + 详情 TTL」,只延长不缩短。追踪集合每进程缓存 5 分钟。
- 站点详情:`kv_cache` TTL 取 `max(站点详情缓存期, siteDetailDays)`。是否重抓仍按站点详情缓存期
(`sites.defaultDetailCacheSec`,站点定义可更长)判断,下载前复核促销和 H&R 走的是这条。
- 改了 TTL:新写入的条目按新值,已有条目保持原过期时间。
## 图片缓存容量 LRU
- 读盘命中时把文件 atime 显式写成当前时间(`utimes`,不依赖挂载的 relatime),mtime 不动。
- `cache.images_lru`(维护组,30 分钟):总量超过 `imagesGb` 时按 atime 从旧到新删到预算的 90%;没超不动。
- `cache.images_clean`(每天,MP `clear_temp`)照旧按 mtime 删超过 `cache.globalImageCacheDays` 天的文件,作兜底。
- 存储面板里图片超过预算 10% 才告警(说明 LRU 没跟上),不给 critical。
## 保留清理任务
都在维护组,开关和周期在定时任务看板改。首次登记在一个周期后才第一次运行。
| key | 目标 | 默认周期 | 删除条件 |
|---|---|---|---|
| `storage.retention.observation` | `observation` | 6 小时 | `created_at` 早于保留期 |
| `storage.retention.sse_event` | `sse_event` | 1 小时 | `at` 早于保留期 |
| `system.heartbeat_cleanup` | 心跳 | 1 小时 | 进程行 24 小时无心跳;采样早于 `heartbeatHours` |
| `storage.retention.task_run` | `task`、引擎 `workflow_runs`、`tasks_run_meta` / `tasks_fact` / `tasks_op_ledger` | 6 小时 | 早于保留期且已结束(task:succeeded / failed / cancelled / dead;run:completed / succeeded / failed / canceled;账本只删 succeeded) |
| `storage.retention.agent_event` | `agent_run_event`、`agent_run_message` | 6 小时 | 早于保留期且会话已结束;消息只删已消费的 |
| `storage.retention.audit` | `audit_event` | 1 天 | `at` 早于保留期,`target_type` 不是 capability |
| `storage.retention.capability_stats` | `audit_event`(capability)、`capability_invocation` | 1 天 | 早于保留期;调用记录不删 running |
| `storage.retention.identity_pool` | `identity_candidate` | 6 小时 | pending + unmatched,`coalesce(last_attempt_at, created_at)` 早于保留期 → 改为 archived |
| `storage.analyze` | 全库 | 7 天 | `ANALYZE` |
| `cache.images_lru` | 图片缓存 | 30 分钟 | 见上 |
| `cache.sweep` | `kv_cache` | 10 分钟 | 已过期 |
分批删除(`modules/system/retention.ts` 的 `deleteInBatches`):
- 每批 5000 行:`delete … where ctid = any(array(select ctid … where <条件> limit 5000))`,单独事务。
- 每批 `statement_timeout` 15 秒、`lock_timeout` 3 秒。超时就结束本次运行,不算失败,摘要写「未删完,下次继续」。
- 每次运行最多 5 分钟,批间停 50 ms。第一次上线积压多时会分几次删完。
- 删除后空间由 autovacuum 回收复用,不会立即还给文件系统。确实要缩文件用 `pg_repack` 或停机 `VACUUM FULL`。
迁移 `0050_storage_retention`:给 `observation(created_at)`、`sse_event(at)`、`audit_event(at)`、`capability_invocation(created_at)`、
`agent_run_message(created_at)`、`tasks_fact(created_at)`、`tasks_run_meta(created_at)`、`tasks_op_ledger(started_at)` 补索引;
按表设 autovacuum:
- 追加加批量删除的大表(observation、sse_event、audit_event、agent_run_event、process_heartbeat_sample、capability_invocation、task、tasks_fact、tasks_run_meta):
`vacuum_scale_factor=0.05`、`analyze_scale_factor=0.02`、`vacuum_insert_scale_factor=0.1`。
- 频繁原地更新的小表(process_heartbeat、rate_limit_bucket、scheduled_job、kv_cache、outbox_event):
`vacuum_scale_factor=0.02`、`vacuum_threshold=100`、`analyze_scale_factor=0.05`。
查看:`select relname, reloptions from pg_class where reloptions is not null;`
## API
### GET /system/storage(admin)
| 字段 | 含义 |
|---|---|
| `level` | 各项最差档:ok / warn / critical,取不到的项 unknown 不参与 |
| `db` | 库体积对照 `pgDataGb` / `pgDataWarnGb`;`tables` 为体积前 15 的表(总 / 表 / 索引字节、估算行数、死元组、最近 autovacuum / analyze) |
| `kvCache[]` | 每个 region 的条数、其中已过期未清理的条数、字节 |
| `images` | 图片缓存用量、文件数、预算 |
| `backups` | 备份目录用量、份数(不含 `.partial`)、预算 |
| `sessionsEvidence` | 会话目录 + `tmp/evidence` 合计、预算 |
| `dataDir[]` | DATA_DIR 一级目录用量,`cache/`、`tmp/` 再细分一层;会话目录在 DATA_DIR 外时单列 |
| `dirsScannedAt` | 目录遍历时间。遍历结果缓存 5 分钟 |
| `cleanup[]` | 每个清理目标:任务 key、保留期说明、开关、最近运行时间 / 状态 / 摘要、下次运行 |
### POST /system/storage/cleanup/{target}(admin)
`target`:`observation`、`sse_event`、`heartbeat`、`task_run`、`agent_event`、`audit`、`capability_stats`、`identity_pool`、`analyze`、`images`、`kv_cache`。
和定时任务「立即执行」一样,只设置 `run_requested_at`,由调度器执行。已在排队或运行中返回 `queued:false`。
任务还没登记(调度器没同步过)返回 404。触发成功写审计 `storage.cleanup`。
### GET /system/health 的 `storage` 段
`{ level, alerts[] }`,level 计入总档位。库体积实时取;目录用量只用缓存值,过期时后台刷新,本次不等
(进程刚启动时目录项是 unknown,不参与汇总)。
## PG 参数
`docker-compose.yml` 的 postgres 服务按 4C / 8G 给 PG 3G 内存:
| 参数 | 缺省 | 环境变量 | 经验值 |
|---|---|---|---|
| 容器内存上限 | 3g | `PG_MEM_LIMIT` | 机器内存的 35–40% |
| `shared_buffers` | 2GB | `PG_SHARED_BUFFERS` | 容器上限的 60–70%(不超过机器内存的 25–30%) |
| `effective_cache_size` | 5GB | `PG_EFFECTIVE_CACHE_SIZE` | 机器内存的 60–70%(只影响规划,不占内存) |
| `work_mem` | 32MB | `PG_WORK_MEM` | 每个排序 / 哈希一份,连接多时调小 |
| `maintenance_work_mem` | 256MB | `PG_MAINTENANCE_WORK_MEM` | VACUUM / 建索引用 |
| `max_wal_size` | 2GB | `PG_MAX_WAL_SIZE` | 批量删除时减少检查点 |
另外 `shm_size: 512mb`(并行查询的动态共享内存走 /dev/shm)。
按机器调整,例如 2C/4G:`PG_MEM_LIMIT=1536m PG_SHARED_BUFFERS=768MB PG_EFFECTIVE_CACHE_SIZE=2GB PG_WORK_MEM=16MB PG_MAINTENANCE_WORK_MEM=128MB`;
8C/16G:`PG_MEM_LIMIT=6g PG_SHARED_BUFFERS=4GB PG_EFFECTIVE_CACHE_SIZE=10GB PG_WORK_MEM=64MB PG_MAINTENANCE_WORK_MEM=512MB`。
写进 `.env` 后 `docker compose up -d postgres` 重建容器生效(数据在卷里,不丢)。确认:`docker compose exec postgres psql -U umt -c 'show shared_buffers'`。
`shared_buffers` 大于容器上限会在启动时 OOM,两者一起改。
## 常见问题
- 面板上 PG 超预算:先看 `tables` 前几名。流水表大就临时调短对应保留期,再 `POST /system/storage/cleanup/<target>`;
死元组多说明 autovacuum 跟不上,可手动 `VACUUM (VERBOSE) <表>`。
- 清理摘要一直是「未删完」:积压太多,每次 5 分钟删不完。可以把任务周期临时调短,或在低峰期多触发几次。
- 图片缓存一直 warn:LRU 任务是否启用、最近一次是否失败;磁盘上是否有大量 `.tmp` 残留(不计入用量,可手动删)。
- 历史曲线只能看 24 小时:这是 `heartbeatHours` 的缺省,要看更久就调大。
+1 -1
View File
@@ -34,7 +34,7 @@ curl -X POST -H "authorization: Bearer $KEY" -H 'content-type: application/json'
- 请求 `GET https://api.search.brave.com/res/v1/web/search`,头 `X-Subscription-Token`,参数 q / count / country / search_lang / safesearch / freshness,超时 10 s。 - 请求 `GET https://api.search.brave.com/res/v1/web/search`,头 `X-Subscription-Token`,参数 q / count / country / search_lang / safesearch / freshness,超时 10 s。
- 错误分类:401/403 → auth,402 → quota,429 → rate(读 `X-RateLimit-Reset` / `Retry-After`),400/422 → bad_request,5xx → upstream,连接失败 → network,超时 → timeout。错误信息里不带 Key。 - 错误分类:401/403 → auth,402 → quota,429 → rate(读 `X-RateLimit-Reset` / `Retry-After`),400/422 → bad_request,5xx → upstream,连接失败 → network,超时 → timeout。错误信息里不带 Key。
- 结果归一化为 `{ title, url, snippet, age, source }`:去 HTML 标签、解码实体、脱敏(URL 参数里的 passkey / token 等换成 `[REDACTED]`)、截断(标题 200 字、摘要 500 字),丢弃非 http(s) 链接。 - 结果归一化为 `{ title, url, snippet, age, source }`:去 HTML 标签、解码实体、脱敏(URL 参数里的 passkey / token 等换成 `[REDACTED]`)、截断(标题 200 字、摘要 500 字),丢弃非 http(s) 链接。
- 缓存:`kv_cache` region `websearch`,key 是 (provider, 查询词, 条数, 地区, 语言, 安全搜索, freshness) 的 sha256,TTL 10 分钟,上限 1000 条(TLRU,同 MP 语义)。过期行由 `cache.sweep` 清理。 - 缓存:`kv_cache` region `websearch`,key 是 (provider, 查询词, 条数, 地区, 语言, 安全搜索, freshness) 的 sha256,TTL 取 `storage.cacheTtl.webSearchDays`(缺省 1 天,CACHE-2 前是 10 分钟),上限 1000 条(TLRU,同 MP 语义)。过期行由 `cache.sweep` 清理。
- 输出 `{ trust: "untrusted", notice, query, results, cached }`,进入模型上下文前 Pi runtime 再整体过一遍 `redactForModel`。 - 输出 `{ trust: "untrusted", notice, query, results, cached }`,进入模型上下文前 Pi runtime 再整体过一遍 `redactForModel`。
## web.fetch ## web.fetch
+9 -4
View File
@@ -98,8 +98,11 @@ export interface TmdbConfig {
maxRetryAfterMs?: number; maxRetryAfterMs?: number;
maxConcurrent?: number; maxConcurrent?: number;
minIntervalMs?: number; minIntervalMs?: number;
/** 默认 max 1024、ttl 24h;空结果(搜索 0 条)用 emptyTtlMs,默认 30min(对照 MP EMPTY_RESULT_CACHE_TTL) */ /**
cache?: { max?: number; ttlMs?: number; emptyTtlMs?: number }; * 默认 max 1024、ttl 24h;空结果(搜索 0 条)用 emptyTtlMs,默认 30min(对照 MP EMPTY_RESULT_CACHE_TTL);
* searchTtlMs:/search/、/find/ 请求的 TTL(缺省同 ttlMs),其余(详情 / 季)用 ttlMs。
*/
cache?: { max?: number; ttlMs?: number; searchTtlMs?: number; emptyTtlMs?: number };
/** /**
* 跨进程共享的请求缓存(服务端传 PG 缓存);给了就不再用进程内 LRU。 * 跨进程共享的请求缓存(服务端传 PG 缓存);给了就不再用进程内 LRU。
* 语义对照 MP @cached(skip_none、业务失败不缓存、results 为空按 emptyTtlMs)。 * 语义对照 MP @cached(skip_none、业务失败不缓存、results 为空按 emptyTtlMs)。
@@ -164,6 +167,8 @@ export class TmdbClient {
for (const k of Object.keys(params).sort()) if (params[k] !== undefined) qs.set(k, params[k]!); for (const k of Object.keys(params).sort()) if (params[k] !== undefined) qs.set(k, params[k]!);
const key = `${path}?${qs}`; const key = `${path}?${qs}`;
const emptyTtl = this.cfg.cache?.emptyTtlMs ?? 30 * 60_000; const emptyTtl = this.cfg.cache?.emptyTtlMs ?? 30 * 60_000;
const baseTtl = this.cfg.cache?.ttlMs ?? 24 * 3600_000;
const ttl = /^\/(search|find)\//.test(path) ? (this.cfg.cache?.searchTtlMs ?? baseTtl) : baseTtl;
if (this.cfg.sharedCache) { if (this.cfg.sharedCache) {
const shared = this.cfg.sharedCache; const shared = this.cfg.sharedCache;
const hit = await shared.get(key); const hit = await shared.get(key);
@@ -171,14 +176,14 @@ export class TmdbClient {
// 网络错误 / 业务失败在 fetchWithRetry 抛出,不会走到写缓存(MP skip_none / skip_if) // 网络错误 / 业务失败在 fetchWithRetry 抛出,不会走到写缓存(MP skip_none / skip_if)
const v = await this.limiter.run(() => this.fetchWithRetry(path, qs)); const v = await this.limiter.run(() => this.fetchWithRetry(path, qs));
if (v !== null && v !== undefined) { if (v !== null && v !== undefined) {
await shared.set(key, v, isEmptyResult(v) ? emptyTtl : (this.cfg.cache?.ttlMs ?? 24 * 3600_000)); await shared.set(key, v, isEmptyResult(v) ? emptyTtl : ttl);
} }
return v as T; return v as T;
} }
return this.cache.getOrLoad( return this.cache.getOrLoad(
key, key,
() => this.limiter.run(() => this.fetchWithRetry(path, qs)), () => this.limiter.run(() => this.fetchWithRetry(path, qs)),
(v) => (isEmptyResult(v) ? emptyTtl : undefined), (v) => (isEmptyResult(v) ? emptyTtl : ttl === baseTtl ? undefined : ttl),
) as Promise<T>; ) as Promise<T>;
} }
+15
View File
@@ -2017,6 +2017,21 @@ export const Api = {
query: ops.SystemHealthHistoryQuery, query: ops.SystemHealthHistoryQuery,
response: ops.SystemHealthHistory, response: ops.SystemHealthHistory,
}), }),
storage: ep({
method: "get",
path: "/system/storage",
scope: "admin",
summary: "存储:表体积前 15、库总体积、各缓存用量与预算、备份占用、DATA_DIR 分目录用量、清理任务状态",
response: ops.SystemStorage,
}),
storageCleanup: ep({
method: "post",
path: "/system/storage/cleanup/{target}",
scope: "admin",
summary: "立即执行某个存储清理任务(排入对应定时任务,已在排队或运行中则合并)",
params: ops.StorageCleanupParam,
response: ops.StorageCleanupResult,
}),
}, },
// ── 引擎面板(仅 admin) ── // ── 引擎面板(仅 admin) ──
+101
View File
@@ -481,9 +481,110 @@ export const SystemHealth = z.object({
), ),
failedRuns24h: NonNegInt, failedRuns24h: NonNegInt,
}), }),
/** 存储预算(CACHE-2):PG 库体积、图片缓存、备份、会话与证据对照 settings.storage.budgets;明细见 GET /system/storage */
storage: z.object({
level: HealthLevel,
/** 超出告警线 / 预算的项(人读文案) */
alerts: z.array(z.string()),
}),
}); });
export type SystemHealth = z.infer<typeof SystemHealth>; export type SystemHealth = z.infer<typeof SystemHealth>;
// ── 存储预算与保留(CACHE-2;GET /system/storage、POST /system/storage/cleanup/{target},仅 admin) ──
/** 手动触发的清理目标;每个对应一个维护组定时任务 */
export const StorageCleanupTarget = z.enum([
"observation",
"sse_event",
"heartbeat",
"task_run",
"agent_event",
"audit",
"capability_stats",
"identity_pool",
"analyze",
"images",
"kv_cache",
]);
export type StorageCleanupTarget = z.infer<typeof StorageCleanupTarget>;
export const StorageCleanupParam = z.object({ target: StorageCleanupTarget });
export const StorageCleanupResult = z.object({
target: StorageCleanupTarget,
/** 对应的定时任务 key */
jobId: z.string(),
/** false = 已在排队或运行中(请求被合并) */
queued: z.boolean(),
});
export type StorageCleanupResult = z.infer<typeof StorageCleanupResult>;
/** 一项占用对照预算 */
export const StorageBudgetUsage = z.object({
usedBytes: NonNegInt.nullable(),
/** 预算(critical 线);无预算为 null */
budgetBytes: NonNegInt.nullable(),
/** 告警线;无单独告警线时为预算的 80% */
warnBytes: NonNegInt.nullable(),
level: HealthLevel,
error: z.string().nullable(),
});
export type StorageBudgetUsage = z.infer<typeof StorageBudgetUsage>;
export const SystemStorage = z.object({
/** 各项最差档(unknown 不参与) */
level: HealthLevel,
checkedAt: Timestamp,
/** 目录用量的统计时间(目录遍历结果缓存 5 分钟) */
dirsScannedAt: Timestamp.nullable(),
db: StorageBudgetUsage.extend({
/** 体积前 15 的表(含索引与 TOAST) */
tables: z.array(
z.object({
schema: z.string(),
name: z.string(),
totalBytes: NonNegInt,
tableBytes: NonNegInt,
indexBytes: NonNegInt,
/** 估算行数(pg_class.reltuples);未 ANALYZE 为 null */
rowsEstimate: NonNegInt.nullable(),
deadRows: NonNegInt.nullable(),
lastAutovacuumAt: Timestamp.nullable(),
lastAnalyzeAt: Timestamp.nullable(),
}),
),
}),
/** PG kv_cache 各 region(条数含未清理的过期行) */
kvCache: z.array(z.object({ region: z.string(), entries: NonNegInt, expired: NonNegInt, bytes: NonNegInt })),
images: StorageBudgetUsage.extend({ files: NonNegInt.nullable() }),
backups: StorageBudgetUsage.extend({ count: NonNegInt.nullable() }),
sessionsEvidence: StorageBudgetUsage,
/** DATA_DIR 下各一级目录(cache 再细分一层)的用量 */
dataDir: z.array(
z.object({
name: z.string(),
path: z.string(),
usedBytes: NonNegInt,
files: NonNegInt,
error: z.string().nullable(),
}),
),
/** 各清理任务的当前设置与最近运行 */
cleanup: z.array(
z.object({
target: StorageCleanupTarget,
jobId: z.string(),
/** 保留期说明(如 "90 天"、"24 小时"、"容量 20 GB") */
retention: z.string().nullable(),
/** 定时任务不存在(未登记 / 调度器未同步)时为 null */
enabled: z.boolean().nullable(),
lastRunAt: Timestamp.nullable(),
lastStatus: z.string().nullable(),
lastSummary: z.string().nullable(),
nextRunAt: Timestamp.nullable(),
}),
),
});
export type SystemStorage = z.infer<typeof SystemStorage>;
export const SystemHealthHistoryQuery = z.object({ export const SystemHealthHistoryQuery = z.object({
/** 回看小时数,1–168,缺省 6 */ /** 回看小时数,1–168,缺省 6 */
hours: z.coerce.number().int().min(1).max(168).default(6), hours: z.coerce.number().int().min(1).max(168).default(6),
+125 -6
View File
@@ -24,11 +24,14 @@ const CacheShape = z.object({
checkIntervalSec: z.number().int().min(10).max(3600), checkIntervalSec: z.number().int().min(10).max(3600),
/** 缓存清理的目标使用率(低水位) */ /** 缓存清理的目标使用率(低水位) */
cleanupTargetPercent: Percent, cleanupTargetPercent: Percent,
/** TMDB 缓存(请求响应 + 识别结果)保留小时数(对照 MP META_CACHE_EXPIRE,缺省 24) */ /**
* @deprecated CACHE-2 起由 storage.cacheTtl(tmdbSearchDays / tmdbDetailDays)取代,服务端不再读取;
* 保留在 schema 里只为兼容已存的 JSON 与旧设置页。
*/
tmdbTtlHours: z.number().int().min(1).max(720), tmdbTtlHours: z.number().int().min(1).max(720),
/** TMDB 缓存每类条数上限,超出删最旧(对照 MP CONF.tmdb,缺省 256;大内存模式 1024) */ /** TMDB 缓存每类条数上限,超出删最旧(对照 MP CONF.tmdb,缺省 256;大内存模式 1024) */
tmdbMaxSize: z.number().int().min(16).max(100_000), tmdbMaxSize: z.number().int().min(16).max(100_000),
/** 空结果 / 未识别的短缓存秒数(对照 MP EMPTY_RESULT_CACHE_TTL,缺省 1800) */ /** @deprecated CACHE-2 起由 storage.cacheTtl.tmdbMissDays 取代,服务端不再读取 */
emptyResultTtlSec: z.number().int().min(0).max(86_400), emptyResultTtlSec: z.number().int().min(0).max(86_400),
/** 图片代理落盘缓存(对照 MP GLOBAL_IMAGE_CACHE,缺省关闭;关闭时仍代理,只是不落盘) */ /** 图片代理落盘缓存(对照 MP GLOBAL_IMAGE_CACHE,缺省关闭;关闭时仍代理,只是不落盘) */
globalImageCache: z.boolean(), globalImageCache: z.boolean(),
@@ -410,7 +413,7 @@ export const DEFAULT_ENRICH_DOMAINS: string[] = [
export const IdentitySettings = z.object({ export const IdentitySettings = z.object({
eagerTmdbLookup: z.boolean(), eagerTmdbLookup: z.boolean(),
/** 待识别池里 unmatched 条目多少天没有新证据就归档(仍可按状态筛出、可手动指定) */ /** @deprecated CACHE-2 起由 storage.retentionDays.identityPoolArchive 取代,服务端不再读取 */
poolArchiveDays: z.number().int().min(1).max(3650), poolArchiveDays: z.number().int().min(1).max(3650),
/** /**
* META-1:订阅别名 / 新作品收录规则命中、TMDB 却查不到时自动尝试一次元数据补全(只出建议,不建作品)。 * META-1:订阅别名 / 新作品收录规则命中、TMDB 却查不到时自动尝试一次元数据补全(只出建议,不建作品)。
@@ -432,6 +435,98 @@ export const IdentitySettings = z.object({
}); });
export type IdentitySettings = z.infer<typeof IdentitySettings>; export type IdentitySettings = z.infer<typeof IdentitySettings>;
// ── 存储预算与保留(CACHE-2,docs/runbook/storage.md)──
const Gb = z.number().min(0.01).max(10_000);
const Days = z.number().int().min(1).max(3650);
export const StorageBudgets = z
.object({
/** PG 数据目录预算(GB):超过 pgDataWarnGb 告警,超过本值 critical */
pgDataGb: Gb,
pgDataWarnGb: Gb,
/** 图片代理缓存容量(GB):超出按最久未访问删除(LRU) */
imagesGb: Gb,
/** DATA_DIR/backups 占用预算(GB) */
backupsGb: Gb,
/** Agent 会话 + 视觉证据(DATA_DIR/agent-sessions、tmp/evidence)预算(GB) */
sessionsEvidenceGb: Gb,
})
.refine((v) => v.pgDataWarnGb <= v.pgDataGb, { message: "PG 告警线不得高于预算" });
export type StorageBudgets = z.infer<typeof StorageBudgets>;
/** 各类流水数据保留期(到期由 storage.retention.* 定时任务分批删除) */
export const StorageRetentionDays = z.object({
/** observation(Feed 观察) */
observation: Days,
/** sse_event(SSE 补读) */
sseEvent: Days,
/** process_heartbeat_sample(小时) */
heartbeatHours: z
.number()
.int()
.min(1)
.max(24 * 90),
/** 已结束的业务任务 task 与引擎 run */
taskRun: Days,
/** 已结束 Agent 会话的 agent_run_event / agent_run_message */
agentEvent: Days,
/** audit_event(能力调用审计除外) */
audit: Days,
/** 能力调用统计:audit_event(target_type=capability)与已结束的 capability_invocation */
capabilityStats: Days,
/** 待识别池 unmatched 条目无新证据多少天后归档 */
identityPoolArchive: Days,
});
export type StorageRetentionDays = z.infer<typeof StorageRetentionDays>;
/** 缓存有效期(天):只改 TTL 的取值来源,缓存语义仍对照 MP(docs/porting/moviepilot-tmdb-cache.md) */
export const StorageCacheTtl = z.object({
/** TMDB 搜索 / find 响应与识别结果 */
tmdbSearchDays: Days,
/** TMDB 详情(/movie/{id}、/tv/{id}、季);追踪中的作品命中时续期 */
tmdbDetailDays: Days,
/** TMDB 空结果与未识别负缓存(MP EMPTY_RESULT_CACHE_TTL) */
tmdbMissDays: Days,
/** 站点详情页扩展内容(kv_cache site.detail) */
siteDetailDays: Days,
/** 网页搜索结果 */
webSearchDays: Days,
});
export type StorageCacheTtl = z.infer<typeof StorageCacheTtl>;
export const StorageSettings = z.object({
budgets: StorageBudgets,
retentionDays: StorageRetentionDays,
cacheTtl: StorageCacheTtl,
});
export type StorageSettings = z.infer<typeof StorageSettings>;
export const DEFAULT_STORAGE_SETTINGS: StorageSettings = {
budgets: { pgDataGb: 25, pgDataWarnGb: 20, imagesGb: 20, backupsGb: 20, sessionsEvidenceGb: 10 },
retentionDays: {
observation: 90,
sseEvent: 7,
heartbeatHours: 24,
taskRun: 60,
agentEvent: 90,
audit: 180,
capabilityStats: 90,
identityPoolArchive: 30,
},
cacheTtl: { tmdbSearchDays: 7, tmdbDetailDays: 30, tmdbMissDays: 1, siteDetailDays: 7, webSearchDays: 1 },
};
/** 存储设置与缺省逐层合并(旧库里缺子字段时补齐) */
export function withStorageDefaults(v: unknown): StorageSettings {
const o = (v && typeof v === "object" ? v : {}) as Partial<Record<keyof StorageSettings, object>>;
return {
budgets: { ...DEFAULT_STORAGE_SETTINGS.budgets, ...(o.budgets ?? {}) },
retentionDays: { ...DEFAULT_STORAGE_SETTINGS.retentionDays, ...(o.retentionDays ?? {}) },
cacheTtl: { ...DEFAULT_STORAGE_SETTINGS.cacheTtl, ...(o.cacheTtl ?? {}) },
} as StorageSettings;
}
export const GlobalSettings = z.object({ export const GlobalSettings = z.object({
automation: AutomationSettings, automation: AutomationSettings,
cache: CacheSettings, cache: CacheSettings,
@@ -445,6 +540,7 @@ export const GlobalSettings = z.object({
naming: NamingSettings, naming: NamingSettings,
identity: IdentitySettings, identity: IdentitySettings,
search: SearchSettings, search: SearchSettings,
storage: StorageSettings,
}); });
export type GlobalSettings = z.infer<typeof GlobalSettings>; export type GlobalSettings = z.infer<typeof GlobalSettings>;
@@ -526,6 +622,7 @@ export const DEFAULT_SETTINGS: GlobalSettings = {
enrichDomains: DEFAULT_ENRICH_DOMAINS, enrichDomains: DEFAULT_ENRICH_DOMAINS,
}, },
search: { defaultSiteMode: "default", siteIds: [] }, search: { defaultSiteMode: "default", siteIds: [] },
storage: DEFAULT_STORAGE_SETTINGS,
}; };
/** PATCH /settings:按分组部分更新;服务端与现值合并后用 GlobalSettings 整体再校验 */ /** PATCH /settings:按分组部分更新;服务端与现值合并后用 GlobalSettings 整体再校验 */
@@ -576,6 +673,13 @@ export const GlobalSettingsUpdate = z.object({
naming: NamingSettings.partial().optional(), naming: NamingSettings.partial().optional(),
identity: IdentitySettings.partial().optional(), identity: IdentitySettings.partial().optional(),
search: SearchSettings.partial().optional(), search: SearchSettings.partial().optional(),
storage: z
.object({
budgets: z.object(StorageBudgets.shape).partial().optional(),
retentionDays: StorageRetentionDays.partial().optional(),
cacheTtl: StorageCacheTtl.partial().optional(),
})
.optional(),
}); });
export type GlobalSettingsUpdate = z.infer<typeof GlobalSettingsUpdate>; export type GlobalSettingsUpdate = z.infer<typeof GlobalSettingsUpdate>;
@@ -602,6 +706,16 @@ export function mergeSettings(current: GlobalSettings, patch: GlobalSettingsUpda
naming: { ...(current.naming ?? DEFAULT_SETTINGS.naming), ...patch.naming }, naming: { ...(current.naming ?? DEFAULT_SETTINGS.naming), ...patch.naming },
identity: { ...(current.identity ?? DEFAULT_SETTINGS.identity), ...patch.identity }, identity: { ...(current.identity ?? DEFAULT_SETTINGS.identity), ...patch.identity },
search: { ...(current.search ?? DEFAULT_SETTINGS.search), ...patch.search }, search: { ...(current.search ?? DEFAULT_SETTINGS.search), ...patch.search },
storage: mergeStorage(current.storage, patch.storage),
};
}
function mergeStorage(current: StorageSettings | undefined, patch: GlobalSettingsUpdate["storage"]): StorageSettings {
const base = withStorageDefaults(current);
return {
budgets: { ...base.budgets, ...patch?.budgets },
retentionDays: { ...base.retentionDays, ...patch?.retentionDays },
cacheTtl: { ...base.cacheTtl, ...patch?.cacheTtl },
}; };
} }
@@ -638,6 +752,10 @@ export const DEPRECATED_SETTINGS_FIELDS = [
"pi.model", "pi.model",
"pi.thinkingLevel", "pi.thinkingLevel",
"pi.apiKeySecretRef", "pi.apiKeySecretRef",
// CACHE-2:由 storage.cacheTtl / storage.retentionDays 取代
"cache.tmdbTtlHours",
"cache.emptyResultTtlSec",
"identity.poolArchiveDays",
] as const; ] as const;
// ── 设置页分区(docs/design/console-ia.md §3.4) ── // ── 设置页分区(docs/design/console-ia.md §3.4) ──
@@ -682,6 +800,7 @@ export const SETTINGS_SECTIONS: readonly SettingsSectionDef[] = [
{ id: "cache", label: "缓存水位" }, { id: "cache", label: "缓存水位" },
{ id: "budget", label: "每日预算" }, { id: "budget", label: "每日预算" },
{ id: "retention", label: "保留" }, { id: "retention", label: "保留" },
{ id: "storage", label: "存储与保留" },
], ],
fields: [ fields: [
"automation.enabled", "automation.enabled",
@@ -695,6 +814,9 @@ export const SETTINGS_SECTIONS: readonly SettingsSectionDef[] = [
"budget.maxConcurrentDownloads", "budget.maxConcurrentDownloads",
"retention.seedRetentionDays", "retention.seedRetentionDays",
"retention.logRetentionDays", "retention.logRetentionDays",
"storage.budgets",
"storage.retentionDays",
"storage.cacheTtl",
], ],
resources: [], resources: [],
}, },
@@ -774,11 +896,8 @@ export const SETTINGS_SECTIONS: readonly SettingsSectionDef[] = [
fields: [ fields: [
"metadata.tmdbApiKeySecretRef", "metadata.tmdbApiKeySecretRef",
"metadata.tmdbBaseUrl", "metadata.tmdbBaseUrl",
"cache.tmdbTtlHours",
"cache.tmdbMaxSize", "cache.tmdbMaxSize",
"cache.emptyResultTtlSec",
"identity.eagerTmdbLookup", "identity.eagerTmdbLookup",
"identity.poolArchiveDays",
"identity.autoEnrich", "identity.autoEnrich",
"identity.enrichDailyBudget", "identity.enrichDailyBudget",
"identity.enrichDomains", "identity.enrichDomains",
@@ -0,0 +1,79 @@
import type { Kysely } from "kysely";
import { execDdl } from "../src/migration-helpers.ts";
// CACHE-2 存储保留(docs/runbook/storage.md):
// - 保留清理按时间列分批删除,给还没有时间索引的高流水表补上(已有:task_created_idx、agent_run_event_at_idx、
// process_heartbeat_sample_at_idx);
// - 按表设置 autovacuum:批量删除 / 追加为主的大表降低触发比例,频繁原地更新的小表更激进,避免死元组堆积。
// 引擎 schema(openworkflow)的表由引擎自己建,这里不动。
const INDEXES: [name: string, ddl: string][] = [
["observation_created_idx", "observation (created_at)"],
["sse_event_at_idx", "sse_event (at)"],
["audit_event_at_idx", "audit_event (at)"],
["capability_invocation_created_idx", "capability_invocation (created_at)"],
["agent_run_message_created_idx", "agent_run_message (created_at)"],
["tasks_fact_created_idx", "tasks_fact (created_at)"],
["tasks_run_meta_created_idx", "tasks_run_meta (created_at)"],
["tasks_op_ledger_started_idx", "tasks_op_ledger (started_at) where status = 'succeeded'"],
];
/** 追加 + 批量删除为主的大表 */
const APPEND_TABLES = [
"observation",
"sse_event",
"audit_event",
"agent_run_event",
"process_heartbeat_sample",
"capability_invocation",
"task",
"tasks_fact",
"tasks_run_meta",
];
const APPEND_OPTS = {
autovacuum_vacuum_scale_factor: "0.05",
autovacuum_analyze_scale_factor: "0.02",
autovacuum_vacuum_insert_scale_factor: "0.1",
};
/** 频繁原地更新的小表(心跳 15 秒一次、令牌桶、调度状态、缓存命中刷新 accessed_at) */
const HOT_TABLES = ["process_heartbeat", "rate_limit_bucket", "scheduled_job", "kv_cache", "outbox_event"];
const HOT_OPTS = {
autovacuum_vacuum_scale_factor: "0.02",
autovacuum_vacuum_threshold: "100",
autovacuum_analyze_scale_factor: "0.05",
};
const setOpts = (tables: string[], o: Record<string, string>) =>
tables
.map(
(t) =>
`alter table ${t} set (${Object.entries(o)
.map(([k, v]) => `${k} = ${v}`)
.join(", ")});`,
)
.join("\n");
const resetOpts = (tables: string[], o: Record<string, string>) =>
tables.map((t) => `alter table ${t} reset (${Object.keys(o).join(", ")});`).join("\n");
// biome-ignore lint/suspicious/noExplicitAny: 迁移不依赖应用表类型
export async function up(db: Kysely<any>): Promise<void> {
await execDdl(
db,
`${INDEXES.map(([n, d]) => `create index if not exists ${n} on ${d};`).join("\n")}
${setOpts(APPEND_TABLES, APPEND_OPTS)}
${setOpts(HOT_TABLES, HOT_OPTS)}
`,
);
}
// biome-ignore lint/suspicious/noExplicitAny: 同上
export async function down(db: Kysely<any>): Promise<void> {
await execDdl(
db,
`${resetOpts(HOT_TABLES, HOT_OPTS)}
${resetOpts(APPEND_TABLES, APPEND_OPTS)}
${INDEXES.map(([n]) => `drop index if exists ${n};`).join("\n")}
`,
);
}
+8
View File
@@ -58,6 +58,14 @@ export class PgCache {
return r.rows.length ? (r.rows[0]!.value as T) : null; return r.rows.length ? (r.rows[0]!.value as T) : null;
} }
/** 续期:未过期的条目把 expires_at 推到 now + ttl(只延长不缩短);返回是否有行被续期 */
async renew(region: string, key: string, ttl: number): Promise<boolean> {
if (ttl <= 0) return false;
const r = await sql`update kv_cache set expires_at = greatest(expires_at, now() + make_interval(secs => ${ttl}))
where region = ${region} and key = ${key} and expires_at > now()`.execute(this.db);
return Number(r.numAffectedRows ?? 0) > 0;
}
async exists(region: string, key: string): Promise<boolean> { async exists(region: string, key: string): Promise<boolean> {
const r = const r =
await sql`select 1 from kv_cache where region = ${region} and key = ${key} and expires_at > now()`.execute( await sql`select 1 from kv_cache where region = ${region} and key = ${key} and expires_at > now()`.execute(
+1 -1
View File
@@ -1,5 +1,5 @@
// 统一从本包取 Kysely 与 sql,避免各包各装一份 kysely // 统一从本包取 Kysely 与 sql,避免各包各装一份 kysely
export { type Kysely, type Selectable, sql, type Transaction } from "kysely"; export { type Kysely, type RawBuilder, type Selectable, sql, type Transaction } from "kysely";
export { export {
type CachedOptions, type CachedOptions,
cachedCall, cachedCall,