In Next dev (Turbopack), each route compiles as its own entry — a module-level Map in turn-liveness.ts was a DIFFERENT instance in the page render, GET /api/chats/[id], and POST /api/chat. The chat route held the live turn (hence the 409), but the page render read an empty registry: initiallyInFlight=false → mount continuation fired a send → 409 → the 'Something went wrong' banner on returning mid-turn. The registry now lives on globalThis (the standard Next singleton channel) — one Map per process regardless of module instances. Semantics unchanged (TTL leak guard, begin/end round trip). Browser-verified end-to-end (headless tester): answer quiz → navigate away → return mid-turn → graded card (no re-prompt), 'Thinking…' shimmer, composer gated, no error banner; teacher response auto-delivered when the background turn finished. Polling GET /api/chats/[id] observed working.
60 lines
2.3 KiB
TypeScript
60 lines
2.3 KiB
TypeScript
/**
|
|
* Server-side turn liveness: which chats currently have a teacher turn in
|
|
* flight. The missing half of "sessions keep running in the background" —
|
|
* together with the write-ahead persist in chat-handler, a returning client
|
|
* can tell "the teacher is mid-turn — don't resurrect, don't start anything"
|
|
* apart from "the record IS current".
|
|
*
|
|
* Lives on globalThis, NOT a module-level Map: Next dev compiles each route
|
|
* as its own entry, so a module-level singleton is a DIFFERENT instance in
|
|
* the page render, GET /api/chats/[id], and POST /api/chat — the page would
|
|
* read an empty registry while the chat route holds the live turn (the bug:
|
|
* returning mid-turn showed the 409 banner instead of the busy state).
|
|
* globalThis spans module instances within the process; it is the standard
|
|
* Next.js singleton channel (same reason Prisma clients live there).
|
|
*
|
|
* Still ephemeral BY DESIGN: a process restart clears it, so there are no
|
|
* stale flags to clean up — a crashed turn recovers via the write-ahead
|
|
* record + mount continuation on the next visit. The TTL is a leak guard for
|
|
* the pathological case where onFinish never fires (it runs even on aborts);
|
|
* chat turns are minutes at most, so 30 minutes is far on the safe side.
|
|
*/
|
|
|
|
interface LhGlobal {
|
|
__lhTurnRegistry?: Map<string, number>;
|
|
}
|
|
|
|
function registry(): Map<string, number> {
|
|
const g = globalThis as typeof globalThis & LhGlobal;
|
|
return (g.__lhTurnRegistry ??= new Map<string, number>());
|
|
}
|
|
|
|
const inFlight = registry();
|
|
const MAX_TURN_MS = 30 * 60_000;
|
|
|
|
/**
|
|
* Mark a turn as starting. Returns false if one is already running for the
|
|
* chat — the caller rejects the request (only one turn may own a chat).
|
|
*/
|
|
export function beginTurn(chatId: string): boolean {
|
|
if (isTurnRunning(chatId)) return false;
|
|
inFlight.set(chatId, Date.now());
|
|
return true;
|
|
}
|
|
|
|
/** Mark the chat's turn as finished (runs in onFinish, aborted or not). */
|
|
export function endTurn(chatId: string): void {
|
|
inFlight.delete(chatId);
|
|
}
|
|
|
|
/** True while the chat's turn is in flight; lazily evicts stale entries. */
|
|
export function isTurnRunning(chatId: string): boolean {
|
|
const since = inFlight.get(chatId);
|
|
if (since === undefined) return false;
|
|
if (Date.now() - since > MAX_TURN_MS) {
|
|
inFlight.delete(chatId);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|