feat: add daemon skeleton with lifecycle management
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
import { Lifecycle } from './lifecycle.js';
|
||||
import type { Config } from '../config/index.js';
|
||||
|
||||
export interface DaemonContext {
|
||||
config: Config;
|
||||
lifecycle: Lifecycle;
|
||||
}
|
||||
|
||||
export async function startDaemon(config: Config): Promise<DaemonContext> {
|
||||
const lifecycle = new Lifecycle();
|
||||
|
||||
// Register signal handlers
|
||||
const signalHandler = () => {
|
||||
lifecycle.shutdown().then(() => process.exit(0));
|
||||
};
|
||||
|
||||
process.on('SIGINT', signalHandler);
|
||||
process.on('SIGTERM', signalHandler);
|
||||
|
||||
lifecycle.onShutdown(async () => {
|
||||
process.off('SIGINT', signalHandler);
|
||||
process.off('SIGTERM', signalHandler);
|
||||
});
|
||||
|
||||
console.log('Flynn daemon started');
|
||||
|
||||
return { config, lifecycle };
|
||||
}
|
||||
|
||||
export { Lifecycle } from './lifecycle.js';
|
||||
@@ -0,0 +1,39 @@
|
||||
import { describe, it, expect, vi } from 'vitest';
|
||||
import { Lifecycle } from './lifecycle.js';
|
||||
|
||||
describe('Lifecycle', () => {
|
||||
it('registers and calls shutdown handlers in reverse order', async () => {
|
||||
const lifecycle = new Lifecycle();
|
||||
const calls: number[] = [];
|
||||
|
||||
lifecycle.onShutdown(async () => { calls.push(1); });
|
||||
lifecycle.onShutdown(async () => { calls.push(2); });
|
||||
lifecycle.onShutdown(async () => { calls.push(3); });
|
||||
|
||||
await lifecycle.shutdown();
|
||||
|
||||
expect(calls).toEqual([3, 2, 1]);
|
||||
});
|
||||
|
||||
it('only shuts down once', async () => {
|
||||
const lifecycle = new Lifecycle();
|
||||
let count = 0;
|
||||
|
||||
lifecycle.onShutdown(async () => { count++; });
|
||||
|
||||
await lifecycle.shutdown();
|
||||
await lifecycle.shutdown();
|
||||
|
||||
expect(count).toBe(1);
|
||||
});
|
||||
|
||||
it('reports running state', async () => {
|
||||
const lifecycle = new Lifecycle();
|
||||
|
||||
expect(lifecycle.isRunning).toBe(true);
|
||||
|
||||
await lifecycle.shutdown();
|
||||
|
||||
expect(lifecycle.isRunning).toBe(false);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,34 @@
|
||||
type ShutdownHandler = () => Promise<void>;
|
||||
|
||||
export class Lifecycle {
|
||||
private shutdownHandlers: ShutdownHandler[] = [];
|
||||
private shuttingDown = false;
|
||||
private _isRunning = true;
|
||||
|
||||
get isRunning(): boolean {
|
||||
return this._isRunning;
|
||||
}
|
||||
|
||||
onShutdown(handler: ShutdownHandler): void {
|
||||
this.shutdownHandlers.push(handler);
|
||||
}
|
||||
|
||||
async shutdown(): Promise<void> {
|
||||
if (this.shuttingDown) return;
|
||||
this.shuttingDown = true;
|
||||
this._isRunning = false;
|
||||
|
||||
console.log('Shutting down...');
|
||||
|
||||
// Execute handlers in reverse order (LIFO)
|
||||
for (const handler of [...this.shutdownHandlers].reverse()) {
|
||||
try {
|
||||
await handler();
|
||||
} catch (error) {
|
||||
console.error('Shutdown handler error:', error);
|
||||
}
|
||||
}
|
||||
|
||||
console.log('Shutdown complete');
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user