feat: implement graceful shutdown

kata-ai · Apr 29, 2020 · cbb2e86 · cbb2e86
2 parents 594e8f5 + 9d1efb4
commit cbb2e86
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 35 deletions.
diff --git a/index.js b/index.js
@@ -49,7 +49,8 @@ module.exports = function (merapi) {
                 if (!isRoutesInMiddleware) app.use(yield router(injector, routes, routerOptions));
 
                 app.start = function () {
-                    app.listen(port, host);
+                    app.__listen = app.listen(port, host);
+                    app.__logger = logger;
                     logger.info(`Starting express on ${host}:${port}`);
                 };
                 return app;
@@ -61,6 +62,17 @@ module.exports = function (merapi) {
                 let app = yield merapi.resolve(this.apps[i]);
                 app.start();
             }
+        },
+
+        *onStop() {
+            for (let i = 0; i < this.apps.length; i++) {
+                let app = yield merapi.resolve(this.apps[i]);
+                if (app.__listen) {
+                    app.__listen.close(() => {
+                        app.__logger.info("Shutting down express plugin...");
+                    })
+                }
+            }
         }
     };
 };
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -27,7 +27,7 @@
   "dependencies": {
     "body-parser": "^1.15.2",
     "express": "^4.14.0",
-    "merapi": "^0.17.1",
+    "merapi": "^0.21.0",
     "type-check": "^0.3.2"
   },
   "devDependencies": {