Implement timeout mechanisms for post-jobs-to-cluster-master features

johnteee · johnteee · commit 2bbbf5b7fa2a · 2018-06-28T21:48:34.000+08:00
diff --git a/README.md b/README.md
@@ -199,6 +199,21 @@ wsgilite.GET('/requestActionOnClusterMaster', async function (request, response,
     return e.errorMessage;
   });
 });
+// Post jobs to cluster master(Timeout cases)
+wsgilite.addClusterMasterRequestHandler(async (worker, msg, handle) => {
+  if (msg && msg.data && msg.data.action === 'streaming_timeout') {
+    var rp = require('request-promise-native');
+    await rp.get('https://www.sample-videos.com/video/mp4/240/big_buck_bunny_240p_30mb.mp4');
+    return 'ok';
+  }
+});
+wsgilite.GET('/requestActionOnClusterMasterTimeout', async function (request, response, meta) {
+  // Timeout for 3000 ms
+  return wsgilite.requestActionOnClusterMaster({action: 'streaming_timeout'}, 3000).catch((e)=>{
+    console.log(`I got error: ${e.errorStacktrace}`);
+    return e.errorStacktrace;
+  });
+});
 // if (require('cluster').isMaster) {
 //   // It could be called on cluster master
 //   wsgilite.requestActionOnClusterMaster({action: 'readrecord'}).catch((e)=>{
diff --git a/demo/simple-routing.js b/demo/simple-routing.js
@@ -154,6 +154,21 @@ wsgilite.GET('/requestActionOnClusterMaster', async function (request, response,
     return e.errorMessage;
   });
 });
+// Post jobs to cluster master(Timeout cases)
+wsgilite.addClusterMasterRequestHandler(async (worker, msg, handle) => {
+  if (msg && msg.data && msg.data.action === 'streaming_timeout') {
+    var rp = require('request-promise-native');
+    await rp.get('https://www.sample-videos.com/video/mp4/240/big_buck_bunny_240p_30mb.mp4');
+    return 'ok';
+  }
+});
+wsgilite.GET('/requestActionOnClusterMasterTimeout', async function (request, response, meta) {
+  // Timeout for 3000 ms
+  return wsgilite.requestActionOnClusterMaster({action: 'streaming_timeout'}, 3000).catch((e)=>{
+    console.log(`I got error: ${e.errorStacktrace}`);
+    return e.errorStacktrace;
+  });
+});
 // if (require('cluster').isMaster) {
 //   // It could be called on cluster master
 //   wsgilite.requestActionOnClusterMaster({action: 'readrecord'}).catch((e)=>{
diff --git a/wsgilite.js b/wsgilite.js
@@ -384,7 +384,7 @@ class WSGILite extends DefSubRoute {
   removeClusterMasterResponseHandler(clusterMasterResponseHandler) {
     this.clusterMasterResponseHandlers = this.clusterMasterResponseHandlers.filter((item)=>item !== clusterMasterResponseHandler);
   }
-  requestActionOnClusterMaster(data) {
+  requestActionOnClusterMaster(data, timeout) {
     let requestId = `${process.pid}_${Date.now()}_${Math.floor(Math.random()*Number.MAX_SAFE_INTEGER)}`;
     let msgRequest = {event: MSG_WSGILITE_DO_THINGS_MASTER, requestId, data};
 
@@ -393,6 +393,16 @@ class WSGILite extends DefSubRoute {
         let worker;
         let handle;
         let msgResponse = {event: MSG_WSGILITE_DO_THINGS_WORKER_SUCCESS, requestId};
+        if (timeout && timeout > 0) {
+          setTimeout(() => {
+            let e = new Error('Timeout');
+            msgResponse.error = e;
+            msgResponse.errorMessage = e.toString();
+            msgResponse.errorStacktrace = e.stack;
+            msgResponse.event = MSG_WSGILITE_DO_THINGS_WORKER_FAILURE;
+            reject(msgResponse);
+          }, timeout);
+        }
 
         this.handleClusterMasterRequest(worker, msgRequest, handle).catch((e)=>{
           msgResponse.error = e;
@@ -418,6 +428,17 @@ class WSGILite extends DefSubRoute {
       };
       this.addClusterMasterResponseHandler(handler);
       process.send(msgRequest);
+
+      if (timeout && timeout > 0) {
+        setTimeout(() => {
+          this.removeClusterMasterResponseHandler(handler);
+
+          let e = new Error('Timeout');
+          let msgResponse = {event: MSG_WSGILITE_DO_THINGS_WORKER_FAILURE, requestId, error: e, errorMessage: e.toString(), errorStacktrace: e.stack}
+          msgResponse.event = MSG_WSGILITE_DO_THINGS_WORKER_FAILURE;
+          reject(msgResponse);
+        }, timeout);
+      }
     });
   }
   handleClusterMasterRequest(worker, msg, handle) {
@@ -429,7 +450,7 @@ class WSGILite extends DefSubRoute {
     const errorHandler = (e) => {
       if (worker && (!errorHandled)) {
         console.log(e);
-        worker.send({event: MSG_WSGILITE_DO_THINGS_WORKER_FAILURE, requestId, error: e, errorMessage: e.toString(), errorStacktrace: e.stacktrace});
+        worker.send({event: MSG_WSGILITE_DO_THINGS_WORKER_FAILURE, requestId, error: e, errorMessage: e.toString(), errorStacktrace: e.stack});
       }
       errorHandled = true;
       return Promise.reject(e);
