63 lines
2.0 KiB
JavaScript
63 lines
2.0 KiB
JavaScript
|
import { spawnSync } from 'node:child_process';
|
||
|
|
import path from 'node:path';
|
||
|
|
|
||
|
|
export function sleepMs(ms) {
|
||
|
|
return new Promise((resolve) => {
|
||
|
|
setTimeout(resolve, ms);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
export function readAgentRunWorkerQueueCounts(root, env) {
|
||
|
|
const result = spawnSync(
|
||
|
|
process.execPath,
|
||
|
|
[path.join(root, 'scripts', 'agent-run-worker.mjs'), '--status'],
|
||
|
|
{ cwd: root, env, encoding: 'utf8' },
|
||
|
|
);
|
||
|
|
const combined = `${result.stdout ?? ''}\n${result.stderr ?? ''}`;
|
||
|
|
const match = combined.match(/\{[\s\S]*\}/);
|
||
|
|
if (!match) return null;
|
||
|
|
try {
|
||
|
|
const payload = JSON.parse(match[0]);
|
||
|
|
const counts = payload?.queue?.statusCounts ?? {};
|
||
|
|
return {
|
||
|
|
queued: Number(counts.queued ?? 0),
|
||
|
|
running: Number(counts.running ?? 0),
|
||
|
|
inFlight: Number(payload?.queue?.inFlight ?? 0),
|
||
|
|
};
|
||
|
|
} catch {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function waitForAgentRunWorkerIdle(root, env, {
|
||
|
|
timeoutMs = Number(process.env.GOOSE_V149_WORKER_IDLE_TIMEOUT_MS ?? 900_000),
|
||
|
|
pollMs = Number(process.env.GOOSE_V149_WORKER_IDLE_POLL_MS ?? 5_000),
|
||
|
|
logPrefix = '[goose-v149-worker-idle]',
|
||
|
|
} = {}) {
|
||
|
|
const deadline = Date.now() + timeoutMs;
|
||
|
|
while (Date.now() < deadline) {
|
||
|
|
const counts = readAgentRunWorkerQueueCounts(root, env);
|
||
|
|
if (counts && counts.running === 0 && counts.queued === 0 && counts.inFlight === 0) {
|
||
|
|
console.log(`${logPrefix} worker idle`);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
console.log(
|
||
|
|
`${logPrefix} waiting `
|
||
|
|
+ `(running=${counts?.running ?? '?'} queued=${counts?.queued ?? '?'} inFlight=${counts?.inFlight ?? '?'})`,
|
||
|
|
);
|
||
|
|
await sleepMs(pollMs);
|
||
|
|
}
|
||
|
|
const counts = readAgentRunWorkerQueueCounts(root, env);
|
||
|
|
throw new Error(
|
||
|
|
`worker not idle within ${timeoutMs}ms `
|
||
|
|
+ `(running=${counts?.running ?? '?'} queued=${counts?.queued ?? '?'})`,
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
export const gooseV149PortalAgentEnv = {
|
||
|
|
GOOSE_V149_PORTAL_RESUME_TIMEOUT_MS:
|
||
|
|
process.env.GOOSE_V149_PORTAL_RESUME_TIMEOUT_MS ?? '900000',
|
||
|
|
GOOSE_V149_PAGE_E2E_TIMEOUT_MS:
|
||
|
|
process.env.GOOSE_V149_PAGE_E2E_TIMEOUT_MS ?? '900000',
|
||
|
|
};
|