@io-orkes/conductor-javascript 1.0.1 → 1.1.0

package/dist/index.mjs

@@ -18,11 +18,7 @@ var __spreadValues = (a, b) => {
 };
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 
-// src/task/TaskRunner.ts
-var DEFAULT_ERROR_MESSAGE = "An unknown error occurred";
-var MAX_RETRIES = 3;
-var noopErrorHandler = (__error) => {
-};
+// src/task/helpers.ts
 var noopLogger = {
   debug: (...args) => {
   },
@@ -31,6 +27,88 @@ var noopLogger = {
   error: (...args) => {
   }
 };
+
+// src/task/Poller.ts
+var Poller = class {
+  constructor(pollFunction, pollerOptions, logger) {
+    this.concurrentCalls = [];
+    this.pollFunction = async () => {
+    };
+    this.isPolling = false;
+    this.options = {
+      pollInterval: 1e3,
+      concurrency: 1
+    };
+    this.logger = noopLogger;
+    this.startPolling = () => {
+      if (this.isPolling) {
+        throw new Error("Runner is already started");
+      }
+      return this.poll();
+    };
+    this.stopPolling = () => {
+      this.isPolling = false;
+      this.concurrentCalls.forEach((call) => call.stop());
+    };
+    this.poll = async () => {
+      if (!this.isPolling) {
+        this.isPolling = true;
+        for (let i = 0; i < this.options.concurrency; i++) {
+          this.concurrentCalls.push(this.singlePoll());
+        }
+      }
+    };
+    this.singlePoll = () => {
+      let poll = this.isPolling;
+      let timeout;
+      const pollingCall = async () => {
+        while (poll) {
+          await this.pollFunction();
+          await new Promise((r) => timeout = setTimeout(() => r(true), this.options.pollInterval));
+        }
+      };
+      return {
+        promise: pollingCall(),
+        stop: () => {
+          clearTimeout(timeout);
+          poll = false;
+          this.logger.debug("stopping single poll call");
+        }
+      };
+    };
+    this.pollFunction = pollFunction;
+    this.options = __spreadValues(__spreadValues({}, this.options), pollerOptions);
+    this.logger = logger || noopLogger;
+  }
+  updateConcurrency(concurrency) {
+    if (concurrency > 0 && concurrency !== this.options.concurrency) {
+      if (concurrency < this.options.concurrency) {
+        const result = this.concurrentCalls.splice(0, this.options.concurrency - concurrency);
+        result.forEach((call) => {
+          call.stop();
+          this.logger.debug("stopping some spawned calls");
+        });
+      } else {
+        for (let i = 0; i < concurrency - this.options.concurrency; i++) {
+          this.concurrentCalls.push(this.singlePoll());
+          this.logger.debug("spawning additional poll calls");
+        }
+      }
+      this.options.concurrency = concurrency;
+    }
+  }
+  updateOptions(options) {
+    const newOptions = __spreadValues(__spreadValues({}, this.options), options);
+    this.updateConcurrency(newOptions.concurrency);
+    this.options = newOptions;
+  }
+};
+
+// src/task/TaskRunner.ts
+var DEFAULT_ERROR_MESSAGE = "An unknown error occurred";
+var MAX_RETRIES = 3;
+var noopErrorHandler = (__error) => {
+};
 var TaskRunner = class {
   constructor({
     worker,
@@ -39,32 +117,24 @@ var TaskRunner = class {
     logger = noopLogger,
     onError: errorHandler = noopErrorHandler
   }) {
-    this.isPolling = false;
     this.startPolling = () => {
-      if (this.isPolling) {
-        throw new Error("Runner is already started");
-      }
-      this.isPolling = true;
-      return this.poll();
+      this.poller.startPolling();
     };
     this.stopPolling = () => {
-      this.isPolling = false;
+      this.poller.stopPolling();
     };
-    this.poll = async () => {
-      while (this.isPolling) {
-        try {
-          const { workerID } = this.options;
-          const task = await this.taskResource.poll(this.worker.taskDefName, workerID, this.options.domain);
-          if (task && task.taskId) {
-            await this.executeTask(task);
-          } else {
-            this.logger.debug(`No tasks for ${this.worker.taskDefName}`);
-          }
-        } catch (unknownError) {
-          this.handleUnknownError(unknownError);
-          this.errorHandler(unknownError);
+    this.pollAndExecute = async () => {
+      try {
+        const { workerID } = this.options;
+        const task = await this.taskResource.poll(this.worker.taskDefName, workerID, this.worker.domain ?? this.options.domain);
+        if (task && task.taskId) {
+          await this.executeTask(task);
+        } else {
+          this.logger.debug(`No tasks for ${this.worker.taskDefName}`);
         }
-        await new Promise((r) => setTimeout(() => r(true), this.options.pollInterval));
+      } catch (unknownError) {
+        this.handleUnknownError(unknownError);
+        this.errorHandler(unknownError);
       }
     };
     this.updateTaskWithRetry = async (task, taskResult) => {
@@ -118,6 +188,18 @@ var TaskRunner = class {
     this.worker = worker;
     this.options = options;
     this.errorHandler = errorHandler;
+    this.poller = new Poller(this.pollAndExecute, { concurrency: options.concurrency, pollInterval: options.pollInterval }, this.logger);
+  }
+  updateOptions(options) {
+    const newOptions = __spreadValues(__spreadValues({}, this.options), options);
+    this.poller.updateOptions({
+      concurrency: newOptions.concurrency,
+      pollInterval: newOptions.pollInterval
+    });
+    this.options = newOptions;
+  }
+  get getOptions() {
+    return this.options;
   }
 };
 
@@ -2087,25 +2169,35 @@ function workerId(options) {
 var TaskManager = class {
   constructor(client, workers, config = {}) {
     this.tasks = {};
+    this.workerManagerWorkerOptions = (worker) => {
+      return __spreadProps(__spreadValues({}, this.taskManageOptions), {
+        concurrency: worker.concurrency ?? this.taskManageOptions.concurrency,
+        domain: worker.domain ?? this.taskManageOptions.domain
+      });
+    };
+    this.updatePollingOptions = (options) => {
+      this.workers.forEach((worker) => {
+        const newOptions = __spreadValues(__spreadValues({}, this.workerManagerWorkerOptions(worker)), options);
+        const runners = this.tasks[worker.taskDefName];
+        runners.forEach((runner) => {
+          runner.updateOptions(newOptions);
+        });
+      });
+    };
     this.startPolling = () => {
       this.workers.forEach((worker) => {
         this.tasks[worker.taskDefName] = [];
-        const options = __spreadProps(__spreadValues({}, this.taskManageOptions), {
-          concurrency: worker.concurrency ?? this.taskManageOptions.concurrency,
-          domain: worker.domain ?? this.taskManageOptions.domain
-        });
+        const options = this.workerManagerWorkerOptions(worker);
         this.logger.debug(`Starting taskDefName=${worker.taskDefName} concurrency=${options.concurrency} domain=${options.domain}`);
-        for (let i = 0; i < options.concurrency; i++) {
-          const runner = new TaskRunner({
-            worker,
-            options,
-            taskResource: this.client.taskResource,
-            logger: this.logger,
-            onError: this.errorHandler
-          });
-          runner.startPolling();
-          this.tasks[worker.taskDefName].push(runner);
-        }
+        const runner = new TaskRunner({
+          worker,
+          options,
+          taskResource: this.client.taskResource,
+          logger: this.logger,
+          onError: this.errorHandler
+        });
+        runner.startPolling();
+        this.tasks[worker.taskDefName].push(runner);
       });
     };
     this.stopPolling = () => {