chore: graceful shutdown

Author: ChaituVR

src/helpers/mysql.ts

@@ -37,4 +37,15 @@ const sequencerDB = mysql.createPool(sequencerConfig);
 
 bluebird.promisifyAll([Pool, Connection]);
 
+export const closeDatabase = (): Promise<void> => {
+  return new Promise(resolve => {
+    hubDB.end(() => {
+      sequencerDB.end(() => {
+        console.log('[mysql] Database connection pools closed');
+        resolve();
+      });
+    });
+  });
+};
+
 export { hubDB as default, sequencerDB };

src/index.ts

@@ -11,6 +11,7 @@ import initMetrics from './helpers/metrics';
 import rateLimit from './helpers/rateLimit';
 import refreshSpacesCache from './helpers/spaces';
 import './helpers/strategies';
+import { closeDatabase } from './helpers/mysql';
 
 const app = express();
 const PORT = process.env.PORT || 3000;
@@ -32,4 +33,26 @@ app.use('/graphql', graphql);
 fallbackLogger(app);
 app.get('/*', (req, res) => res.redirect('/api'));
 
-app.listen(PORT, () => log.info(`Started on: http://localhost:${PORT}`));
+const server = app.listen(PORT, () =>
+  log.info(`Started on: http://localhost:${PORT}`)
+);
+
+const gracefulShutdown = async (signal: string) => {
+  console.log(`Received ${signal}. Starting graceful shutdown...`);
+
+  server.close(async () => {
+    console.log('Express server closed.');
+
+    try {
+      await closeDatabase();
+      console.log('Graceful shutdown completed.');
+      process.exit(0);
+    } catch (error) {
+      console.error('Error during shutdown:', error);
+      process.exit(1);
+    }
+  });
+};
+
+process.on('SIGINT', () => gracefulShutdown('SIGINT'));
+process.on('SIGTERM', () => gracefulShutdown('SIGTERM'));