root
68 lines
2.3 KiB
TypeScript
68 lines
2.3 KiB
TypeScript
import { fileURLToPath } from "node:url";
|
|
import { NativeConnection, Worker } from "@temporalio/worker";
|
|
import { createServerContext } from "./bootstrap";
|
|
import { loadConfig } from "./config";
|
|
import { createActivities } from "./dispatcher";
|
|
import { requiredOption } from "./runtime-args";
|
|
import { temporalAddress, TemporalGateway } from "./temporal-client";
|
|
|
|
const config = loadConfig(requiredOption("--config"));
|
|
const runtimeId = requiredOption("--runtime");
|
|
const target = config.runtime.serverTargets[runtimeId];
|
|
if (!target) throw new Error(`runtime.serverTargets.${runtimeId} does not exist`);
|
|
|
|
const context = createServerContext(config, target);
|
|
const connection = await NativeConnection.connect({ address: temporalAddress(config) });
|
|
const worker = await Worker.create({
|
|
connection,
|
|
namespace: config.temporal.namespace,
|
|
taskQueue: config.temporal.taskQueue,
|
|
workflowsPath: fileURLToPath(new URL("./workflows.ts", import.meta.url)),
|
|
activities: createActivities({ lottery: context.service, scores: context.monitor }),
|
|
});
|
|
const temporal = await TemporalGateway.connect(config);
|
|
const schedule = await temporal.ensureScoreSchedule();
|
|
let state: "ready" | "stopping" = "ready";
|
|
const health = Bun.serve({
|
|
hostname: target.workerHealthHost,
|
|
port: target.workerHealthPort,
|
|
fetch: () => Response.json({
|
|
ok: state === "ready",
|
|
component: "apistate-worker",
|
|
state,
|
|
namespace: config.temporal.namespace,
|
|
taskQueue: config.temporal.taskQueue,
|
|
schedule,
|
|
}, { status: state === "ready" ? 200 : 503 }),
|
|
});
|
|
|
|
console.log(JSON.stringify({
|
|
ok: true,
|
|
component: "apistate-worker",
|
|
runtime: runtimeId,
|
|
health: health.url.toString(),
|
|
temporalNamespace: config.temporal.namespace,
|
|
temporalTaskQueue: config.temporal.taskQueue,
|
|
schedule,
|
|
valuesPrinted: false,
|
|
}));
|
|
|
|
let stopping = false;
|
|
async function stop(): Promise<void> {
|
|
if (stopping) return;
|
|
stopping = true;
|
|
state = "stopping";
|
|
health.stop(true);
|
|
worker.shutdown();
|
|
}
|
|
for (const signal of ["SIGINT", "SIGTERM"] as const) process.on(signal, () => void stop());
|
|
|
|
try {
|
|
await worker.run();
|
|
} finally {
|
|
await temporal.close();
|
|
await connection.close();
|
|
context.close();
|
|
console.log(JSON.stringify({ ok: true, component: "apistate-worker", state: "stopped", valuesPrinted: false }));
|
|
}
|