codehooks-js 1.3.20 → 1.3.21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codehooks-js",
-  "version": "1.3.20",
+  "version": "1.3.21",
   "type": "module",
   "description": "Codehooks.io official library - provides express.JS like syntax",
   "main": "index.js",

package/workflow/engine.mjs

@@ -23,7 +23,8 @@ class Workflow extends EventEmitter {
     #description = null; // workflow description
     #defaultStepOptions = { // default step options
         timeout: 30000, // timeout for a step
-        maxRetries: 3 // maximum number of retries for a step
+        maxRetries: 3, // maximum number of retries for a step
+        workers: 1 // number of workers for a step
     };
 
     /**
@@ -154,6 +155,36 @@ class Workflow extends EventEmitter {
         return this.#steps;
     }
 
+    /**
+     * Get the number of workers for a specific step
+     * @param {string} stepName - Name of the step
+     * @returns {number} The number of workers for the step
+     */
+    getWorkersForStep(stepName) {
+        const stepConfig = this.#steps[stepName];
+        return stepConfig?.workers ?? this.#defaultStepOptions.workers ?? 1;
+    }
+
+    /**
+     * Set the default number of workers for steps
+     * @param {number} workers - Number of workers
+     * @throws {Error} If workers is not a positive number
+     */
+    setDefaultWorkers(workers) {
+        if (typeof workers !== 'number' || workers <= 0) {
+            throw new Error('Workers must be a positive number');
+        }
+        this.#defaultStepOptions.workers = workers;
+    }
+
+    /**
+     * Get the default number of workers
+     * @returns {number} The default number of workers
+     */
+    getDefaultWorkers() {
+        return this.#defaultStepOptions.workers;
+    }
+
     /**
      * Configure the steps engine
      * @param {Object} config - Configuration object
@@ -161,6 +192,7 @@ class Workflow extends EventEmitter {
      * @param {string} config.queuePrefix - Queue prefix
      * @param {number} config.timeout - Timeout in milliseconds
      * @param {number} config.maxStepCount - Maximum step count
+     * @param {number} config.workers - Default number of workers for steps
      * @param {Object} config.steps - Steps configuration
      */
     configure(config) {
@@ -176,6 +208,9 @@ class Workflow extends EventEmitter {
         if (config.maxStepCount) {
             this.setMaxStepCount(config.maxStepCount);
         }
+        if (config.workers) {
+            this.setDefaultWorkers(config.workers);
+        }
         if (config.steps) {
             this.setStepsConfig(config.steps);
         }
@@ -370,7 +405,7 @@ class Workflow extends EventEmitter {
                                         state: mergedState,
                                         options: options,
                                         instanceId: instanceId
-                                    });
+                                    }, { workers: this.getWorkersForStep(step) });
                                 }
                             } else {
                                 console.debug('enqueue step', nextStep, instanceId);
@@ -380,7 +415,7 @@ class Workflow extends EventEmitter {
                                     state: mergedState,
                                     options: options,
                                     instanceId: instanceId
-                                });
+                                }, { workers: this.getWorkersForStep(nextStep) });
                             }
                             this.emit('stepEnqueued', { workflowName: stepsName, step: nextStep, state: newState, instanceId });
                             resolve();
@@ -508,7 +543,7 @@ class Workflow extends EventEmitter {
                     state: newState,
                     instanceId: newState._id,
                     options: {}
-                });
+                }, { workers: this.getWorkersForStep(firstStepName) });
                 resolve(newState);
             } catch (error) {
                 console.error('Error starting workflow:', error.message);
@@ -589,7 +624,7 @@ class Workflow extends EventEmitter {
                 state: state,
                 options: {},
                 instanceId: instanceId
-            });
+            }, { workers: this.getWorkersForStep(state.nextStep) });
 
             resolve({ instanceId });
         });