Merge pull request #728 from PierreBrisorgueil/gracefulShutdown

feat(lib): add graceful shutdown, prevention for docker ✨

Author: PierreBrisorgueil

lib/app.js

@@ -2,8 +2,8 @@
  * Module dependencies.
  */
 const chalk = require('chalk');
-const http = require('http');
-const https = require('https');
+const nodeHttp = require('http');
+const nodeHttps = require('https');
 
 const config = require('../config');
 const mongooseService = require('./services/mongoose');
@@ -104,6 +104,7 @@ exports.start = async () => {
   let db;
   let orm;
   let app;
+  let http;
 
   try {
     ({ db, orm, app } = await bootstrap());
@@ -112,15 +113,36 @@ exports.start = async () => {
   }
 
   try {
-    if (config.secure && config.secure.credentials) await https.createServer(config.secure.credentials, app).listen(config.port, config.host);
-    else await http.createServer(app).listen(config.port, config.host);
+    if (config.secure && config.secure.credentials) http = await nodeHttps.createServer(config.secure.credentials, app).listen(config.port, config.host);
+    else http = await nodeHttp.createServer(app).listen(config.port, config.host);
     logConfiguration();
     return {
       db,
       orm,
       app,
+      http,
     };
   } catch (e) {
     throw new Error(e);
   }
 };
+
+// Shut down the server
+exports.shutdown = async (server) => {
+  try {
+    server.then(async (value) => {
+      await mongooseService.disconnect();
+      // add sequelize
+      value.http.close((err) => {
+        console.info(chalk.yellow('Server closed'));
+        if (err) {
+          console.info(chalk.red('Error on server close.', err));
+          process.exitCode = 1;
+        }
+        process.exit();
+      });
+    });
+  } catch (e) {
+    throw new Error(e);
+  }
+};

server.js

@@ -2,9 +2,20 @@
  * Module dependencies.
  */
 
+const chalk = require('chalk');
 const app = require('./lib/app');
 
-app.start().catch((e) => {
+const server = app.start().catch((e) => {
   console.log(`server failed: ${e.message}`);
   throw (e);
 });
+
+process.on('SIGINT', () => {
+  console.info(chalk.blue(' SIGINT Graceful shutdown ', new Date().toISOString()));
+  app.shutdown(server);
+});
+
+process.on('SIGTERM', () => {
+  console.info(chalk.blue(' SIGTERM Graceful shutdown ', new Date().toISOString()));
+  app.shutdown(server);
+});