PEER-208 Add health check and helmet

Signed-off-by: SeeuSim <[email protected]>

Author: SeeuSim

backend/user/package.json

@@ -21,6 +21,7 @@
     "drizzle-orm": "^0.33.0",
     "express": "^4.21.0",
     "express-rate-limit": "^7.4.0",
+    "helmet": "^7.1.0",
     "http-status-codes": "^2.3.0",
     "jsonwebtoken": "^9.0.2",
     "pino": "^9.4.0",

backend/user/src/index.ts

@@ -1,9 +1,10 @@
-import app from './server';
+import app, { dbHealthCheck } from './server';
 import { logger } from './lib/utils';
 
 const port = process.env.PORT || 8001;
 
 const listenMessage = `App listening on port: ${port}`;
-app.listen(port, () => {
+app.listen(port, async () => {
+  await dbHealthCheck();
   logger.info(listenMessage);
 });

backend/user/src/server.ts

@@ -1,21 +1,37 @@
 import express, { json } from 'express';
 import pino from 'pino-http';
-import { db, users } from './lib/db';
+import { db } from './lib/db';
 import authRoutes from './routes/auth';
+import helmet from 'helmet';
+import { sql } from 'drizzle-orm';
+import { logger } from './lib/utils';
+import { exit } from 'process';
 
 const app = express();
 app.use(pino());
 app.use(json());
+app.use(helmet());
 app.use('/auth', authRoutes);
 app.get('/', async (_req, res) => {
   res.json({
     message: 'OK',
   });
 });
 
+export const dbHealthCheck = async () => {
+  try {
+    await db.execute(sql`SELECT 1`);
+    logger.info('Connected to DB');
+  } catch (error) {
+    const { message } = error as Error;
+    logger.error('Cannot connect to DB: ' + message);
+    exit(1);
+  }
+};
+
 // Ensure DB service is up before running.
 app.get('/test-db', async (_req, res) => {
-  await db.select().from(users);
+  await dbHealthCheck();
   res.json({ message: 'OK ' });
 });