From e19bdbf6d7ee7ecaee1ae5bdf8a3f6c9ebec1c54 Mon Sep 17 00:00:00 2001
From: Jan Oberhauser <jan.oberhauser@gmail.com>
Date: Thu, 19 Sep 2019 14:00:14 +0200
Subject: [PATCH] :rocket: Include DB check in health-check

---
 packages/cli/src/Server.ts | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/packages/cli/src/Server.ts b/packages/cli/src/Server.ts
index e38573fec0..2d42801033 100644
--- a/packages/cli/src/Server.ts
+++ b/packages/cli/src/Server.ts
@@ -3,6 +3,9 @@ import {
 	dirname as pathDirname,
 	join as pathJoin,
 } from 'path';
+import {
+	getConnectionManager,
+} from "typeorm";
 import * as bodyParser from 'body-parser';
 import * as history from 'connect-history-api-fallback';
 import * as requestPromise from 'request-promise-native';
@@ -226,12 +229,29 @@ class App {
 		// ----------------------------------------
 
 
-		// Creates a new workflow
-		this.app.get('/healthz', ResponseHelper.send(async (req: express.Request, res: express.Response): Promise<IDataObject> => {
-			return {
+		// Does very basic health check
+		this.app.get('/healthz', async (req: express.Request, res: express.Response) => {
+
+			const connectionManager = getConnectionManager();
+
+			if (connectionManager.connections.length === 0) {
+				const error = new ResponseHelper.ResponseError('No Database connection found!', undefined, 503);
+				return ResponseHelper.sendErrorResponse(res, error);
+			}
+
+			if (connectionManager.connections[0].isConnected === false) {
+				// Connection is not active
+				const error = new ResponseHelper.ResponseError('Database connection not active!', undefined, 503);
+				return ResponseHelper.sendErrorResponse(res, error);
+			}
+
+			// Everything fine
+			const responseData = {
 				status: 'ok',
 			};
-		}));
+
+			ResponseHelper.sendSuccessResponse(res, responseData, true, 200);
+		});