Add graceful shutdown handler for SIGTERM and SIGINT

examples/basic-server.ts

@@ -1,4 +1,4 @@
-import { createServer } from '@linkforty/core';
+import { createServer, registerGracefulShutdown } from '@linkforty/core';
 
 async function start() {
   const server = await createServer({
@@ -18,6 +18,8 @@ async function start() {
     host: '0.0.0.0',
   });
 
+  registerGracefulShutdown(server);
+
   console.log('LinkForty server running on http://localhost:3000');
   console.log('');
   console.log('API Endpoints:');

src/index.ts

@@ -1,7 +1,7 @@
 import Fastify, { FastifyInstance } from 'fastify';
 import cors from '@fastify/cors';
 import redis from '@fastify/redis';
-import { initializeDatabase, DatabaseOptions } from './lib/database.js';
+import { initializeDatabase, DatabaseOptions, db } from './lib/database.js';
 import { redirectRoutes } from './routes/redirect.js';
 import { linkRoutes } from './routes/links.js';
 import { analyticsRoutes } from './routes/analytics.js';
@@ -55,6 +55,24 @@ export async function createServer(options: ServerOptions = {}) {
   return fastify;
 }
 
+export function registerGracefulShutdown(fastify: FastifyInstance): void {
+  const shutdown = async (signal: string) => {
+    fastify.log.info(`Received ${signal}, shutting down gracefully`);
+    try {
+      await fastify.close();
+      await db.end();
+      fastify.log.info('Shutdown complete');
+      process.exit(0);
+    } catch (err) {
+      fastify.log.error(err, 'Error during shutdown');
+      process.exit(1);
+    }
+  };
+
+  process.once('SIGTERM', () => shutdown('SIGTERM'));
+  process.once('SIGINT', () => shutdown('SIGINT'));
+}
+
 // Re-export utilities and types
 export * from './lib/utils.js';
 export * from './lib/database.js';