70 lines
2.2 KiB
TypeScript
70 lines
2.2 KiB
TypeScript
import type { ServerControl } from "./server-control";
|
|
import type { Logger } from "sync-client";
|
|
|
|
export class ServerManager {
|
|
private readonly activeServers = new Set<ServerControl>();
|
|
private readonly logger: Logger;
|
|
private isShuttingDown = false;
|
|
|
|
public constructor(logger: Logger) {
|
|
this.logger = logger;
|
|
}
|
|
|
|
public track(server: ServerControl): void {
|
|
this.activeServers.add(server);
|
|
}
|
|
|
|
public untrack(server: ServerControl): void {
|
|
this.activeServers.delete(server);
|
|
}
|
|
|
|
public async stopAll(): Promise<void> {
|
|
if (this.isShuttingDown) {
|
|
return;
|
|
}
|
|
this.isShuttingDown = true;
|
|
|
|
const servers = Array.from(this.activeServers);
|
|
// eslint-disable-next-line no-restricted-properties
|
|
await Promise.all(
|
|
servers.map(async (server) => {
|
|
try {
|
|
await server.stop();
|
|
} catch {
|
|
// Best-effort cleanup during shutdown
|
|
}
|
|
})
|
|
);
|
|
}
|
|
|
|
public installSignalHandlers(): void {
|
|
process.on("SIGINT", () => {
|
|
this.logger.info("Received SIGINT, shutting down...");
|
|
void this.stopAll()
|
|
.catch(() => {
|
|
/* no-op */
|
|
})
|
|
.then(() => process.exit(130));
|
|
});
|
|
|
|
process.on("SIGTERM", () => {
|
|
this.logger.info("Received SIGTERM, shutting down...");
|
|
void this.stopAll()
|
|
.catch(() => {
|
|
/* no-op */
|
|
})
|
|
.then(() => process.exit(143));
|
|
});
|
|
|
|
// Last-resort synchronous cleanup. Runs even when the process is
|
|
// exiting via process.exit() from unhandledRejection /
|
|
// uncaughtException — paths where async stopAll() cannot complete.
|
|
// Kill only the server processes owned by this runner. SIGKILL also
|
|
// terminates a SIGSTOP'd process; forceKillSync's SIGCONT is harmless.
|
|
process.on("exit", () => {
|
|
for (const server of this.activeServers) {
|
|
server.forceKillSync();
|
|
}
|
|
});
|
|
}
|
|
}
|