refactor(error): improve database connection health check

Author: m-mdy-m

src/app.ts

@@ -28,11 +28,11 @@ async function main() {
   });
 
   process.on('unhandledRejection', (reason) => {
-    console.error('Unhandled Rejection:', reason);
+    logger.error(`${reason}:`, 'Unhandled Rejection');
   });
 
-  process.on('uncaughtException', async (err) => {
-    console.error('Uncaught Exception:', err);
+  process.on('uncaughtException', async (err: any) => {
+    logger.error(`${err}:`, 'Uncaught Exception');
     await shutdown(botInstance, 1);
   });
 }
@@ -41,7 +41,7 @@ async function shutdown(botInstance: CopBot, exitCode = 0) {
   try {
     const db = ServiceProvider.getInstance();
     await db.close();
-    logger.info('Database closed.');
+    logger.warn('Database closed.');
     await botInstance.stop();
   } catch (err) {
     console.error('Error during shutdown:', err);

src/bot/index.ts

@@ -62,8 +62,19 @@ export class CopBot {
           }
         });
         app.get('/health', async (_, res) => {
-          const isHealthy = await ServiceProvider.getInstance().healthCheck();
-          res.status(isHealthy ? 200 : 500).send(isHealthy ? 'OK' : 'Database not reachable');
+          try {
+            const isHealthy = await ServiceProvider.getInstance().healthCheck();
+            if (isHealthy) {
+              res.status(200).send('Database is healthy');
+              logger.info('Database connection is healthy.');
+            } else {
+              res.status(500).send('Database not reachable');
+              logger.error('Database not reachable.');
+            }
+          } catch (error: any) {
+            res.status(500).send('Database connection error');
+            logger.error(`Database health check failed: ${error.message}`);
+          }
         });
         const port = process.env.PORT || 3000;
         app.listen(port, async () => {

src/database/ConnectionPool.ts

@@ -72,6 +72,9 @@ export class ConnectionPool {
     return new Pool({
       connectionString,
       ssl: this._isProduction === 'production' ? { rejectUnauthorized: false } : false,
+      connectionTimeoutMillis: 500,
+      max: 10,
+      idleTimeoutMillis: 500,
     });
   }
 }

src/database/models/User.ts

@@ -3,7 +3,7 @@ import { User } from '../../types/database/TablesTypes';
 import { DatabaseService } from '../service/Database';
 export class UserService {
   private _db: DatabaseService;
-  constructor(private _client: PoolClient) {
+  constructor(_client: PoolClient) {
     this._db = new DatabaseService(_client);
   }
   async create(user: Omit<User, 'id'>): Promise<User> {

src/service/database/ServiceProvider.ts

@@ -58,6 +58,7 @@ export class ServiceProvider {
   async healthCheck(): Promise<boolean> {
     try {
       const client = await this.getPoolClint();
+      await client.query('SELECT NOW()');
       client.release();
       console.log('Database is healthy.');
       return true;