Split task execution: run preparation, provider cooldowns, heartbeat

Pure move to keep task-execution-service under 500 lines.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
ZacharyZhang-NYandClaude Opus 5.5 committed 2026-10-08 05:27:14 -04:00
1 parent 95e607bea7
commit 12e2d9dcd5
4 files changed
+381 -329

No files matched your search

+77
View File
@@ -0,0 +1,77 @@
/**
* How long a provider's tasks stay parked after it reports a transient
* failure (usage limit, rate limit, overload). Long enough not to hammer a
* closed door every 5s tick, short enough to probe again the same hour.
*/
const PROVIDER_TRANSIENT_FAILURE_COOLDOWN_MS = 10 * 60_000;
/** In-memory parks for providers that reported a transient failure. */
export class ProviderCooldowns {
private readonly parks = new Map<string, { until: number; label: string; interactiveProbedAt?: number }>();
constructor(private readonly now: () => number) {}
/**
* Parks a provider after it reported a transient failure (usage limit, rate
* limit, overload), so the queue stops feeding it doomed runs every tick.
* The park is in-memory by design: after a restart the first probe run
* either succeeds or re-arms the cooldown at the cost of one request.
*/
park(provider: { id: string; label: string }, message: string): void {
const active = this.parks.get(provider.id);
const wasParked = active !== undefined && active.until > this.now();
this.parks.set(provider.id, {
until: this.now() + PROVIDER_TRANSIENT_FAILURE_COOLDOWN_MS,
label: provider.label,
// A failed probe must not unlock endless probes: re-arming the park
// keeps the spent-probe mark until the window expires naturally.
interactiveProbedAt: wasParked ? active?.interactiveProbedAt : undefined,
});
if (!wasParked) {
const minutes = Math.round(PROVIDER_TRANSIENT_FAILURE_COOLDOWN_MS / 60_000);
console.warn(`[task-queue] provider "${provider.label}" reported a transient failure ("${message}"); its tasks are parked for ${minutes} minutes.`);
}
}
/**
* Providers currently inside a transient-failure cooldown. Expired parks are
* pruned (with a resume log line) so the next tick probes the provider again.
*/
cooledProviderIds(): string[] {
const active: string[] = [];
for (const [id, park] of this.parks) {
if (park.until > this.now()) {
active.push(id);
} else {
this.parks.delete(id);
console.info(`[task-queue] provider "${park.label}" cooldown ended; its tasks resume.`);
}
}
return active;
}
/**
* Cooled providers that even the interactive lane must skip: their one
* founder-probe for this park window is already spent. A cooled provider
* not yet probed stays interactively claimable — one human message is
* allowed through to learn whether the outage was a blip, instead of the
* founder sitting out the whole cooldown for a single failed background
* request.
*/
interactiveCooledProviderIds(): string[] {
return this.cooledProviderIds().filter((id) => this.parks.get(id)?.interactiveProbedAt !== undefined);
}
/** Marks the cooled provider's founder-probe as spent (no-op when healthy). */
noteInteractiveProbe(providerId: string): void {
const park = this.parks.get(providerId);
if (park) park.interactiveProbedAt = this.now();
}
/** A served request is proof of health: the park ends for everyone at once. */
clear(provider: { id: string; label: string }): void {
if (this.parks.delete(provider.id)) {
console.info(`[task-queue] provider "${provider.label}" served a request; cooldown lifted early.`);
}
}
}