Handle termination signals to stop container faster

src/index.ts

@@ -1,3 +1,18 @@
+import http from "http";
+import process from "process";
+
+import db from "./db";
+import logger from "./logger";
 import { startServer } from "./server";
 
-startServer();
+function shutdown(server: http.Server): void {
+  logger.info("Shutting down");
+  server.close(() => {
+    db.destroy();
+    process.exit(1);
+  });
+}
+
+const server = startServer();
+process.on("SIGINT", () => shutdown(server));
+process.on("SIGTERM", () => shutdown(server));

src/server.ts

@@ -18,9 +18,7 @@ export function startServer(): http.Server {
   const broadcast = webSocketServer(httpServer);
 
   const feedPoller = pollFeed(broadcast);
-  httpServer.on("close", () => {
-    feedPoller.destroy();
-  });
+  httpServer.on("close", () => feedPoller.destroy());
 
   return httpServer;
 }