@io-orkes/conductor-javascript 1.2.0-rc.5 → 1.2.0-rc.7

package/dist/index.mjs

@@ -2210,9 +2210,9 @@ var Poller = class {
     /**
      * Stops Polling for work
      */
-    this.stopPolling = () => {
+    this.stopPolling = async () => {
+      await Promise.all(this.concurrentCalls.map((call) => call.stop()));
       this.polling = false;
-      this.concurrentCalls.forEach((call) => call.stop());
     };
     this.poll = async () => {
       if (!this.polling) {
@@ -2229,17 +2229,18 @@ var Poller = class {
         while (poll) {
           await this.pollFunction();
           await new Promise(
-            (r) => timeout = setTimeout(() => r(true), this.options.pollInterval)
+            (r) => poll ? timeout = setTimeout(() => r(true), this.options.pollInterval) : r(true)
           );
         }
       };
       return {
         promise: pollingCall(),
-        stop: () => {
+        stop: () => new Promise((r) => {
           clearTimeout(timeout);
           poll = false;
           this.logger.debug("stopping single poll call");
-        }
+          r(true);
+        })
       };
     };
     this.pollFunction = pollFunction;
@@ -2302,8 +2303,8 @@ var TaskRunner = class {
     /**
      * Stops Polling for work
      */
-    this.stopPolling = () => {
-      this.poller.stopPolling();
+    this.stopPolling = async () => {
+      await this.poller.stopPolling();
     };
     this.pollAndExecute = async () => {
       try {
@@ -2469,9 +2470,11 @@ var TaskManager = class {
     /**
      * Stops polling for tasks
      */
-    this.stopPolling = () => {
+    this.stopPolling = async () => {
       for (const taskType in this.tasks) {
-        this.tasks[taskType].forEach((runner) => runner.stopPolling());
+        await Promise.all(
+          this.tasks[taskType].map((runner) => runner.stopPolling())
+        );
         this.tasks[taskType] = [];
       }
       this.polling = false;