add health check status for services

src/db/connection.ts

@@ -46,6 +46,21 @@ export function disconnect() {
   return mongoose.connection.close();
 }
 
+/**
+ * Health check status
+ */
+export const mongoStatus = () => {
+  return new Promise((resolve, reject) => {
+    mongoose.connection.db.admin().ping((err, result) => {
+      if (err) {
+        return reject(err);
+      }
+
+      return resolve(result);
+    });
+  });
+};
+
 const schema = makeExecutableSchema({
   typeDefs,
   resolvers,

src/index.ts

@@ -21,14 +21,14 @@ import {
   readFileRequest,
   registerOnboardHistory,
 } from './data/utils';
-import { connect } from './db/connection';
+import { connect, mongoStatus } from './db/connection';
 import { debugBase, debugExternalApi, debugInit } from './debuggers';
 import { identifyCustomer, trackCustomEvent, trackViewPageEvent, updateCustomerProperty } from './events';
-import { initConsumer } from './messageBroker';
+import { initConsumer, rabbitMQStatus } from './messageBroker';
 import { importer, uploader } from './middlewares/fileMiddleware';
 import userMiddleware from './middlewares/userMiddleware';
 import widgetsMiddleware from './middlewares/widgetsMiddleware';
-import { initRedis } from './redisClient';
+import { initRedis, redisStatus } from './redisClient';
 import init from './startup';
 
 // load environment variables
@@ -126,7 +126,28 @@ app.get('/download-template', async (req: any, res) => {
 });
 
 // for health check
-app.get('/status', async (_req, res) => {
+app.get('/status', async (_req, res, next) => {
+  try {
+    await mongoStatus();
+  } catch (e) {
+    debugBase('MongoDB is not running');
+    next(e);
+  }
+
+  try {
+    await redisStatus();
+  } catch (e) {
+    debugBase('Redis is not running');
+    next(e);
+  }
+
+  try {
+    await rabbitMQStatus();
+  } catch (e) {
+    debugBase('RabbitMQ is not running');
+    next(e);
+  }
+
   res.end('ok');
 });
 

src/messageBroker.ts

@@ -1,4 +1,4 @@
-import * as amqplib from 'amqplib';
+import amqplib from 'amqplib';
 import * as dotenv from 'dotenv';
 import * as uuid from 'uuid';
 import {
@@ -162,3 +162,23 @@ export const initConsumer = async () => {
     }
   });
 };
+
+/**
+ * Health check rabbitMQ
+ */
+export const rabbitMQStatus = async () => {
+  return new Promise((resolve, reject) => {
+    // tslint:disable-next-line:no-submodule-imports
+    import('amqplib/callback_api')
+      .then(amqp => {
+        amqp.connect(RABBITMQ_HOST, error => {
+          if (error) {
+            return reject(error);
+          }
+
+          return resolve('ok');
+        });
+      })
+      .catch(e => reject(e));
+  });
+};

src/redisClient.ts

@@ -129,3 +129,19 @@ export const removeFromArray = (setKey: string, setMember: string) => {
     });
   });
 };
+
+/**
+ * Health check status
+ * retryStrategy - get response immediately
+ */
+export const redisStatus = () => {
+  return new Promise((resolve, reject) => {
+    client.ping((error, result) => {
+      if (error) {
+        return reject(error);
+      }
+
+      return resolve(result);
+    });
+  });
+};