rootandClaude Fable 5.1
1262340856
- TaskEngine port + OpenWorkflowEngine ([email protected].1, schema openworkflow): enqueue/signal/cancel/get/list/startWorker/stop - defineTask registry with zod input schema; 9 stub task definitions (§3.1-8) - TaskContext: step.run/sleep/waitForFact, redacting logger, traceId, attemptToken, abortSignal driven by PG cancel flag - signal persists fact to tasks_fact first; waitForFact checks DB before waiting - OutboxRelay: FOR UPDATE SKIP LOCKED, idempotencyKey outbox:<id> - withAttemptToken fencing on tasks_op_ledger (own migration, merged in W2) - tests: kill -9 crash recovery (child worker), cancel, idempotency, concurrent relays, early fact, lease-loss fencing Co-Authored-By: Claude Fable 5.1 <[email protected]>
103 lines
3.4 KiB
TypeScript
103 lines
3.4 KiB
TypeScript
import { Kysely, PostgresDialect, sql } from "kysely";
|
|
import pg from "pg";
|
|
import { migrateTasksTables, OpenWorkflowEngine, silentSink, type TaskRegistry } from "../src";
|
|
|
|
export const PG_URL =
|
|
process.env.TASKS_TEST_DATABASE_URL ?? process.env.DATABASE_URL ?? "postgres://umt:umt@localhost:55433/umt";
|
|
|
|
/** 探测开发库;不可达时集成测试跳过(CI 机器应提供 55433) */
|
|
export async function pgAvailable(): Promise<boolean> {
|
|
const c = new pg.Client({ connectionString: PG_URL, connectionTimeoutMillis: 2000 });
|
|
try {
|
|
await c.connect();
|
|
await c.end();
|
|
return true;
|
|
} catch {
|
|
console.warn(`[tasks] PostgreSQL ${PG_URL} 不可达,跳过集成测试`);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// biome-ignore lint/suspicious/noExplicitAny: 测试直接用原生 SQL
|
|
export type TestDb = Kysely<any>;
|
|
|
|
/** 每个测试文件用独立 schema:业务表 <name>,引擎表 <name>_ow */
|
|
export function testSchemas(suffix: string) {
|
|
const biz = `tasks_test_${suffix}`;
|
|
return { biz, ow: `${biz}_ow` };
|
|
}
|
|
|
|
export function connectDb(schema: string): TestDb {
|
|
const pool = new pg.Pool({ connectionString: PG_URL, max: 6, options: `-c search_path=${schema}` });
|
|
return new Kysely({ dialect: new PostgresDialect({ pool }) });
|
|
}
|
|
|
|
export async function resetSchemas(schemas: { biz: string; ow: string }) {
|
|
const admin = connectDb("public");
|
|
try {
|
|
await sql.raw(`DROP SCHEMA IF EXISTS ${schemas.biz} CASCADE`).execute(admin);
|
|
await sql.raw(`DROP SCHEMA IF EXISTS ${schemas.ow} CASCADE`).execute(admin);
|
|
await sql.raw(`CREATE SCHEMA ${schemas.biz}`).execute(admin);
|
|
} finally {
|
|
await admin.destroy();
|
|
}
|
|
}
|
|
|
|
export async function dropSchemas(schemas: { biz: string; ow: string }) {
|
|
const admin = connectDb("public");
|
|
try {
|
|
await sql.raw(`DROP SCHEMA IF EXISTS ${schemas.biz} CASCADE`).execute(admin);
|
|
await sql.raw(`DROP SCHEMA IF EXISTS ${schemas.ow} CASCADE`).execute(admin);
|
|
} finally {
|
|
await admin.destroy();
|
|
}
|
|
}
|
|
|
|
/** 与 packages/db 0001_init 的 outbox_event 同构 */
|
|
export async function createOutboxTable(db: TestDb) {
|
|
await sql`CREATE TABLE outbox_event (
|
|
id uuid PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
at timestamptz NOT NULL DEFAULT now(),
|
|
type text NOT NULL,
|
|
payload jsonb NOT NULL DEFAULT '{}'::jsonb,
|
|
published_at timestamptz)`.execute(db);
|
|
}
|
|
|
|
export async function createSideEffectTable(db: TestDb) {
|
|
await sql`CREATE TABLE IF NOT EXISTS side_effect (
|
|
id bigserial PRIMARY KEY, run_id text NOT NULL, step text NOT NULL, note text, pid int,
|
|
at timestamptz NOT NULL DEFAULT now())`.execute(db);
|
|
}
|
|
|
|
export async function sideEffects(db: TestDb, runId: string) {
|
|
const r = await sql<{ step: string; note: string | null; pid: number }>`
|
|
SELECT step, note, pid FROM side_effect WHERE run_id = ${runId} ORDER BY id`.execute(db);
|
|
return r.rows;
|
|
}
|
|
|
|
export async function setupEngine(db: TestDb, schemas: { ow: string }, registry: TaskRegistry) {
|
|
await migrateTasksTables(db);
|
|
return OpenWorkflowEngine.connect({
|
|
databaseUrl: PG_URL,
|
|
db,
|
|
registry,
|
|
schema: schemas.ow,
|
|
logSink: silentSink,
|
|
cancelPollMs: 200,
|
|
});
|
|
}
|
|
|
|
export async function waitFor<T>(
|
|
fn: () => Promise<T | null | undefined | false>,
|
|
timeoutMs: number,
|
|
label = "condition",
|
|
) {
|
|
const deadline = Date.now() + timeoutMs;
|
|
for (;;) {
|
|
const v = await fn();
|
|
if (v) return v;
|
|
if (Date.now() > deadline) throw new Error(`timeout waiting for ${label}`);
|
|
await new Promise((r) => setTimeout(r, 200));
|
|
}
|
|
}
|