feat: add shadow LangGraph orchestrator runtime
This commit is contained in:
@@ -0,0 +1,65 @@
|
||||
import { pathToFileURL } from 'node:url';
|
||||
import { createOrchestratorApp } from './app.mjs';
|
||||
import { createOrchestratorCheckpoint } from './checkpoint.mjs';
|
||||
import { createLangGraphOrchestratorRuntime } from './runtime.mjs';
|
||||
|
||||
function positivePort(value, fallback = 8093) {
|
||||
const parsed = Number(value);
|
||||
if (!Number.isInteger(parsed) || parsed < 1 || parsed > 65535) return fallback;
|
||||
return parsed;
|
||||
}
|
||||
|
||||
export async function startOrchestratorServer({
|
||||
env = process.env,
|
||||
logger = console,
|
||||
} = {}) {
|
||||
const checkpoint = await createOrchestratorCheckpoint({
|
||||
mode: env.MEMIND_ORCHESTRATOR_CHECKPOINT_MODE,
|
||||
connectionString: env.MEMIND_ORCHESTRATOR_DATABASE_URL,
|
||||
schema: env.MEMIND_ORCHESTRATOR_DATABASE_SCHEMA,
|
||||
});
|
||||
const runtime = createLangGraphOrchestratorRuntime({
|
||||
checkpointer: checkpoint.checkpointer,
|
||||
checkpointKind: checkpoint.kind,
|
||||
durable: checkpoint.durable,
|
||||
checkpointProbe: checkpoint.probe,
|
||||
});
|
||||
const app = createOrchestratorApp({
|
||||
runtime,
|
||||
serviceToken: env.MEMIND_ORCHESTRATOR_SERVICE_TOKEN,
|
||||
});
|
||||
const host = String(env.MEMIND_ORCHESTRATOR_HOST ?? '127.0.0.1').trim() || '127.0.0.1';
|
||||
const port = positivePort(env.MEMIND_ORCHESTRATOR_PORT, 8093);
|
||||
const server = await new Promise((resolve, reject) => {
|
||||
const listening = app.listen(port, host, () => resolve(listening));
|
||||
listening.once('error', reject);
|
||||
});
|
||||
logger.log(`[orchestrator] listening on http://${host}:${port} (${checkpoint.kind})`);
|
||||
|
||||
async function close() {
|
||||
await new Promise((resolve, reject) => {
|
||||
server.close((error) => (error ? reject(error) : resolve()));
|
||||
});
|
||||
await checkpoint.close();
|
||||
}
|
||||
|
||||
return { app, server, runtime, checkpoint, close };
|
||||
}
|
||||
|
||||
const isEntrypoint = process.argv[1]
|
||||
&& import.meta.url === pathToFileURL(process.argv[1]).href;
|
||||
|
||||
if (isEntrypoint) {
|
||||
const running = await startOrchestratorServer();
|
||||
const shutdown = async (signal) => {
|
||||
console.log(`[orchestrator] received ${signal}, shutting down`);
|
||||
await running.close();
|
||||
process.exit(0);
|
||||
};
|
||||
process.once('SIGINT', () => void shutdown('SIGINT'));
|
||||
process.once('SIGTERM', () => void shutdown('SIGTERM'));
|
||||
}
|
||||
|
||||
export const orchestratorServerInternals = {
|
||||
positivePort,
|
||||
};
|
||||
Reference in New Issue
Block a user