Merge pull request #4369 from anoophm/handle_SIGTERM

Issue #TG 267 fix: graceful shutdown of server 

src/app/package.json

@@ -5,18 +5,18 @@
   "buildNumber": "1",
   "private": true,
   "devDependencies": {
-    "codacy-coverage": "3.0.0",
     "babel": "6.23.0",
     "chai": "4.2.0",
-    "chai-spies": "1.0.0",
     "chai-http": "4.2.0",
+    "chai-spies": "1.0.0",
+    "codacy-coverage": "3.0.0",
     "gulp": "4.0.2",
-    "gulp-deploy-azure-cdn": "2.1.0",
+    "gulp-brotli": "1.2.1",
     "gulp-clean": "0.4.0",
     "gulp-decompress": "2.0.2",
+    "gulp-deploy-azure-cdn": "2.1.0",
     "gulp-download": "0.0.1",
     "gulp-gzip": "1.4.2",
-    "gulp-brotli": "1.2.1",
     "gulp-rename": "2.0.0",
     "gulp-rev": "8.1.1",
     "gulp-rev-collector": "1.3.1",
@@ -78,7 +78,7 @@
   },
   "dependencies": {
     "@project-sunbird/discussion-service": "0.0.6",
-    "@project-sunbird/ext-framework-server": "0.1.10",
+    "@project-sunbird/ext-framework-server": "0.1.27",
     "@project-sunbird/form-service": "0.0.12",
     "@project-sunbird/program": "0.0.5",
     "@project-sunbird/review-comment": "0.0.8",
@@ -98,6 +98,7 @@
     "express-session": "1.15.3",
     "googleapis": "45.0.0",
     "helmet": "3.21.2",
+    "http-graceful-shutdown": "^2.3.2",
     "http-status-codes": "1.3.0",
     "jsonwebtoken": "7.4.1",
     "kafka-node": "4.1.1",

src/app/server.js

 'use strict'
 const express = require('express');
+const gracefulShutdown = require('http-graceful-shutdown');
 const proxy = require('express-http-proxy')
 const session = require('express-session')
 const path = require('path')
@@ -88,8 +89,8 @@ app.get('/service/health', healthService.createAndValidateRequestBody, healthSer
 
 app.get("/latex/convert", latexService.convert);
 app.post("/latex/convert", bodyParser.json({ limit: '1mb' }), latexService.convert);
-app.post('/user/v2/accept/tnc', bodyParser.json({limit: '1mb'}), userService.acceptTnc);
-app.get('/user/v1/switch/:userId', bodyParser.json({limit: '1mb'}),keycloak.protect(), userService.switchUser);
+app.post('/user/v2/accept/tnc', bodyParser.json({ limit: '1mb' }), userService.acceptTnc);
+app.get('/user/v1/switch/:userId', bodyParser.json({ limit: '1mb' }), keycloak.protect(), userService.switchUser);
 
 require('./routes/desktopAppRoutes.js')(app) // desktop app routes
 
@@ -178,6 +179,7 @@ function runApp() {
       envHelper.defaultChannelId = _.get(channelData, 'result.response.content[0].hashTagId'); // needs to be added in envVariable file
       logger.info({ msg: `app running on port ${envHelper.PORTAL_PORT}` })
     })
+    handleShutDowns();
     portal.server.keepAliveTimeout = 60000 * 5;
   })
 }
@@ -216,4 +218,22 @@ process.on('uncaughtException', (err) => {
   console.log('Uncaught Exception', err)
   process.exit(1);
 });
+
+function handleShutDowns() {
+
+  const cleanup = signal => new Promise(async (resolve, reject) => { // close db connections
+    await frameworkAPI.closeCassandraConnections();
+    logger.info({ msg: `Closed db connection after ${signal} signal.` })
+    resolve();
+  });
+
+  gracefulShutdown(portal.server, {
+    signals: 'SIGINT SIGTERM',
+    timeout: 60 * 1000, // forcefully shutdown if not closed gracefully after 1 min
+    onShutdown: cleanup,
+    finally: () => logger.info({ msg: 'Server gracefully shut down.'}),
+    development: process.env.sunbird_environment === 'local' ? true : false, // in dev mode skip graceful shutdown 
+  });
+}
+
 exports.close = () => portal.server.close()