feat: add graceful server shutdown with database and socket disconnection

Author: muneebhashone

src/lib/database.ts

@@ -12,3 +12,8 @@ export const connectDatabase = async () => {
 		process.exit(1);
 	}
 };
+
+export const disconnectDatabase = async () => {
+	await mongoose.disconnect();
+	logger.info("Database disconnected");
+};

src/main.ts

@@ -14,7 +14,7 @@ import session from "express-session";
 import helmet from "helmet";
 import morgan from "morgan";
 import config from "./config/config.service";
-import { connectDatabase } from "./lib/database";
+import { connectDatabase, disconnectDatabase } from "./lib/database";
 import logger, { httpLogger } from "./lib/logger.service";
 import { useSocketIo } from "./lib/realtime.server";
 import redisStore from "./lib/session.store";
@@ -27,17 +27,17 @@ import YAML from "yaml";
 import { convertDocumentationToYaml } from "./openapi/swagger-doc-generator";
 import globalErrorHandler from "./utils/globalErrorHandler";
 
-const boostrapServer = async () => {
-	await connectDatabase();
-
-	const app = express();
+const app = express();
 
 	app.set("trust proxy", true);
 
 	const server = createServer(app);
 
 	const io = useSocketIo(server);
 
+const boostrapServer = async () => {
+	await connectDatabase();
+
 	app.use((req, _, next) => {
 		req.io = io;
 		next();
@@ -116,3 +116,17 @@ boostrapServer().catch((err) => {
 	logger.error(err.message);
 	process.exit(1);
 });
+
+for (const signal of ["SIGINT", "SIGTERM"]) {
+	process.on(signal, async () => {
+		await disconnectDatabase();
+		logger.info("Server is shutting down...");
+		io.disconnectSockets(true);
+		logger.info("Server disconnected from sockets");
+		server.close();
+		logger.info("Server closed");
+		process.exit(0);
+	});
+}
+
+

src/modules/user/user.controller.ts

@@ -40,6 +40,7 @@ export const handleCreateSuperAdmin = async (
 	_: Request<unknown, unknown, unknown>,
 	res: Response,
 ) => {
+	
 	const user = await createUser({
 		email: config.ADMIN_EMAIL,
 		name: "Super Admin",