modify: app 실행과 index 업데이트, 그에 따른 dbms connection 관련 설정 업데이트

Author: Nuung

src/configs/db.config.ts

@@ -27,7 +27,14 @@ if (process.env.NODE_ENV === 'production') {
 
 const pool = new Pool(poolConfig);
 
-async function initializeDatabase(): Promise<void> {
+/**
+ * 데이터베이스 연결을 초기화하고 TimescaleDB 확장을 보장
+ * 최대 3회 재시도하며, 실패 시 서버를 종료
+ *
+ * @throws 연결에 3회 실패하면 서버가 종료
+ * @returns {Promise<void>} 연결 및 확장 완료 시 resolve되는 프로미스
+ */
+export async function initializeDatabase(): Promise<void> {
   const maxRetries = 3;
   let delay = 800;
 
@@ -65,9 +72,21 @@ async function initializeDatabase(): Promise<void> {
   }
 }
 
-initializeDatabase().catch((error) => {
-  logger.error('데이터베이스 초기화 중 예상치 못한 오류:', error);
-  process.exit(1); // 치명적 오류시 서버 종료
-});
+
+/**
+ * 데이터베이스 커넥션 풀을 안정적으로 직접 종료하는 함수
+ *
+ * @throws 데이터베이스 연결 종료에 실패할 경우 에러
+ * @returns {Promise<void>} 데이터베이스 연결이 종료되면 resolve되는 프로미스
+ */
+export async function closeDatabase(): Promise<void> {
+  try {
+    await pool.end();
+    logger.info('데이터베이스 연결이 정상적으로 종료되었습니다.');
+  } catch (error) {
+    logger.error('데이터베이스 연결 종료 중 오류 발생:', error);
+    throw error;
+  }
+}
 
 export default pool;

src/index.ts

@@ -1,45 +1,91 @@
 import app from '@/app';
 import logger from '@/configs/logger.config';
 import { closeCache } from './configs/cache.config';
+import { closeDatabase, initializeDatabase } from './configs/db.config';
+import { Server } from 'http';
 
 const port = parseInt(process.env.PORT || '8080', 10);
 
-const server = app.listen(port, () => {
-  logger.info(`Server running on port ${port}`);
-  logger.info(`Environment: ${process.env.NODE_ENV}`);
-  if (process.env.NODE_ENV !== 'production') {
-    logger.info(`API Docs: http://localhost:${port}/api-docs`);
+
+async function startServer() {
+  try {
+    // 데이터베이스 초기화
+    await initializeDatabase();
+    logger.info('데이터베이스 초기화 완료');
+
+    // 서버 시작
+    const server = app.listen(port, () => {
+      logger.info(`Server running on port ${port}`);
+      logger.info(`Environment: ${process.env.NODE_ENV}`);
+      if (process.env.NODE_ENV !== 'production') {
+        logger.info(`API Docs: http://localhost:${port}/api-docs`);
+      }
+      logger.info(`Health Check: http://localhost:${port}/health`);
+    });
+
+    // Graceful shutdown 핸들러 설정
+    setupGracefulShutdown(server);
+
+  } catch (error) {
+    logger.error('서버 시작 중 오류 발생:', error);
+    process.exit(1);
   }
-  logger.info(`Health Check: http://localhost:${port}/health`);
-});
+}
+
+function setupGracefulShutdown(server: Server) {
+  // 기본적인 graceful shutdown 추가
+  const gracefulShutdown = async (signal: string) => {
+    logger.info(`${signal} received, shutting down gracefully`);
+
+    try {
+      // HTTP 서버 종료
+      await new Promise<void>((resolve) => {
+        server.close(() => {
+          logger.info('HTTP server closed');
+          resolve();
+        });
+      });
 
-// 기본적인 graceful shutdown 추가
-const gracefulShutdown = async (signal: string) => {
-  logger.info(`${signal} received, shutting down gracefully`);
-  await closeCache();
+      // 데이터베이스 연결 종료
+      await closeDatabase();
 
-  server.close(() => {
-    logger.info('HTTP server closed');
-    process.exit(0);
+      // 캐시 연결 종료
+      await closeCache();
+
+      logger.info('Graceful shutdown completed');
+      process.exit(0);
+    } catch (error) {
+      logger.error('Graceful shutdown 중 오류 발생:', error);
+      process.exit(1);
+    }
+  };
+
+  // 시그널 핸들러 등록
+  process.on('SIGTERM', async () => gracefulShutdown('SIGTERM'));
+  process.on('SIGINT', async () => gracefulShutdown('SIGINT'));
+
+  // 예상치 못한 에러 처리
+  process.on('uncaughtException', async (error) => {
+    logger.error('Uncaught Exception:', error);
+    await gracefulShutdown('UNCAUGHT_EXCEPTION');
+  });
+
+  process.on('unhandledRejection', async (reason, promise) => {
+    logger.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    await gracefulShutdown('UNHANDLED_REJECTION');
   });
 
   // 강제 종료 타이머 (10초)
-  setTimeout(() => {
+  const forceExitTimer = setTimeout(() => {
     logger.error('Could not close connections in time, forcefully shutting down');
     process.exit(1);
   }, 10000);
-};
-
-process.on('SIGTERM', async () => gracefulShutdown('SIGTERM'));
-process.on('SIGINT', async () => gracefulShutdown('SIGINT'));
 
-// 예상치 못한 에러 처리
-process.on('uncaughtException', async (error) => {
-  logger.error('Uncaught Exception:', error);
-  await gracefulShutdown('UNCAUGHT_EXCEPTION');
-});
+  // graceful shutdown이 완료되면 타이머 해제
+  process.on('exit', () => {
+    clearTimeout(forceExitTimer);
+  });
+}
 
-process.on('unhandledRejection', async (reason, promise) => {
-  logger.error('Unhandled Rejection at:', promise, 'reason:', reason);
-  await gracefulShutdown('UNHANDLED_REJECTION');
-});
+// 서버 시작
+startServer();