@cydm/happy-elves 0.1.0-beta.491 → 0.1.0-beta.495
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/apps/cli/package.json +1 -1
- package/apps/daemon/dist/orchestration/remote-work-budget.d.ts +1 -0
- package/apps/daemon/dist/orchestration/remote-work-budget.js +51 -0
- package/apps/daemon/dist/orchestration/runner.js +116 -32
- package/apps/daemon/dist/orchestration/store.d.ts +9 -0
- package/apps/daemon/dist/orchestration/store.js +28 -5
- package/apps/daemon/dist/orchestration/watcher.d.ts +7 -1
- package/apps/daemon/dist/orchestration/watcher.js +78 -5
- package/apps/daemon/dist/relay/connection.js +27 -0
- package/apps/daemon/dist/relay/remote-command.js +4 -0
- package/apps/daemon/dist/runtime/headless-event-projector.d.ts +12 -1
- package/apps/daemon/dist/runtime/headless-event-projector.js +55 -5
- package/apps/daemon/dist/runtime/headless-history-reader.d.ts +1 -0
- package/apps/daemon/dist/runtime/headless-history-reader.js +17 -5
- package/apps/daemon/dist/session/native-history-refresh.js +15 -1
- package/apps/daemon/dist/session/runtime-truth.d.ts +1 -0
- package/apps/daemon/dist/session/runtime-truth.js +13 -1
- package/apps/daemon/package.json +1 -1
- package/apps/relay/dist/packet-store.js +93 -15
- package/apps/relay/dist/storage-worker.js +4 -1
- package/apps/relay/dist/websocket.js +75 -10
- package/package.json +1 -1
- package/packages/runtime/dist/index.d.ts +4 -0
- package/packages/shared/dist/protocol-schemas.d.ts +2 -0
- package/packages/shared/dist/protocol-schemas.js +2 -0
- package/packages/shared/dist/protocol-types.d.ts +2 -0
- package/packages/shared/dist/protocol.d.ts +2 -0
- package/packages/shared/dist/protocol.js +2 -0
package/apps/cli/package.json
CHANGED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export declare function scheduleRemoteOrchestrationWork<T>(machineId: string, run: () => Promise<T>): Promise<T>;
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* One process-wide budget for every remote orchestration operation.
|
|
3
|
+
*
|
|
4
|
+
* Transcript catch-up, startup reconcile, and queue/handler initiated work
|
|
5
|
+
* all use this scheduler. Keeping the fence here prevents separate entry
|
|
6
|
+
* points from opening more sockets than the peer can service.
|
|
7
|
+
*/
|
|
8
|
+
const maxRemoteWork = 4;
|
|
9
|
+
const maxRemoteWorkPerMachine = 2;
|
|
10
|
+
const pending = [];
|
|
11
|
+
const activeByMachine = new Map();
|
|
12
|
+
let active = 0;
|
|
13
|
+
let draining = false;
|
|
14
|
+
export function scheduleRemoteOrchestrationWork(machineId, run) {
|
|
15
|
+
return new Promise((resolve, reject) => {
|
|
16
|
+
pending.push({ machineId, run: async () => await run(), resolve: (value) => resolve(value), reject });
|
|
17
|
+
void drainRemoteOrchestrationWork();
|
|
18
|
+
});
|
|
19
|
+
}
|
|
20
|
+
async function drainRemoteOrchestrationWork() {
|
|
21
|
+
if (draining)
|
|
22
|
+
return;
|
|
23
|
+
draining = true;
|
|
24
|
+
try {
|
|
25
|
+
for (;;) {
|
|
26
|
+
if (active >= maxRemoteWork)
|
|
27
|
+
return;
|
|
28
|
+
const index = pending.findIndex((work) => (activeByMachine.get(work.machineId) ?? 0) < maxRemoteWorkPerMachine);
|
|
29
|
+
if (index < 0)
|
|
30
|
+
return;
|
|
31
|
+
const work = pending.splice(index, 1)[0];
|
|
32
|
+
if (!work)
|
|
33
|
+
return;
|
|
34
|
+
active += 1;
|
|
35
|
+
activeByMachine.set(work.machineId, (activeByMachine.get(work.machineId) ?? 0) + 1);
|
|
36
|
+
void work.run().then(work.resolve, work.reject).finally(() => {
|
|
37
|
+
active -= 1;
|
|
38
|
+
const machineActive = activeByMachine.get(work.machineId) ?? 1;
|
|
39
|
+
if (machineActive <= 1)
|
|
40
|
+
activeByMachine.delete(work.machineId);
|
|
41
|
+
else
|
|
42
|
+
activeByMachine.set(work.machineId, machineActive - 1);
|
|
43
|
+
void drainRemoteOrchestrationWork();
|
|
44
|
+
});
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
finally {
|
|
48
|
+
draining = false;
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=remote-work-budget.js.map
|
|
@@ -12,55 +12,129 @@ import { encryptedSessionMetadata } from "../session/metadata.js";
|
|
|
12
12
|
import { loadSessionProjections } from "../session/store.js";
|
|
13
13
|
import { appendAttention, expireInflightLeases, leaseNextAttention, latestAttentionForTask, markAttentionDelivered, markAttentionDeliveryFailed, markSubscriptionLiveness, recordTurnRuntimeBinding, reopenUnresolvedArtifactFeedbackAttentions, readOrchestrationStore, } from "./store.js";
|
|
14
14
|
import { attentionFromReconcile, catchUpSubscriptionEvents } from "./watcher.js";
|
|
15
|
+
import { scheduleRemoteOrchestrationWork } from "./remote-work-budget.js";
|
|
15
16
|
import { resolveWakeIdentity } from "./wake-identity.js";
|
|
16
17
|
const activeDeliveryTicks = new Set();
|
|
17
18
|
const reconciledSubscriptions = new Set();
|
|
19
|
+
const remoteScanStates = new Map();
|
|
20
|
+
const pendingRemoteReconciles = new Map();
|
|
21
|
+
let orchestrationTickInFlight = false;
|
|
18
22
|
const defaultStaleMs = 30 * 60 * 1000;
|
|
23
|
+
const defaultRemoteScanIntervalMs = 10_000;
|
|
24
|
+
const defaultRemoteScanBackoffMs = 5_000;
|
|
25
|
+
const maxRemoteScanBackoffMs = 60_000;
|
|
19
26
|
export async function tickOrchestration(config) {
|
|
20
|
-
|
|
21
|
-
await expireInflightLeases();
|
|
22
|
-
await scanRemoteSubscriptions(config);
|
|
23
|
-
await reconcileSubscriptionsOnce(config);
|
|
24
|
-
await markStaleSubscriptions(config);
|
|
25
|
-
await reopenUnresolvedArtifactFeedbackAttentions();
|
|
26
|
-
const busyConsumerSessionIds = new Set();
|
|
27
|
-
for (const sessionId of sessions.keys()) {
|
|
28
|
-
if (sessionTurns.activeTurnId(sessionId))
|
|
29
|
-
busyConsumerSessionIds.add(sessionId);
|
|
30
|
-
}
|
|
31
|
-
const attention = await leaseNextAttention({
|
|
32
|
-
consumerSessionIds: new Set(sessions.keys()),
|
|
33
|
-
busyConsumerSessionIds,
|
|
34
|
-
leaseMs: deliveryLeaseMs(),
|
|
35
|
-
});
|
|
36
|
-
if (!attention)
|
|
27
|
+
if (orchestrationTickInFlight)
|
|
37
28
|
return;
|
|
38
|
-
|
|
39
|
-
if (activeDeliveryTicks.has(key))
|
|
40
|
-
return;
|
|
41
|
-
activeDeliveryTicks.add(key);
|
|
29
|
+
orchestrationTickInFlight = true;
|
|
42
30
|
try {
|
|
43
|
-
await
|
|
31
|
+
await ensureProjectedSessionsLoaded();
|
|
32
|
+
await expireInflightLeases();
|
|
33
|
+
// This only schedules bounded, per-subscription remote work. It never
|
|
34
|
+
// waits for a remote provider, so a slow/offline peer cannot delay local
|
|
35
|
+
// reconcile, artifact-feedback delivery, or another peer.
|
|
36
|
+
await scanRemoteSubscriptions(config);
|
|
37
|
+
await reconcileSubscriptionsOnce(config);
|
|
38
|
+
await markStaleSubscriptions(config);
|
|
39
|
+
await reopenUnresolvedArtifactFeedbackAttentions();
|
|
40
|
+
const busyConsumerSessionIds = new Set();
|
|
41
|
+
for (const sessionId of sessions.keys()) {
|
|
42
|
+
if (sessionTurns.activeTurnId(sessionId))
|
|
43
|
+
busyConsumerSessionIds.add(sessionId);
|
|
44
|
+
}
|
|
45
|
+
const attention = await leaseNextAttention({
|
|
46
|
+
consumerSessionIds: new Set(sessions.keys()),
|
|
47
|
+
busyConsumerSessionIds,
|
|
48
|
+
leaseMs: deliveryLeaseMs(),
|
|
49
|
+
});
|
|
50
|
+
if (!attention)
|
|
51
|
+
return;
|
|
52
|
+
const key = `${attention.taskId}:${attention.eventCursor}`;
|
|
53
|
+
if (activeDeliveryTicks.has(key))
|
|
54
|
+
return;
|
|
55
|
+
activeDeliveryTicks.add(key);
|
|
56
|
+
try {
|
|
57
|
+
await deliverAttention(config, attention);
|
|
58
|
+
}
|
|
59
|
+
finally {
|
|
60
|
+
activeDeliveryTicks.delete(key);
|
|
61
|
+
}
|
|
44
62
|
}
|
|
45
63
|
finally {
|
|
46
|
-
|
|
64
|
+
orchestrationTickInFlight = false;
|
|
47
65
|
}
|
|
48
66
|
}
|
|
49
67
|
async function scanRemoteSubscriptions(config) {
|
|
50
68
|
const store = await readOrchestrationStore();
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
69
|
+
const now = Date.now();
|
|
70
|
+
const candidates = store.subscriptions
|
|
71
|
+
.filter((subscription) => subscription.status === "active")
|
|
72
|
+
.filter((subscription) => (subscription.crewMachineId ?? config.machineId) !== config.machineId)
|
|
73
|
+
.map((subscription) => {
|
|
74
|
+
const key = remoteSubscriptionKey(subscription);
|
|
75
|
+
const state = remoteScanStates.get(key) ?? { nextAt: 0, backoffMs: 0 };
|
|
76
|
+
remoteScanStates.set(key, state);
|
|
77
|
+
const persistedAt = Date.parse(subscription.lastRemoteScanAt ?? "");
|
|
78
|
+
const persistedNextAt = Number.isFinite(persistedAt) ? persistedAt + remoteScanIntervalMs() : 0;
|
|
79
|
+
return { subscription, key, state, dueAt: Math.max(state.nextAt, persistedNextAt) };
|
|
80
|
+
})
|
|
81
|
+
.filter((candidate) => candidate.dueAt <= now)
|
|
82
|
+
.sort((left, right) => left.dueAt - right.dueAt);
|
|
83
|
+
for (const candidate of candidates) {
|
|
84
|
+
if (candidate.state.inFlight)
|
|
56
85
|
continue;
|
|
57
|
-
|
|
58
|
-
|
|
86
|
+
candidate.state.nextAt = Number.POSITIVE_INFINITY;
|
|
87
|
+
// catchUpSubscriptionEvents owns the budget for the actual remote RPC.
|
|
88
|
+
// Do not acquire it here as well: an outer scan slot waiting for an inner
|
|
89
|
+
// catch-up slot would deadlock a saturated peer backlog.
|
|
90
|
+
const operation = runRemoteSubscriptionScan(config, candidate.subscription, candidate.state)
|
|
91
|
+
.finally(() => {
|
|
92
|
+
candidate.state.inFlight = undefined;
|
|
93
|
+
});
|
|
94
|
+
candidate.state.inFlight = operation;
|
|
95
|
+
}
|
|
96
|
+
void drainRemoteReconciles();
|
|
97
|
+
}
|
|
98
|
+
function enqueueRemoteReconcile(config, subscription) {
|
|
99
|
+
pendingRemoteReconciles.set(remoteSubscriptionKey(subscription), { config, subscription });
|
|
100
|
+
void drainRemoteReconciles();
|
|
101
|
+
}
|
|
102
|
+
async function drainRemoteReconciles() {
|
|
103
|
+
for (const [key, pending] of pendingRemoteReconciles) {
|
|
104
|
+
const crewMachineId = pending.subscription.crewMachineId ?? pending.config.machineId;
|
|
105
|
+
pendingRemoteReconciles.delete(key);
|
|
106
|
+
void scheduleRemoteOrchestrationWork(crewMachineId, () => reconcileRemoteSubscription(pending.config, pending.subscription))
|
|
107
|
+
.catch((error) => {
|
|
108
|
+
console.error(`Remote orchestration reconcile failed for ${pending.subscription.taskId}: ${error instanceof Error ? error.message : String(error)}`);
|
|
59
109
|
});
|
|
60
110
|
}
|
|
61
111
|
}
|
|
112
|
+
async function runRemoteSubscriptionScan(config, subscription, state) {
|
|
113
|
+
let result;
|
|
114
|
+
try {
|
|
115
|
+
result = await scanRemoteTranscript(config, subscription);
|
|
116
|
+
}
|
|
117
|
+
catch (error) {
|
|
118
|
+
console.error(`Remote orchestration transcript scan crashed for ${subscription.taskId}: ${error instanceof Error ? error.message : String(error)}`);
|
|
119
|
+
result = { status: "failed" };
|
|
120
|
+
}
|
|
121
|
+
if (result.status === "ok") {
|
|
122
|
+
state.backoffMs = 0;
|
|
123
|
+
state.nextAt = Date.now() + remoteScanIntervalMs();
|
|
124
|
+
return;
|
|
125
|
+
}
|
|
126
|
+
state.backoffMs = Math.min(maxRemoteScanBackoffMs, state.backoffMs > 0 ? state.backoffMs * 2 : defaultRemoteScanBackoffMs);
|
|
127
|
+
state.nextAt = Date.now() + state.backoffMs;
|
|
128
|
+
}
|
|
129
|
+
function remoteSubscriptionKey(subscription) {
|
|
130
|
+
return `${subscription.crewMachineId ?? "local"}:${subscription.taskId}:${subscription.crewSessionId}:${subscription.turnId}`;
|
|
131
|
+
}
|
|
62
132
|
async function scanRemoteTranscript(config, subscription) {
|
|
63
|
-
await catchUpSubscriptionEvents({ config, subscription });
|
|
133
|
+
const result = await catchUpSubscriptionEvents({ config, subscription });
|
|
134
|
+
if (result.status === "failed") {
|
|
135
|
+
console.error(`Remote orchestration transcript scan failed for ${subscription.taskId}`);
|
|
136
|
+
}
|
|
137
|
+
return result;
|
|
64
138
|
}
|
|
65
139
|
async function reconcileSubscriptionsOnce(config) {
|
|
66
140
|
const store = await readOrchestrationStore();
|
|
@@ -76,7 +150,10 @@ async function reconcileSubscriptionsOnce(config) {
|
|
|
76
150
|
if (latest?.eventTypes.some((type) => type === "terminal" || type === "failure"))
|
|
77
151
|
continue;
|
|
78
152
|
if (crewMachineId !== config.machineId) {
|
|
79
|
-
|
|
153
|
+
// Remote reconcile shares the same bounded/fair scheduler as transcript
|
|
154
|
+
// catch-up. Startup recovery must not create an unbounded second fleet
|
|
155
|
+
// of sockets for one slow peer.
|
|
156
|
+
enqueueRemoteReconcile(config, subscription);
|
|
80
157
|
continue;
|
|
81
158
|
}
|
|
82
159
|
await reconcileLocalSubscription(config, subscription, "startup");
|
|
@@ -428,6 +505,13 @@ function deliveryLeaseMs() {
|
|
|
428
505
|
const parsed = Number(raw);
|
|
429
506
|
return Number.isFinite(parsed) && parsed > 0 ? parsed : 5 * 60 * 1000;
|
|
430
507
|
}
|
|
508
|
+
function remoteScanIntervalMs() {
|
|
509
|
+
const raw = process.env.HAPPY_ELVES_ORCHESTRATION_REMOTE_SCAN_MS;
|
|
510
|
+
if (!raw)
|
|
511
|
+
return defaultRemoteScanIntervalMs;
|
|
512
|
+
const parsed = Number(raw);
|
|
513
|
+
return Number.isFinite(parsed) && parsed >= 1_000 ? parsed : defaultRemoteScanIntervalMs;
|
|
514
|
+
}
|
|
431
515
|
function internalSocket() {
|
|
432
516
|
return {
|
|
433
517
|
readyState: WebSocket.CLOSED,
|
|
@@ -89,6 +89,10 @@ export type OrchestrationSubscriptionRecord = {
|
|
|
89
89
|
lastSeenAt?: string;
|
|
90
90
|
lastRemoteScanAt?: string;
|
|
91
91
|
remoteEventSeq?: number;
|
|
92
|
+
/** Last provider item used as the forward catch-up anchor for remote scans. */
|
|
93
|
+
remoteEventItemId?: string;
|
|
94
|
+
/** Append-only runtime event cursor; bound to the exact Crew session/turn. */
|
|
95
|
+
remoteEventCursor?: string;
|
|
92
96
|
createdAt: string;
|
|
93
97
|
updatedAt: string;
|
|
94
98
|
};
|
|
@@ -236,7 +240,12 @@ export declare function upsertSubscription(input: Omit<OrchestrationSubscription
|
|
|
236
240
|
export declare function activeSubscriptionsForTurn(crewSessionId: string, turnId: string): Promise<OrchestrationSubscriptionRecord[]>;
|
|
237
241
|
export declare function markSubscriptionLiveness(crewSessionId: string, turnId: string, at?: string): Promise<void>;
|
|
238
242
|
export declare function markSubscriptionRemoteScan(taskId: string, input: {
|
|
243
|
+
crewMachineId?: string;
|
|
244
|
+
crewSessionId?: string;
|
|
245
|
+
turnId?: string;
|
|
239
246
|
remoteEventSeq?: number;
|
|
247
|
+
remoteEventItemId?: string;
|
|
248
|
+
remoteEventCursor?: string;
|
|
240
249
|
at?: string;
|
|
241
250
|
}): Promise<void>;
|
|
242
251
|
export declare function appendAttention(input: AttentionInput): Promise<TaskAttentionRecord[]>;
|
|
@@ -712,11 +712,18 @@ export async function upsertSubscription(input) {
|
|
|
712
712
|
function upsertSubscriptionRecord(store, input, now) {
|
|
713
713
|
const cursor = input.cursor ?? cursorFromNumber(store.nextCursor);
|
|
714
714
|
const index = store.subscriptions.findIndex((record) => record.taskId === input.taskId);
|
|
715
|
+
const previous = index >= 0 ? store.subscriptions[index] : undefined;
|
|
716
|
+
const previousRemoteEventSeq = previous?.remoteEventSeq;
|
|
717
|
+
const crewMachineId = input.crewMachineId ?? previous?.crewMachineId;
|
|
718
|
+
const bindingUnchanged = Boolean(previous &&
|
|
719
|
+
previous.crewMachineId === crewMachineId &&
|
|
720
|
+
previous.crewSessionId === input.crewSessionId &&
|
|
721
|
+
previous.turnId === input.turnId);
|
|
715
722
|
const next = {
|
|
716
|
-
...(
|
|
723
|
+
...(previous ? { createdAt: previous.createdAt } : { createdAt: now }),
|
|
717
724
|
taskId: input.taskId,
|
|
718
725
|
...(input.consumerMachineId ? { consumerMachineId: input.consumerMachineId } : {}),
|
|
719
|
-
...(
|
|
726
|
+
...(crewMachineId ? { crewMachineId } : {}),
|
|
720
727
|
consumerSessionId: input.consumerSessionId,
|
|
721
728
|
crewSessionId: input.crewSessionId,
|
|
722
729
|
turnId: input.turnId,
|
|
@@ -725,9 +732,13 @@ function upsertSubscriptionRecord(store, input, now) {
|
|
|
725
732
|
eventTypes: normalizeEventTypes(input.eventTypes),
|
|
726
733
|
cursor,
|
|
727
734
|
status: input.status ?? "active",
|
|
728
|
-
...(input.lastSeenAt ? { lastSeenAt: input.lastSeenAt } : {}),
|
|
729
|
-
...(input.lastRemoteScanAt ? { lastRemoteScanAt: input.lastRemoteScanAt } : {}),
|
|
730
|
-
...(Number.isInteger(
|
|
735
|
+
...(input.lastSeenAt ? { lastSeenAt: input.lastSeenAt } : bindingUnchanged && previous?.lastSeenAt ? { lastSeenAt: previous.lastSeenAt } : {}),
|
|
736
|
+
...(bindingUnchanged && previous?.lastRemoteScanAt ? { lastRemoteScanAt: previous.lastRemoteScanAt } : input.lastRemoteScanAt ? { lastRemoteScanAt: input.lastRemoteScanAt } : {}),
|
|
737
|
+
...(bindingUnchanged && Number.isInteger(previousRemoteEventSeq)
|
|
738
|
+
? { remoteEventSeq: previousRemoteEventSeq }
|
|
739
|
+
: Number.isInteger(input.remoteEventSeq) ? { remoteEventSeq: input.remoteEventSeq } : {}),
|
|
740
|
+
...(bindingUnchanged && previous?.remoteEventItemId ? { remoteEventItemId: previous.remoteEventItemId } : input.remoteEventItemId ? { remoteEventItemId: input.remoteEventItemId } : {}),
|
|
741
|
+
...(bindingUnchanged && previous?.remoteEventCursor ? { remoteEventCursor: previous.remoteEventCursor } : input.remoteEventCursor ? { remoteEventCursor: input.remoteEventCursor } : {}),
|
|
731
742
|
updatedAt: now,
|
|
732
743
|
};
|
|
733
744
|
if (index >= 0)
|
|
@@ -759,11 +770,21 @@ export async function markSubscriptionRemoteScan(taskId, input) {
|
|
|
759
770
|
for (const record of current.subscriptions) {
|
|
760
771
|
if (record.taskId !== taskId || record.status !== "active")
|
|
761
772
|
continue;
|
|
773
|
+
if (input.crewMachineId !== undefined && record.crewMachineId !== input.crewMachineId)
|
|
774
|
+
continue;
|
|
775
|
+
if (input.crewSessionId !== undefined && record.crewSessionId !== input.crewSessionId)
|
|
776
|
+
continue;
|
|
777
|
+
if (input.turnId !== undefined && record.turnId !== input.turnId)
|
|
778
|
+
continue;
|
|
762
779
|
record.lastRemoteScanAt = at;
|
|
763
780
|
const remoteEventSeq = input.remoteEventSeq;
|
|
764
781
|
if (remoteEventSeq !== undefined && Number.isInteger(remoteEventSeq)) {
|
|
765
782
|
record.remoteEventSeq = Math.max(record.remoteEventSeq ?? 0, remoteEventSeq);
|
|
766
783
|
}
|
|
784
|
+
if (input.remoteEventItemId)
|
|
785
|
+
record.remoteEventItemId = input.remoteEventItemId;
|
|
786
|
+
if (input.remoteEventCursor)
|
|
787
|
+
record.remoteEventCursor = input.remoteEventCursor;
|
|
767
788
|
record.updatedAt = at;
|
|
768
789
|
}
|
|
769
790
|
return current;
|
|
@@ -1430,6 +1451,8 @@ function normalizeSubscriptionRecord(value, localMachineId) {
|
|
|
1430
1451
|
...(typeof value.providerTurnId === "string" && value.providerTurnId ? { providerTurnId: value.providerTurnId } : {}),
|
|
1431
1452
|
...(typeof value.lastRemoteScanAt === "string" && value.lastRemoteScanAt ? { lastRemoteScanAt: value.lastRemoteScanAt } : {}),
|
|
1432
1453
|
...(typeof remoteEventSeq === "number" && Number.isInteger(remoteEventSeq) && remoteEventSeq >= 0 ? { remoteEventSeq } : {}),
|
|
1454
|
+
...(typeof value.remoteEventItemId === "string" && value.remoteEventItemId ? { remoteEventItemId: value.remoteEventItemId } : {}),
|
|
1455
|
+
...(typeof value.remoteEventCursor === "string" && value.remoteEventCursor ? { remoteEventCursor: value.remoteEventCursor } : {}),
|
|
1433
1456
|
};
|
|
1434
1457
|
}
|
|
1435
1458
|
function isTurnBindingRecord(value) {
|
|
@@ -12,10 +12,16 @@ export declare function observeTurnReconcile(input: {
|
|
|
12
12
|
config: DaemonConfig;
|
|
13
13
|
result: TurnReconcileResult;
|
|
14
14
|
}): Promise<void>;
|
|
15
|
+
export type SubscriptionCatchUpResult = {
|
|
16
|
+
status: "ok" | "unreachable" | "failed";
|
|
17
|
+
maxSeq?: number;
|
|
18
|
+
nextAfter?: string;
|
|
19
|
+
lastItemId?: string;
|
|
20
|
+
};
|
|
15
21
|
export declare function catchUpSubscriptionEvents(input: {
|
|
16
22
|
config: DaemonConfig;
|
|
17
23
|
subscription: OrchestrationSubscriptionRecord;
|
|
18
|
-
}): Promise<
|
|
24
|
+
}): Promise<SubscriptionCatchUpResult>;
|
|
19
25
|
export declare function attentionFromReconcile(result: TurnReconcileResult): {
|
|
20
26
|
eventType: TaskAttentionEventType;
|
|
21
27
|
reason: string;
|
|
@@ -1,8 +1,13 @@
|
|
|
1
1
|
import { randomId, } from "../../../../packages/shared/dist/index.js";
|
|
2
2
|
import { appendAudit } from "../audit.js";
|
|
3
|
-
import { requestRemoteMachineResult } from "../relay/remote-command.js";
|
|
3
|
+
import { RemoteCommandError, requestRemoteMachineResult } from "../relay/remote-command.js";
|
|
4
4
|
import { runtimeRead } from "../session/runtime-truth.js";
|
|
5
5
|
import { activeSubscriptionsForTurn, appendAttention, markSubscriptionLiveness, markSubscriptionRemoteScan, recordTurnRuntimeBinding, } from "./store.js";
|
|
6
|
+
import { scheduleRemoteOrchestrationWork } from "./remote-work-budget.js";
|
|
7
|
+
const remoteCatchupsInFlight = new Map();
|
|
8
|
+
const legacyRemoteCursorFallbackUntil = new Map();
|
|
9
|
+
const legacyRemoteCursorFallbackMs = 5 * 60 * 1000;
|
|
10
|
+
const initialRemoteEventCursor = "event_000000000000";
|
|
6
11
|
export async function observeSessionEvent(input) {
|
|
7
12
|
const attention = attentionFromEvent(input.payload);
|
|
8
13
|
if (!attention) {
|
|
@@ -95,6 +100,27 @@ export async function observeTurnReconcile(input) {
|
|
|
95
100
|
}
|
|
96
101
|
export async function catchUpSubscriptionEvents(input) {
|
|
97
102
|
const crewMachineId = input.subscription.crewMachineId ?? input.config.machineId;
|
|
103
|
+
if (crewMachineId !== input.config.machineId) {
|
|
104
|
+
const key = `${crewMachineId}:${input.subscription.taskId}:${input.subscription.crewSessionId}:${input.subscription.turnId}`;
|
|
105
|
+
const existing = remoteCatchupsInFlight.get(key);
|
|
106
|
+
if (existing)
|
|
107
|
+
return await existing;
|
|
108
|
+
const operation = scheduleRemoteOrchestrationWork(crewMachineId, () => catchUpSubscriptionEventsInternal(input)).finally(() => {
|
|
109
|
+
if (remoteCatchupsInFlight.get(key) === operation)
|
|
110
|
+
remoteCatchupsInFlight.delete(key);
|
|
111
|
+
});
|
|
112
|
+
remoteCatchupsInFlight.set(key, operation);
|
|
113
|
+
return await operation;
|
|
114
|
+
}
|
|
115
|
+
return await catchUpSubscriptionEventsInternal(input);
|
|
116
|
+
}
|
|
117
|
+
async function catchUpSubscriptionEventsInternal(input) {
|
|
118
|
+
const crewMachineId = input.subscription.crewMachineId ?? input.config.machineId;
|
|
119
|
+
const remoteCursorEnabled = crewMachineId !== input.config.machineId && canUseRemoteEventCursor(crewMachineId);
|
|
120
|
+
const remoteCursor = remoteCursorEnabled
|
|
121
|
+
? input.subscription.remoteEventCursor
|
|
122
|
+
?? (input.subscription.remoteEventSeq ? `event_${String(input.subscription.remoteEventSeq).padStart(12, "0")}` : initialRemoteEventCursor)
|
|
123
|
+
: undefined;
|
|
98
124
|
try {
|
|
99
125
|
const result = crewMachineId === input.config.machineId
|
|
100
126
|
? await runtimeRead(input.config, {
|
|
@@ -118,6 +144,8 @@ export async function catchUpSubscriptionEvents(input) {
|
|
|
118
144
|
action: "turn.eventsPage",
|
|
119
145
|
sessionId: input.subscription.crewSessionId,
|
|
120
146
|
turnId: input.subscription.turnId,
|
|
147
|
+
...(remoteCursor ? { after: remoteCursor } : {}),
|
|
148
|
+
...(input.subscription.remoteEventItemId && !remoteCursor ? { afterItemId: input.subscription.remoteEventItemId } : {}),
|
|
121
149
|
...(input.subscription.runtimeTurnId ?? input.subscription.providerTurnId
|
|
122
150
|
? { runtimeTurnId: input.subscription.runtimeTurnId ?? input.subscription.providerTurnId }
|
|
123
151
|
: {}),
|
|
@@ -137,7 +165,14 @@ export async function catchUpSubscriptionEvents(input) {
|
|
|
137
165
|
errorCode: "errorCode" in result ? result.errorCode : undefined,
|
|
138
166
|
reason: "reason" in result ? result.reason : undefined,
|
|
139
167
|
});
|
|
140
|
-
return;
|
|
168
|
+
return { status: "unreachable" };
|
|
169
|
+
}
|
|
170
|
+
// A rolling-upgrade peer may accept the old runtimeRead command but omit
|
|
171
|
+
// the forward cursor in its response. Treat that as the legacy path for
|
|
172
|
+
// this machine; in particular, never advance a durable cursor from a
|
|
173
|
+
// page whose peer cannot prove append-only progress.
|
|
174
|
+
if (crewMachineId !== input.config.machineId && remoteCursorEnabled && !result.cursorSupported && !result.nextAfter) {
|
|
175
|
+
legacyRemoteCursorFallbackUntil.set(crewMachineId, Date.now() + legacyRemoteCursorFallbackMs);
|
|
141
176
|
}
|
|
142
177
|
const observed = await observeSubscriptionRuntimeEvents({
|
|
143
178
|
config: input.config,
|
|
@@ -146,11 +181,28 @@ export async function catchUpSubscriptionEvents(input) {
|
|
|
146
181
|
events: result.events,
|
|
147
182
|
skipThroughSeq: crewMachineId === input.config.machineId ? 0 : input.subscription.remoteEventSeq ?? 0,
|
|
148
183
|
});
|
|
149
|
-
if (crewMachineId !== input.config.machineId
|
|
150
|
-
|
|
184
|
+
if (crewMachineId !== input.config.machineId) {
|
|
185
|
+
const canAdvance = Boolean(result.nextAfter);
|
|
186
|
+
await markSubscriptionRemoteScan(input.subscription.taskId, {
|
|
187
|
+
crewMachineId,
|
|
188
|
+
crewSessionId: input.subscription.crewSessionId,
|
|
189
|
+
turnId: input.subscription.turnId,
|
|
190
|
+
...(canAdvance && observed.maxSeq > (input.subscription.remoteEventSeq ?? 0) ? { remoteEventSeq: observed.maxSeq } : {}),
|
|
191
|
+
...(canAdvance && observed.lastItemId ? { remoteEventItemId: observed.lastItemId } : {}),
|
|
192
|
+
...(canAdvance && result.nextAfter ? { remoteEventCursor: result.nextAfter } : {}),
|
|
193
|
+
});
|
|
151
194
|
}
|
|
195
|
+
return {
|
|
196
|
+
status: "ok",
|
|
197
|
+
maxSeq: observed.maxSeq,
|
|
198
|
+
...(result.nextAfter ? { nextAfter: result.nextAfter } : {}),
|
|
199
|
+
...(observed.lastItemId ? { lastItemId: observed.lastItemId } : {}),
|
|
200
|
+
};
|
|
152
201
|
}
|
|
153
202
|
catch (error) {
|
|
203
|
+
if (crewMachineId !== input.config.machineId && remoteCursorEnabled && isRemoteEventCursorCompatibilityFailure(error)) {
|
|
204
|
+
legacyRemoteCursorFallbackUntil.set(crewMachineId, Date.now() + legacyRemoteCursorFallbackMs);
|
|
205
|
+
}
|
|
154
206
|
await appendOrchestrationAudit(input.config, "orchestration.subscription.catchup_failed", `Subscription catch-up failed for ${input.subscription.taskId}`, {
|
|
155
207
|
taskId: input.subscription.taskId,
|
|
156
208
|
crewMachineId,
|
|
@@ -158,17 +210,38 @@ export async function catchUpSubscriptionEvents(input) {
|
|
|
158
210
|
turnId: input.subscription.turnId,
|
|
159
211
|
error: error instanceof Error ? error.message : String(error),
|
|
160
212
|
});
|
|
213
|
+
return { status: "failed" };
|
|
161
214
|
}
|
|
162
215
|
}
|
|
216
|
+
function canUseRemoteEventCursor(crewMachineId) {
|
|
217
|
+
const until = legacyRemoteCursorFallbackUntil.get(crewMachineId);
|
|
218
|
+
if (until === undefined)
|
|
219
|
+
return true;
|
|
220
|
+
if (until <= Date.now()) {
|
|
221
|
+
legacyRemoteCursorFallbackUntil.delete(crewMachineId);
|
|
222
|
+
return true;
|
|
223
|
+
}
|
|
224
|
+
return false;
|
|
225
|
+
}
|
|
226
|
+
function isRemoteEventCursorCompatibilityFailure(error) {
|
|
227
|
+
if (!(error instanceof RemoteCommandError))
|
|
228
|
+
return false;
|
|
229
|
+
return error.code === "REMOTE_COMMAND_TIMEOUT" ||
|
|
230
|
+
error.code === "REMOTE_MACHINE_ERROR" ||
|
|
231
|
+
/INVALID|UNKNOWN|UNRECOGNIZED|SCHEMA|PROTOCOL|CURSOR|AFTER/u.test(error.code);
|
|
232
|
+
}
|
|
163
233
|
async function observeSubscriptionRuntimeEvents(input) {
|
|
164
234
|
const crewMachineId = input.crewMachineId ?? input.subscription.crewMachineId ?? input.config.machineId;
|
|
165
235
|
const skipThroughSeq = input.skipThroughSeq ?? 0;
|
|
166
236
|
let maxSeq = skipThroughSeq;
|
|
167
237
|
let observed = 0;
|
|
238
|
+
let lastItemId;
|
|
168
239
|
for (const event of [...input.events].sort((left, right) => (left.seq ?? 0) - (right.seq ?? 0))) {
|
|
169
240
|
const seq = event.seq ?? 0;
|
|
170
241
|
if (seq > 0)
|
|
171
242
|
maxSeq = Math.max(maxSeq, seq);
|
|
243
|
+
if (event.itemId && seq >= maxSeq)
|
|
244
|
+
lastItemId = event.itemId;
|
|
172
245
|
if (seq > 0 && seq <= skipThroughSeq)
|
|
173
246
|
continue;
|
|
174
247
|
if (event.turnId !== input.subscription.turnId)
|
|
@@ -190,7 +263,7 @@ async function observeSubscriptionRuntimeEvents(input) {
|
|
|
190
263
|
});
|
|
191
264
|
observed += 1;
|
|
192
265
|
}
|
|
193
|
-
return { maxSeq, observed };
|
|
266
|
+
return { maxSeq, observed, ...(lastItemId ? { lastItemId } : {}) };
|
|
194
267
|
}
|
|
195
268
|
function attentionFromEvent(payload) {
|
|
196
269
|
if (payload.type === "option_request") {
|
|
@@ -84,6 +84,22 @@ export async function handleServerMessage(ws, config, message) {
|
|
|
84
84
|
capability: message.capability,
|
|
85
85
|
},
|
|
86
86
|
});
|
|
87
|
+
// A detached relay operation failure is recoverable only after the
|
|
88
|
+
// authenticated stream is recreated. Keeping this socket open would
|
|
89
|
+
// leave durable packets stranded: the relay has already rejected the
|
|
90
|
+
// replay/live send, while the daemon's normal outbox drain is only
|
|
91
|
+
// kicked on a new socket. Close the stream for storage-generation
|
|
92
|
+
// failures so start() performs its bounded reconnect loop. Do not do
|
|
93
|
+
// this for ordinary command validation errors.
|
|
94
|
+
if (shouldReconnectAfterRelayError(message.code) && ws.readyState === WebSocket.OPEN) {
|
|
95
|
+
try {
|
|
96
|
+
ws.close();
|
|
97
|
+
}
|
|
98
|
+
catch {
|
|
99
|
+
// The close event/error path still performs cleanup if the socket
|
|
100
|
+
// implementation rejects a redundant close.
|
|
101
|
+
}
|
|
102
|
+
}
|
|
87
103
|
return;
|
|
88
104
|
}
|
|
89
105
|
if (message.type === "machine:createSession") {
|
|
@@ -268,6 +284,14 @@ async function handleRelayWireMessage(ws, config, message) {
|
|
|
268
284
|
packetId: message.packetId,
|
|
269
285
|
},
|
|
270
286
|
});
|
|
287
|
+
if (shouldReconnectAfterRelayError(message.code) && ws.readyState === WebSocket.OPEN) {
|
|
288
|
+
try {
|
|
289
|
+
ws.close();
|
|
290
|
+
}
|
|
291
|
+
catch {
|
|
292
|
+
// Cleanup is also driven by the close/error listeners below.
|
|
293
|
+
}
|
|
294
|
+
}
|
|
271
295
|
return;
|
|
272
296
|
}
|
|
273
297
|
if (message.type !== "server:packet")
|
|
@@ -338,6 +362,9 @@ async function handleRelayWireMessage(ws, config, message) {
|
|
|
338
362
|
ackIncomingPacket(ws, message);
|
|
339
363
|
}
|
|
340
364
|
}
|
|
365
|
+
function shouldReconnectAfterRelayError(code) {
|
|
366
|
+
return code === "RELAY_STORAGE_TIMEOUT" || code === "RELAY_STORAGE_UNAVAILABLE" || code === "RELAY_STORAGE_CLOSED";
|
|
367
|
+
}
|
|
341
368
|
function errorCode(error) {
|
|
342
369
|
if (!error || typeof error !== "object")
|
|
343
370
|
return undefined;
|
|
@@ -51,6 +51,10 @@ export async function publishRemoteControllerCommand(config, input) {
|
|
|
51
51
|
// durable daemon connection (with its own cursor) must remain the
|
|
52
52
|
// only consumer that can acknowledge those commands.
|
|
53
53
|
cursor: Number.MAX_SAFE_INTEGER,
|
|
54
|
+
// This ephemeral RPC only needs the session-to-machine routing table
|
|
55
|
+
// to publish one targeted command. Do not make every status/read call
|
|
56
|
+
// download the account's retained transcript and artifact payloads.
|
|
57
|
+
snapshotMode: "routing",
|
|
54
58
|
}));
|
|
55
59
|
});
|
|
56
60
|
ws.addEventListener("error", () => {
|
|
@@ -32,11 +32,22 @@ export declare function selectThreadEventsAfterItem(client: HeadlessRuntimeClien
|
|
|
32
32
|
events: HeadlessRuntimeEvent[];
|
|
33
33
|
anchorFound: boolean;
|
|
34
34
|
}>;
|
|
35
|
+
/**
|
|
36
|
+
* Read a forward page from the append-only runtime event sequence. Unlike an
|
|
37
|
+
* item anchor this cursor advances for every provider event, including
|
|
38
|
+
* repeated updates to one item and terminal events without an item id.
|
|
39
|
+
*/
|
|
40
|
+
export declare function selectThreadEventsAfterCursor(client: HeadlessRuntimeClient, threadId: string, afterCursor: string, limit: number): Promise<{
|
|
41
|
+
events: HeadlessRuntimeEvent[];
|
|
42
|
+
hasMoreBefore: boolean;
|
|
43
|
+
nextAfter?: string;
|
|
44
|
+
}>;
|
|
35
45
|
export declare function historicalEventsFromRuntimeEvents(runtimeEvents: HeadlessRuntimeEvent[]): RuntimeHistoricalEvent[];
|
|
36
|
-
export declare function limitHistoricalEventsByBytes(events: RuntimeHistoricalEvent[], maxBytes: number | undefined): {
|
|
46
|
+
export declare function limitHistoricalEventsByBytes(events: RuntimeHistoricalEvent[], maxBytes: number | undefined, direction?: "before" | "after"): {
|
|
37
47
|
events: RuntimeHistoricalEvent[];
|
|
38
48
|
truncated: boolean;
|
|
39
49
|
nextBefore?: string;
|
|
50
|
+
nextAfter?: string;
|
|
40
51
|
};
|
|
41
52
|
export declare function terminalHistoricalEventFromTurn(turn: Turn & {
|
|
42
53
|
status: "completed" | "interrupted" | "failed";
|
|
@@ -330,6 +330,26 @@ export async function selectThreadEventsAfterItem(client, threadId, afterItemId,
|
|
|
330
330
|
events: sortRuntimeEventsByCursor([...selected, ...terminalEvents]),
|
|
331
331
|
};
|
|
332
332
|
}
|
|
333
|
+
/**
|
|
334
|
+
* Read a forward page from the append-only runtime event sequence. Unlike an
|
|
335
|
+
* item anchor this cursor advances for every provider event, including
|
|
336
|
+
* repeated updates to one item and terminal events without an item id.
|
|
337
|
+
*/
|
|
338
|
+
export async function selectThreadEventsAfterCursor(client, threadId, afterCursor, limit) {
|
|
339
|
+
const page = await client.listThreadEvents({
|
|
340
|
+
threadId,
|
|
341
|
+
order: "asc",
|
|
342
|
+
afterCursor,
|
|
343
|
+
limit,
|
|
344
|
+
});
|
|
345
|
+
const events = sortRuntimeEventsByCursor(page.events);
|
|
346
|
+
const nextAfter = events.at(-1)?.cursor;
|
|
347
|
+
return {
|
|
348
|
+
events,
|
|
349
|
+
hasMoreBefore: false,
|
|
350
|
+
...(nextAfter ? { nextAfter } : {}),
|
|
351
|
+
};
|
|
352
|
+
}
|
|
333
353
|
export function historicalEventsFromRuntimeEvents(runtimeEvents) {
|
|
334
354
|
const deltaItemIds = runtimeEvents
|
|
335
355
|
.filter((event) => event.method === "item/delta")
|
|
@@ -384,27 +404,57 @@ export function historicalEventsFromRuntimeEvents(runtimeEvents) {
|
|
|
384
404
|
.sort((a, b) => (a.occurredAt ?? 0) - (b.occurredAt ?? 0) || (a.seq ?? 0) - (b.seq ?? 0))
|
|
385
405
|
.map((event, index) => ({ ...event, seq: event.seq ?? index + 1 }));
|
|
386
406
|
}
|
|
387
|
-
export function limitHistoricalEventsByBytes(events, maxBytes) {
|
|
407
|
+
export function limitHistoricalEventsByBytes(events, maxBytes, direction = "before") {
|
|
388
408
|
if (maxBytes === undefined || !Number.isFinite(maxBytes) || maxBytes <= 0)
|
|
389
409
|
return { events, truncated: false };
|
|
390
410
|
const groups = groupHistoricalEvents(events);
|
|
391
411
|
const selectedNewestFirst = [];
|
|
392
412
|
let bytes = 0;
|
|
393
|
-
|
|
413
|
+
const orderedGroups = direction === "after" ? groups : [...groups].reverse();
|
|
414
|
+
for (const group of orderedGroups) {
|
|
394
415
|
const groupBytes = Buffer.byteLength(JSON.stringify(group.events), "utf8");
|
|
395
416
|
if (bytes + groupBytes > maxBytes) {
|
|
396
417
|
if (selectedNewestFirst.length === 0) {
|
|
418
|
+
if (direction === "after") {
|
|
419
|
+
// A single item can be larger than the page budget. Emit a marker
|
|
420
|
+
// for that item, then consume as many following complete groups as
|
|
421
|
+
// fit. The cursor advances only through groups represented in the
|
|
422
|
+
// response; it must never jump to the raw page tail.
|
|
423
|
+
const marker = historyTruncatedMarker(group.events, maxBytes, group.events.length);
|
|
424
|
+
const kept = [];
|
|
425
|
+
let markerBytes = Buffer.byteLength(JSON.stringify([marker]), "utf8");
|
|
426
|
+
for (const following of orderedGroups.slice(1)) {
|
|
427
|
+
const followingBytes = Buffer.byteLength(JSON.stringify(following.events), "utf8");
|
|
428
|
+
if (markerBytes + followingBytes > maxBytes)
|
|
429
|
+
break;
|
|
430
|
+
kept.push(following);
|
|
431
|
+
markerBytes += followingBytes;
|
|
432
|
+
}
|
|
433
|
+
const consumed = [group, ...kept];
|
|
434
|
+
const lastConsumedEvent = consumed.at(-1)?.events.at(-1);
|
|
435
|
+
return {
|
|
436
|
+
events: [marker, ...kept.flatMap((item) => item.events)],
|
|
437
|
+
truncated: true,
|
|
438
|
+
...(lastConsumedEvent?.seq ? { nextAfter: headlessCursorFromSeq(lastConsumedEvent.seq) } : {}),
|
|
439
|
+
};
|
|
440
|
+
}
|
|
397
441
|
return {
|
|
398
442
|
events: [historyTruncatedMarker(group.events, maxBytes, group.events.length)],
|
|
399
443
|
truncated: true,
|
|
400
|
-
...(
|
|
444
|
+
...(direction === "before"
|
|
445
|
+
? (group.oldestCursor ? { nextBefore: group.oldestCursor } : {})
|
|
446
|
+
: {}),
|
|
401
447
|
};
|
|
402
448
|
}
|
|
403
|
-
const keptGroups = [...selectedNewestFirst].reverse();
|
|
449
|
+
const keptGroups = direction === "after" ? selectedNewestFirst : [...selectedNewestFirst].reverse();
|
|
450
|
+
const lastKeptEvent = keptGroups.at(-1)?.events.at(-1);
|
|
404
451
|
return {
|
|
405
452
|
events: keptGroups.flatMap((item) => item.events),
|
|
406
453
|
truncated: true,
|
|
407
|
-
...(keptGroups[0]?.oldestCursor ? { nextBefore: keptGroups[0].oldestCursor } : {}),
|
|
454
|
+
...(direction === "before" && keptGroups[0]?.oldestCursor ? { nextBefore: keptGroups[0].oldestCursor } : {}),
|
|
455
|
+
...(direction === "after" && lastKeptEvent?.seq
|
|
456
|
+
? { nextAfter: headlessCursorFromSeq(lastKeptEvent.seq) }
|
|
457
|
+
: {}),
|
|
408
458
|
};
|
|
409
459
|
}
|
|
410
460
|
selectedNewestFirst.push(group);
|
|
@@ -1,20 +1,25 @@
|
|
|
1
|
-
import { limitHistoricalEventsByBytes, historicalEventsFromThread, mapHeadlessRuntimeEventsToHistoricalEvents, selectThreadEventsAfterItem, selectThreadEventsBeforeCursor, selectThreadEventsBeforeItem, selectThreadEventsForTurn, } from "./headless-event-projector.js";
|
|
1
|
+
import { limitHistoricalEventsByBytes, historicalEventsFromThread, mapHeadlessRuntimeEventsToHistoricalEvents, selectThreadEventsAfterCursor, selectThreadEventsAfterItem, selectThreadEventsBeforeCursor, selectThreadEventsBeforeItem, selectThreadEventsForTurn, } from "./headless-event-projector.js";
|
|
2
2
|
const defaultEventLimit = 200;
|
|
3
3
|
export async function readHeadlessHistoricalEventPage(input) {
|
|
4
|
+
assertValidHeadlessEventCursor(input.after);
|
|
4
5
|
assertValidHeadlessEventCursor(input.before);
|
|
5
6
|
const limitItems = clampLimit(input.limitItems);
|
|
6
7
|
const selection = await selectHeadlessRuntimeEvents(input, limitItems);
|
|
7
8
|
const runtimeEvents = selection.events;
|
|
8
9
|
const allEvents = await authoritativeTurnEvents(input, runtimeEvents) ?? mapHeadlessRuntimeEventsToHistoricalEvents(runtimeEvents);
|
|
9
|
-
const
|
|
10
|
+
const visibleEvents = input.turnId ? allEvents.filter((event) => event.turnId === input.turnId) : allEvents;
|
|
11
|
+
const limited = limitHistoricalEventsByBytes(visibleEvents, input.maxBytes, input.after ? "after" : "before");
|
|
10
12
|
const includeSessionHead = !input.turnId && !input.before && !input.beforeItemId;
|
|
11
13
|
const head = includeSessionHead ? latestHeadFromEvents(allEvents) : {};
|
|
12
14
|
const nextBefore = limited.nextBefore ?? selection.nextBefore;
|
|
13
15
|
const hasMoreBefore = Boolean((selection.hasMoreBefore || limited.truncated) && nextBefore);
|
|
16
|
+
const nextAfter = limited.truncated ? limited.nextAfter : selection.nextAfter;
|
|
14
17
|
return {
|
|
15
18
|
events: limited.events,
|
|
16
19
|
hasMoreBefore,
|
|
17
20
|
...(hasMoreBefore && nextBefore ? { nextBefore } : {}),
|
|
21
|
+
...(nextAfter ? { nextAfter } : {}),
|
|
22
|
+
...(input.after ? { cursorSupported: true } : {}),
|
|
18
23
|
...(head.currentHead ? { currentHead: head.currentHead } : {}),
|
|
19
24
|
...(head.lastTurnId ? { lastTurnId: head.lastTurnId } : {}),
|
|
20
25
|
};
|
|
@@ -28,7 +33,7 @@ async function authoritativeTurnEvents(input, runtimeEvents) {
|
|
|
28
33
|
// plus a repeated terminal snapshot for the same item, or a late native
|
|
29
34
|
// reconciliation. Identity is itemId/turnId, never text, so legitimate
|
|
30
35
|
// repeated answers from distinct items remain visible.
|
|
31
|
-
if (input.before || input.beforeItemId || input.afterItemId)
|
|
36
|
+
if (input.after || input.before || input.beforeItemId || input.afterItemId)
|
|
32
37
|
return undefined;
|
|
33
38
|
// A few internal readers (and older provider adapters during rolling
|
|
34
39
|
// upgrades) only expose the event reader. Canonical rebuild is an
|
|
@@ -69,13 +74,20 @@ async function listAllThreadTurns(client, threadId) {
|
|
|
69
74
|
return turns;
|
|
70
75
|
}
|
|
71
76
|
async function selectHeadlessRuntimeEvents(input, limitItems) {
|
|
72
|
-
if (input.
|
|
73
|
-
return await
|
|
77
|
+
if (input.after) {
|
|
78
|
+
return await selectThreadEventsAfterCursor(input.client, input.threadId, input.after, limitItems);
|
|
74
79
|
}
|
|
80
|
+
// An after-item anchor is a thread-level forward cursor. It must take
|
|
81
|
+
// precedence over turn filtering so callers can advance a subscription
|
|
82
|
+
// without re-reading the first page of a long turn on every poll. The
|
|
83
|
+
// runtime-truth layer filters the returned page back to the requested turn.
|
|
75
84
|
if (input.afterItemId) {
|
|
76
85
|
const selection = await selectThreadEventsAfterItem(input.client, input.threadId, input.afterItemId, limitItems);
|
|
77
86
|
return { events: selection.events, hasMoreBefore: selection.anchorFound };
|
|
78
87
|
}
|
|
88
|
+
if (input.turnId) {
|
|
89
|
+
return await selectThreadEventsForTurn(input.client, input.threadId, input.turnId, limitItems);
|
|
90
|
+
}
|
|
79
91
|
if (input.beforeItemId) {
|
|
80
92
|
const selection = await selectThreadEventsBeforeItem(input.client, input.threadId, input.beforeItemId, limitItems);
|
|
81
93
|
return { events: selection.events, hasMoreBefore: selection.hasMoreBefore };
|
|
@@ -98,7 +98,21 @@ export async function refreshNativeHistoryForLoadedSession(session, config, requ
|
|
|
98
98
|
const entry = { generation, promise: operation, startedAt: Date.now() };
|
|
99
99
|
orchestrationRefreshes.set(refreshKey, entry);
|
|
100
100
|
try {
|
|
101
|
-
const
|
|
101
|
+
const outcome = await Promise.race([
|
|
102
|
+
operation.then((result) => ({ kind: "result", result })),
|
|
103
|
+
new Promise((resolve) => setTimeout(() => resolve({ kind: "timeout" }), orchestrationRefreshMaxWaitMs)),
|
|
104
|
+
]);
|
|
105
|
+
if (outcome.kind === "timeout") {
|
|
106
|
+
entry.abandoned = true;
|
|
107
|
+
if (orchestrationRefreshes.get(refreshKey) === entry)
|
|
108
|
+
orchestrationRefreshes.delete(refreshKey);
|
|
109
|
+
orchestrationRefreshGenerations.set(refreshKey, (orchestrationRefreshGenerations.get(refreshKey) ?? 0) + 1);
|
|
110
|
+
// The first caller must have the same bounded wait as followers. Drop
|
|
111
|
+
// the provider client so the old IPC cannot poison the next refresh.
|
|
112
|
+
void getHeadlessRuntimeManager(session.cwd).invalidateClientForBinding(session.runtimeBinding, session.cwd);
|
|
113
|
+
return sessions.get(session.sessionId) ?? session;
|
|
114
|
+
}
|
|
115
|
+
const result = outcome.result;
|
|
102
116
|
if (!entry.abandoned && generation !== (orchestrationRefreshGenerations.get(refreshKey) ?? 0)) {
|
|
103
117
|
if (orchestrationRefreshes.get(refreshKey) === entry)
|
|
104
118
|
orchestrationRefreshes.delete(refreshKey);
|
|
@@ -17,7 +17,14 @@ export async function runtimeRead(config, input) {
|
|
|
17
17
|
// Reconcile that history before projecting turn.result/status so an
|
|
18
18
|
// upgrade/restart repairs old same-turn duplicates through the official
|
|
19
19
|
// read path, without requiring an internal syncNativeHistory call.
|
|
20
|
-
|
|
20
|
+
// Event catch-up reads the authoritative Headless event log and must stay
|
|
21
|
+
// independent from the provider-native history refresh. The latter can
|
|
22
|
+
// involve a slow/remote provider IPC call; putting it in every polling
|
|
23
|
+
// page made one stale Crew subscription block all other orchestration.
|
|
24
|
+
// Result/status/list retain the refresh for legacy native-history repair.
|
|
25
|
+
if (input.action !== "turn.eventsPage") {
|
|
26
|
+
session = await refreshNativeHistoryForLoadedSession(session, config, `runtime_truth_refresh_${input.requestId}`);
|
|
27
|
+
}
|
|
21
28
|
runtimeSessionId = runtimeSessionIdForSession(session);
|
|
22
29
|
if (!runtimeSessionId)
|
|
23
30
|
return unreachable(input, `Session ${input.sessionId} does not have a runtime session binding`, undefined, "SESSION_BINDING_MISSING");
|
|
@@ -90,6 +97,7 @@ async function readTurnList(input, session, runtimeSessionId) {
|
|
|
90
97
|
before: input.before,
|
|
91
98
|
beforeItemId: input.beforeItemId,
|
|
92
99
|
afterItemId: input.afterItemId,
|
|
100
|
+
after: input.after,
|
|
93
101
|
limitItems: input.limitItems ?? 400,
|
|
94
102
|
maxBytes: input.maxBytes ?? 240_000,
|
|
95
103
|
});
|
|
@@ -207,6 +215,7 @@ async function readTurnEventsPage(input, session, runtimeSessionId) {
|
|
|
207
215
|
before: input.before,
|
|
208
216
|
beforeItemId: input.beforeItemId,
|
|
209
217
|
afterItemId: input.afterItemId,
|
|
218
|
+
after: input.after,
|
|
210
219
|
limitItems: input.limitItems ?? 300,
|
|
211
220
|
maxBytes: input.maxBytes ?? 260_000,
|
|
212
221
|
});
|
|
@@ -223,6 +232,8 @@ async function readTurnEventsPage(input, session, runtimeSessionId) {
|
|
|
223
232
|
events: events.map((event) => runtimeReadEvent(event, input.turnId, runtimeTurnId)),
|
|
224
233
|
hasMoreBefore: page.hasMoreBefore,
|
|
225
234
|
nextBefore: page.nextBefore,
|
|
235
|
+
...(page.nextAfter ? { nextAfter: page.nextAfter } : {}),
|
|
236
|
+
...(page.cursorSupported ? { cursorSupported: true } : {}),
|
|
226
237
|
...(page.currentHead ? { currentHead: page.currentHead } : {}),
|
|
227
238
|
...(page.lastTurnId ? { lastTurnId: page.lastTurnId } : {}),
|
|
228
239
|
};
|
|
@@ -252,6 +263,7 @@ async function readRuntimePage(session, input) {
|
|
|
252
263
|
...(input.turnId ? { turnId: input.turnId } : {}),
|
|
253
264
|
...(input.before ? { before: input.before } : {}),
|
|
254
265
|
...(input.beforeItemId ? { beforeItemId: input.beforeItemId } : {}),
|
|
266
|
+
...(input.after ? { after: input.after } : {}),
|
|
255
267
|
...(input.afterItemId ? { afterItemId: input.afterItemId } : {}),
|
|
256
268
|
limitItems: input.limitItems ?? 200,
|
|
257
269
|
maxBytes: input.maxBytes ?? 200_000,
|
package/apps/daemon/package.json
CHANGED
|
@@ -72,10 +72,21 @@ export function initPacketRelayTables(db) {
|
|
|
72
72
|
|
|
73
73
|
CREATE INDEX IF NOT EXISTS idx_packets_to_node
|
|
74
74
|
ON packets(space_id, to_node_id, id);
|
|
75
|
+
-- Replay admission must be able to reject already-acked packets without
|
|
76
|
+
-- touching the encrypted payload column. The old query selected
|
|
77
|
+
-- packets.* while walking every packet addressed to a node, so a node
|
|
78
|
+
-- with a large, fully-acked history could read gigabytes of overflow
|
|
79
|
+
-- pages before returning an empty batch.
|
|
80
|
+
CREATE INDEX IF NOT EXISTS idx_packets_unacked_lookup
|
|
81
|
+
ON packets(space_id, to_node_id, id, expires_at, packet_id);
|
|
75
82
|
CREATE INDEX IF NOT EXISTS idx_packets_channel
|
|
76
83
|
ON packets(space_id, channel_id, stream_id, id);
|
|
84
|
+
CREATE INDEX IF NOT EXISTS idx_packets_channel_id
|
|
85
|
+
ON packets(space_id, channel_id, id);
|
|
77
86
|
CREATE INDEX IF NOT EXISTS idx_packets_expires
|
|
78
87
|
ON packets(expires_at);
|
|
88
|
+
CREATE INDEX IF NOT EXISTS idx_packets_received_at
|
|
89
|
+
ON packets(received_at);
|
|
79
90
|
CREATE INDEX IF NOT EXISTS idx_retained_space_key
|
|
80
91
|
ON retained_packets(space_id, retained_key);
|
|
81
92
|
`);
|
|
@@ -192,24 +203,62 @@ export function listRetainedPackets(db, input) {
|
|
|
192
203
|
export function listPacketsForSubscription(db, input) {
|
|
193
204
|
const cursor = input.cursor ?? 0;
|
|
194
205
|
const limit = input.limit ?? 500;
|
|
195
|
-
const
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
206
|
+
const streamId = input.subscription.streamId;
|
|
207
|
+
const rows = [];
|
|
208
|
+
// Keep the historical OR semantics, but execute each arm independently so
|
|
209
|
+
// SQLite can seek through idx_packets_to_node or idx_packets_channel instead
|
|
210
|
+
// of scanning and sorting every packet in the account. A packet can match
|
|
211
|
+
// both arms; the merge below de-duplicates by the stable packets.id cursor.
|
|
212
|
+
// The node arm is intentionally unconditional. `nodeId` is the authenticated
|
|
213
|
+
// connection identity and was always part of the old OR predicate, even for
|
|
214
|
+
// an empty/channel-only subscription. `subscription.toNodeId` is a snapshot
|
|
215
|
+
// label, not a permission to redirect this reader to another node.
|
|
216
|
+
{
|
|
217
|
+
const sql = streamId === undefined
|
|
218
|
+
? `SELECT * FROM packets
|
|
219
|
+
WHERE space_id = ? AND to_node_id = ? AND id > ?
|
|
220
|
+
AND (expires_at IS NULL OR expires_at > ?)
|
|
221
|
+
ORDER BY id ASC LIMIT ?`
|
|
222
|
+
: `SELECT * FROM packets
|
|
223
|
+
WHERE space_id = ? AND to_node_id = ? AND stream_id = ? AND id > ?
|
|
224
|
+
AND (expires_at IS NULL OR expires_at > ?)
|
|
225
|
+
ORDER BY id ASC LIMIT ?`;
|
|
226
|
+
const values = streamId === undefined
|
|
227
|
+
? [input.spaceId, input.nodeId, cursor, input.now, limit]
|
|
228
|
+
: [input.spaceId, input.nodeId, streamId, cursor, input.now, limit];
|
|
229
|
+
rows.push(...statement(db, sql).all(...values));
|
|
230
|
+
}
|
|
231
|
+
if (input.subscription.channelId !== undefined) {
|
|
232
|
+
const sql = streamId === undefined
|
|
233
|
+
? `SELECT * FROM packets
|
|
234
|
+
WHERE space_id = ? AND channel_id = ? AND id > ?
|
|
235
|
+
AND (expires_at IS NULL OR expires_at > ?)
|
|
236
|
+
ORDER BY id ASC LIMIT ?`
|
|
237
|
+
: `SELECT * FROM packets
|
|
238
|
+
WHERE space_id = ? AND channel_id = ? AND stream_id = ? AND id > ?
|
|
239
|
+
AND (expires_at IS NULL OR expires_at > ?)
|
|
240
|
+
ORDER BY id ASC LIMIT ?`;
|
|
241
|
+
const values = streamId === undefined
|
|
242
|
+
? [input.spaceId, input.subscription.channelId, cursor, input.now, limit]
|
|
243
|
+
: [input.spaceId, input.subscription.channelId, streamId, cursor, input.now, limit];
|
|
244
|
+
rows.push(...statement(db, sql).all(...values));
|
|
245
|
+
}
|
|
246
|
+
const uniqueRows = new Map();
|
|
247
|
+
for (const row of rows)
|
|
248
|
+
uniqueRows.set(row.id, row);
|
|
249
|
+
return [...uniqueRows.values()]
|
|
250
|
+
.sort((left, right) => left.id - right.id)
|
|
251
|
+
.slice(0, limit)
|
|
252
|
+
.map(packetFromRow);
|
|
208
253
|
}
|
|
209
254
|
export function listUnackedPacketsForNode(db, input) {
|
|
210
255
|
const cursor = input.cursor ?? 0;
|
|
211
256
|
const limit = input.limit ?? 500;
|
|
212
|
-
|
|
257
|
+
// First find candidate row ids using a covering routing index. Do not
|
|
258
|
+
// select packets.* in this anti-join: encrypted_payload is commonly a
|
|
259
|
+
// multi-kilobyte value stored in overflow pages, and loading it for every
|
|
260
|
+
// already-acked packet turns an empty replay into a large disk read.
|
|
261
|
+
const candidateRows = statement(db, `SELECT packets.id
|
|
213
262
|
FROM packets
|
|
214
263
|
LEFT JOIN packet_acks
|
|
215
264
|
ON packet_acks.space_id = packets.space_id
|
|
@@ -222,13 +271,42 @@ export function listUnackedPacketsForNode(db, input) {
|
|
|
222
271
|
AND packet_acks.packet_id IS NULL
|
|
223
272
|
ORDER BY packets.id ASC
|
|
224
273
|
LIMIT ?`).all(input.nodeId, input.spaceId, cursor, input.nodeId, input.now, limit);
|
|
274
|
+
if (candidateRows.length === 0)
|
|
275
|
+
return [];
|
|
276
|
+
// Fetch the encrypted rows only for the bounded set that really needs to
|
|
277
|
+
// be replayed. The ids are already ordered, but retain an explicit ORDER
|
|
278
|
+
// BY so the FIFO/server-sequence contract remains clear to SQLite.
|
|
279
|
+
const placeholders = candidateRows.map(() => "?").join(", ");
|
|
280
|
+
const rows = statement(db, `SELECT * FROM packets
|
|
281
|
+
WHERE space_id = ? AND id IN (${placeholders})
|
|
282
|
+
ORDER BY id ASC`).all(input.spaceId, ...candidateRows.map((row) => row.id));
|
|
225
283
|
return rows.map(packetFromRow);
|
|
226
284
|
}
|
|
227
285
|
export function prunePackets(db, input) {
|
|
228
286
|
if (input.maxAgeMs <= 0)
|
|
229
287
|
return;
|
|
230
288
|
const oldestReceivedAt = input.now - input.maxAgeMs;
|
|
231
|
-
|
|
289
|
+
// Retention runs on the same serialized storage worker as packet replay.
|
|
290
|
+
// Bound each pass so a large historical database cannot occupy the worker
|
|
291
|
+
// past the operation deadline. The two indexed arms avoid the old OR
|
|
292
|
+
// predicate's full packets-table scan when there is nothing to prune.
|
|
293
|
+
const candidates = statement(db, `WITH expired AS (
|
|
294
|
+
SELECT id FROM packets
|
|
295
|
+
WHERE expires_at IS NOT NULL AND expires_at <= ?
|
|
296
|
+
LIMIT 1000
|
|
297
|
+
), stale AS (
|
|
298
|
+
SELECT id FROM packets
|
|
299
|
+
WHERE received_at <= ?
|
|
300
|
+
LIMIT 1000
|
|
301
|
+
)
|
|
302
|
+
SELECT id
|
|
303
|
+
FROM (SELECT id FROM expired UNION SELECT id FROM stale)
|
|
304
|
+
ORDER BY id ASC
|
|
305
|
+
LIMIT 1000`).all(input.now, oldestReceivedAt);
|
|
306
|
+
if (candidates.length === 0)
|
|
307
|
+
return;
|
|
308
|
+
const placeholders = candidates.map(() => "?").join(", ");
|
|
309
|
+
statement(db, `DELETE FROM packets WHERE id IN (${placeholders})`).run(...candidates.map((row) => row.id));
|
|
232
310
|
}
|
|
233
311
|
export function maxPacketSeq(db, spaceId) {
|
|
234
312
|
const row = statement(db, "SELECT COALESCE(MAX(id), 0) AS seq FROM packets WHERE space_id = ?").get(spaceId);
|
|
@@ -13,6 +13,7 @@ import { ackPacket, getPacketById, getRetainedPacket, insertPacket, listPacketsF
|
|
|
13
13
|
if (parentPort && workerData?.relayStorageWorker === true) {
|
|
14
14
|
const dbPath = String(workerData.dbPath);
|
|
15
15
|
const testStorageDelayMs = Math.max(0, Number(workerData.testStorageDelayMs) || 0);
|
|
16
|
+
const testStorageDelayFile = typeof workerData.testStorageDelayFile === "string" ? workerData.testStorageDelayFile : undefined;
|
|
16
17
|
mkdirSync(path.dirname(dbPath), { recursive: true });
|
|
17
18
|
const db = new DatabaseSync(dbPath);
|
|
18
19
|
db.exec("PRAGMA journal_mode = WAL");
|
|
@@ -26,8 +27,9 @@ if (parentPort && workerData?.relayStorageWorker === true) {
|
|
|
26
27
|
parentPort.postMessage({ type: "ready" });
|
|
27
28
|
parentPort.on("message", (request) => {
|
|
28
29
|
try {
|
|
29
|
-
if (testStorageDelayMs > 0)
|
|
30
|
+
if (testStorageDelayMs > 0 && (!testStorageDelayFile || existsSync(testStorageDelayFile))) {
|
|
30
31
|
Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, testStorageDelayMs);
|
|
32
|
+
}
|
|
31
33
|
const value = request.op === "controlPlane"
|
|
32
34
|
? runControlPlaneAction(controlPlane, request.action, request.input)
|
|
33
35
|
: request.op === "directTicketCreate"
|
|
@@ -252,6 +254,7 @@ export function createRelayDurableStorage(dbPath) {
|
|
|
252
254
|
dbPath,
|
|
253
255
|
pairingTtlMs: 600_000,
|
|
254
256
|
testStorageDelayMs: process.env.HAPPY_ELVES_TEST_STORAGE_DELAY_MS,
|
|
257
|
+
testStorageDelayFile: process.env.HAPPY_ELVES_TEST_STORAGE_DELAY_FILE,
|
|
255
258
|
},
|
|
256
259
|
});
|
|
257
260
|
nextWorker.unref();
|
|
@@ -158,9 +158,12 @@ export function registerWebsocketRoute(app, context, originAllowed) {
|
|
|
158
158
|
});
|
|
159
159
|
await sendSnapshot(context, openedConnection, message.snapshotMode ?? "full");
|
|
160
160
|
broadcastPresence(context, openedConnection.spaceId, await presenceWithManifest(context, openedConnection.spaceId, presence));
|
|
161
|
-
void replayNodePackets(context, openedConnection, message.cursor)
|
|
161
|
+
void replayNodePackets(context, openedConnection, message.cursor).catch((error) => {
|
|
162
|
+
reportDetachedOperationFailure(openedConnection, error);
|
|
163
|
+
});
|
|
162
164
|
}).catch((error) => {
|
|
163
165
|
app.log.warn({ error }, "failed to persist relay presence");
|
|
166
|
+
reportDetachedOperationFailure(openedConnection, error, true);
|
|
164
167
|
});
|
|
165
168
|
}).catch((error) => {
|
|
166
169
|
fail(error instanceof Error ? error.message : "Websocket authentication failed", "AUTH_TOKEN_INVALID");
|
|
@@ -210,10 +213,15 @@ export function registerWebsocketRoute(app, context, originAllowed) {
|
|
|
210
213
|
if (!connection)
|
|
211
214
|
return;
|
|
212
215
|
const closedConnection = connection;
|
|
216
|
+
closedConnection.closed = true;
|
|
213
217
|
closedConnection.publishChainClosed = true;
|
|
214
218
|
if (closedConnection.expiryTimer)
|
|
215
219
|
globalThis.clearTimeout(closedConnection.expiryTimer);
|
|
216
|
-
|
|
220
|
+
if (closedConnection.cursorFlushTimer) {
|
|
221
|
+
globalThis.clearTimeout(closedConnection.cursorFlushTimer);
|
|
222
|
+
closedConnection.cursorFlushTimer = undefined;
|
|
223
|
+
}
|
|
224
|
+
void flushPendingCursor(context, closedConnection).catch(() => undefined);
|
|
217
225
|
unregisterConnection(closedConnection);
|
|
218
226
|
const remaining = connectionsByNode.get(nodeKey(closedConnection.spaceId, closedConnection.nodeId));
|
|
219
227
|
const stillOnline = Boolean(remaining && remaining.size > 0);
|
|
@@ -544,7 +552,9 @@ function pumpSubscriptionReplays(context, connection) {
|
|
|
544
552
|
if (!state || state.running)
|
|
545
553
|
continue;
|
|
546
554
|
state.queued = false;
|
|
547
|
-
void runSubscriptionReplay(context, connection, key, state)
|
|
555
|
+
void runSubscriptionReplay(context, connection, key, state).catch((error) => {
|
|
556
|
+
reportDetachedOperationFailure(connection, error);
|
|
557
|
+
});
|
|
548
558
|
}
|
|
549
559
|
}
|
|
550
560
|
async function runSubscriptionReplay(context, connection, key, state) {
|
|
@@ -613,7 +623,9 @@ function deliverPacket(context, packet) {
|
|
|
613
623
|
if (packet.toNodeId) {
|
|
614
624
|
const connections = connectionsByNode.get(nodeKey(packet.spaceId, packet.toNodeId));
|
|
615
625
|
for (const connection of connections ?? []) {
|
|
616
|
-
void sendPacket(context, connection, packet, "live")
|
|
626
|
+
void sendPacket(context, connection, packet, "live").catch((error) => {
|
|
627
|
+
reportDetachedOperationFailure(connection, error);
|
|
628
|
+
});
|
|
617
629
|
delivered.add(connection);
|
|
618
630
|
}
|
|
619
631
|
}
|
|
@@ -623,7 +635,9 @@ function deliverPacket(context, packet) {
|
|
|
623
635
|
continue;
|
|
624
636
|
if (!connection.subscriptions.some((subscription) => subscriptionMatches(connection, subscription, packet)))
|
|
625
637
|
continue;
|
|
626
|
-
void sendPacket(context, connection, packet, "live")
|
|
638
|
+
void sendPacket(context, connection, packet, "live").catch((error) => {
|
|
639
|
+
reportDetachedOperationFailure(connection, error);
|
|
640
|
+
});
|
|
627
641
|
}
|
|
628
642
|
}
|
|
629
643
|
async function sendPacket(context, connection, packet, source) {
|
|
@@ -666,6 +680,8 @@ async function sendPacket(context, connection, packet, source) {
|
|
|
666
680
|
return true;
|
|
667
681
|
}
|
|
668
682
|
function queueCursorUpdate(context, connection, serverSeq) {
|
|
683
|
+
if (connection.closed)
|
|
684
|
+
return;
|
|
669
685
|
connection.pendingCursorServerSeq = Math.max(connection.pendingCursorServerSeq ?? 0, serverSeq);
|
|
670
686
|
if (connection.cursorFlushTimer)
|
|
671
687
|
return;
|
|
@@ -674,13 +690,35 @@ function queueCursorUpdate(context, connection, serverSeq) {
|
|
|
674
690
|
// keeps high-rate replay from turning every packet into an fsync.
|
|
675
691
|
connection.cursorFlushTimer = setTimeout(() => {
|
|
676
692
|
connection.cursorFlushTimer = undefined;
|
|
677
|
-
void flushPendingCursor(context, connection);
|
|
693
|
+
void flushPendingCursor(context, connection).catch(() => undefined);
|
|
678
694
|
}, 100);
|
|
679
695
|
connection.cursorFlushTimer.unref?.();
|
|
680
696
|
}
|
|
681
697
|
async function flushPendingCursor(context, connection) {
|
|
698
|
+
if (connection.cursorFlushInFlight)
|
|
699
|
+
return await connection.cursorFlushInFlight;
|
|
700
|
+
const run = flushPendingCursorOnce(context, connection);
|
|
701
|
+
let trackedRun;
|
|
702
|
+
trackedRun = run.finally(() => {
|
|
703
|
+
if (connection.cursorFlushInFlight === trackedRun)
|
|
704
|
+
connection.cursorFlushInFlight = undefined;
|
|
705
|
+
// A newer packet can arrive while the write is in flight. Keep one
|
|
706
|
+
// coalesced follow-up timer, but never resurrect retries after the
|
|
707
|
+
// socket has closed.
|
|
708
|
+
if (!connection.closed && connection.pendingCursorServerSeq !== undefined && !connection.cursorFlushTimer) {
|
|
709
|
+
connection.cursorFlushTimer = setTimeout(() => {
|
|
710
|
+
connection.cursorFlushTimer = undefined;
|
|
711
|
+
void flushPendingCursor(context, connection).catch(() => undefined);
|
|
712
|
+
}, 250);
|
|
713
|
+
connection.cursorFlushTimer.unref?.();
|
|
714
|
+
}
|
|
715
|
+
});
|
|
716
|
+
connection.cursorFlushInFlight = trackedRun;
|
|
717
|
+
await trackedRun;
|
|
718
|
+
}
|
|
719
|
+
async function flushPendingCursorOnce(context, connection) {
|
|
682
720
|
if (connection.cursorFlushTimer) {
|
|
683
|
-
clearTimeout(connection.cursorFlushTimer);
|
|
721
|
+
globalThis.clearTimeout(connection.cursorFlushTimer);
|
|
684
722
|
connection.cursorFlushTimer = undefined;
|
|
685
723
|
}
|
|
686
724
|
const serverSeq = connection.pendingCursorServerSeq;
|
|
@@ -700,7 +738,8 @@ async function flushPendingCursor(context, connection) {
|
|
|
700
738
|
connection.pendingCursorServerSeq = Math.max(connection.pendingCursorServerSeq ?? 0, serverSeq);
|
|
701
739
|
// Cursor persistence is recoverable; do not terminate the relay process
|
|
702
740
|
// because a worker/database write was temporarily unavailable.
|
|
703
|
-
|
|
741
|
+
// The single-flight finalizer above owns the one retry timer. In
|
|
742
|
+
// particular, a close cannot leave an unbounded timer chain behind.
|
|
704
743
|
}
|
|
705
744
|
}
|
|
706
745
|
function closeSlowReceiver(connection) {
|
|
@@ -874,7 +913,7 @@ function scheduleConnectionExpiry(context, connection, expiresAt) {
|
|
|
874
913
|
if (!current || current.revoked_at)
|
|
875
914
|
return;
|
|
876
915
|
if (current.expires_at && current.expires_at > context.now()) {
|
|
877
|
-
void scheduleNext();
|
|
916
|
+
void scheduleNext().catch(() => undefined);
|
|
878
917
|
return;
|
|
879
918
|
}
|
|
880
919
|
if (!current.expires_at)
|
|
@@ -888,7 +927,33 @@ function scheduleConnectionExpiry(context, connection, expiresAt) {
|
|
|
888
927
|
}).catch(() => undefined);
|
|
889
928
|
}, Math.min(delay, MAX_EXPIRY_TIMER_DELAY_MS));
|
|
890
929
|
};
|
|
891
|
-
void scheduleNext();
|
|
930
|
+
void scheduleNext().catch(() => undefined);
|
|
931
|
+
}
|
|
932
|
+
function reportDetachedOperationFailure(connection, error, forceClose = false) {
|
|
933
|
+
if (connection.socket.readyState !== 1)
|
|
934
|
+
return;
|
|
935
|
+
const record = error && typeof error === "object" ? error : {};
|
|
936
|
+
const code = typeof record.code === "string" ? record.code : "RELAY_COMMAND_FAILED";
|
|
937
|
+
const message = typeof record.message === "string" ? record.message : String(error);
|
|
938
|
+
send(connection.socket, { type: "server:error", code, message });
|
|
939
|
+
// A detached replay/live send has already lost its in-memory continuation.
|
|
940
|
+
// Close the stream for storage-generation failures so old daemons and
|
|
941
|
+
// controllers, which only reconnect on close, replay their durable outbox
|
|
942
|
+
// and retained packets. The stored ACK boundary remains authoritative; no
|
|
943
|
+
// packet is acknowledged merely because the connection is being closed.
|
|
944
|
+
if (forceClose || isRecoverableStorageFailure(code)) {
|
|
945
|
+
connection.closed = true;
|
|
946
|
+
try {
|
|
947
|
+
connection.socket.close(1011, code);
|
|
948
|
+
}
|
|
949
|
+
catch {
|
|
950
|
+
// The close/error listener still performs cleanup for transports that
|
|
951
|
+
// reject a redundant close.
|
|
952
|
+
}
|
|
953
|
+
}
|
|
954
|
+
}
|
|
955
|
+
function isRecoverableStorageFailure(code) {
|
|
956
|
+
return code === "RELAY_STORAGE_TIMEOUT" || code === "RELAY_STORAGE_UNAVAILABLE" || code === "RELAY_STORAGE_CLOSED";
|
|
892
957
|
}
|
|
893
958
|
async function isScopedControllerConnection(context, connection) {
|
|
894
959
|
const token = await context.tokenRecordAsync(connection.authToken);
|
package/package.json
CHANGED
|
@@ -250,6 +250,10 @@ export type RuntimeHistoricalEventPage = {
|
|
|
250
250
|
events: RuntimeHistoricalEvent[];
|
|
251
251
|
hasMoreBefore?: boolean;
|
|
252
252
|
nextBefore?: string;
|
|
253
|
+
/** Forward event cursor for incremental readers. */
|
|
254
|
+
nextAfter?: string;
|
|
255
|
+
/** True when the reader understood a forward cursor, including an empty page. */
|
|
256
|
+
cursorSupported?: boolean;
|
|
253
257
|
currentHead?: string;
|
|
254
258
|
lastTurnId?: string;
|
|
255
259
|
};
|
|
@@ -1113,6 +1113,8 @@ export declare const runtimeReadCommandResultSchema: z.ZodDiscriminatedUnion<[z.
|
|
|
1113
1113
|
}, z.core.$strict>>;
|
|
1114
1114
|
hasMoreBefore: z.ZodOptional<z.ZodBoolean>;
|
|
1115
1115
|
nextBefore: z.ZodOptional<z.ZodString>;
|
|
1116
|
+
nextAfter: z.ZodOptional<z.ZodString>;
|
|
1117
|
+
cursorSupported: z.ZodOptional<z.ZodBoolean>;
|
|
1116
1118
|
currentHead: z.ZodOptional<z.ZodString>;
|
|
1117
1119
|
lastTurnId: z.ZodOptional<z.ZodString>;
|
|
1118
1120
|
errorCode: z.ZodOptional<z.ZodString>;
|
|
@@ -487,6 +487,8 @@ export const runtimeReadCommandResultSchema = z.discriminatedUnion("action", [
|
|
|
487
487
|
events: z.array(runtimeReadEventSchema),
|
|
488
488
|
hasMoreBefore: z.boolean().optional(),
|
|
489
489
|
nextBefore: z.string().min(1).optional(),
|
|
490
|
+
nextAfter: z.string().min(1).optional(),
|
|
491
|
+
cursorSupported: z.boolean().optional(),
|
|
490
492
|
currentHead: z.string().min(1).optional(),
|
|
491
493
|
lastTurnId: z.string().min(1).optional(),
|
|
492
494
|
errorCode: z.string().min(1).optional(),
|
|
@@ -349,6 +349,8 @@ export type RuntimeTurnEventsPageResult = {
|
|
|
349
349
|
events: RuntimeReadEvent[];
|
|
350
350
|
hasMoreBefore?: boolean;
|
|
351
351
|
nextBefore?: string;
|
|
352
|
+
nextAfter?: string;
|
|
353
|
+
cursorSupported?: boolean;
|
|
352
354
|
currentHead?: string;
|
|
353
355
|
lastTurnId?: string;
|
|
354
356
|
errorCode?: string;
|
|
@@ -138,6 +138,7 @@ export type ControllerClientMessage = {
|
|
|
138
138
|
runtimeTurnId?: string;
|
|
139
139
|
before?: string;
|
|
140
140
|
beforeItemId?: string;
|
|
141
|
+
after?: string;
|
|
141
142
|
afterItemId?: string;
|
|
142
143
|
limitItems?: number;
|
|
143
144
|
maxBytes?: number;
|
|
@@ -767,6 +768,7 @@ export type MachineCommand = {
|
|
|
767
768
|
runtimeTurnId?: string;
|
|
768
769
|
before?: string;
|
|
769
770
|
beforeItemId?: string;
|
|
771
|
+
after?: string;
|
|
770
772
|
afterItemId?: string;
|
|
771
773
|
limitItems?: number;
|
|
772
774
|
maxBytes?: number;
|
|
@@ -155,6 +155,7 @@ const controllerMessageSchema = z.union([
|
|
|
155
155
|
runtimeTurnId: z.string().min(1).optional(),
|
|
156
156
|
before: z.string().min(1).optional(),
|
|
157
157
|
beforeItemId: z.string().min(1).optional(),
|
|
158
|
+
after: z.string().min(1).optional(),
|
|
158
159
|
afterItemId: z.string().min(1).optional(),
|
|
159
160
|
limitItems: z.number().int().positive().max(1000).optional(),
|
|
160
161
|
maxBytes: z.number().int().positive().max(2_000_000).optional(),
|
|
@@ -885,6 +886,7 @@ const machineCommandSchema = z.union([
|
|
|
885
886
|
runtimeTurnId: z.string().min(1).optional(),
|
|
886
887
|
before: z.string().min(1).optional(),
|
|
887
888
|
beforeItemId: z.string().min(1).optional(),
|
|
889
|
+
after: z.string().min(1).optional(),
|
|
888
890
|
afterItemId: z.string().min(1).optional(),
|
|
889
891
|
limitItems: z.number().int().positive().max(1000).optional(),
|
|
890
892
|
maxBytes: z.number().int().positive().max(2_000_000).optional(),
|