feat(rpc/service): shutdown process on disconnect in service adapter

CheerlessCloud · Oct 4, 2018 · 4097905 · 4097905
1 parent f0d46a1
commit 4097905
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/Service.js b/src/Service.js
@@ -154,7 +154,13 @@ class RpcService extends AdapterConsumer {
     stopSignal = 'SIGINT',
     gracefulStopTimeout = 60 * 1000,
   }: { stopSignal?: string, gracefulStopTimeout?: number } = {}) {
+    let shutdownInProcess = false;
     const endCallback = async () => {
+      if (shutdownInProcess) {
+        return;
+      }
+      shutdownInProcess = true;
+
       try {
         await this.destroy({ gracefulStopTimeout });
         process.exit(0);
@@ -170,6 +176,10 @@ class RpcService extends AdapterConsumer {
     };
 
     process.once(stopSignal, endCallback);
+
+    const adapter = this._getAdapter();
+    // @todo exit with status 0
+    adapter._eventBus.once('disconnect', endCallback);
   }
 }