@ynode/cluster 1.1.0 → 1.2.0

package/README.md

@@ -62,6 +62,9 @@ The `run(startWorker, options)` function accepts the following options:
 | `scaleDownThreshold` | `number` | `10` | Event loop lag (ms) threshold to trigger scaling down. |
 | `scalingCooldown` | `number` | `10000` | Minimum time (ms) between scaling actions. |
 | `scaleDownGrace` | `number` | `30000` | Grace period (ms) after scaling up before scaling down is allowed. |
+| `autoScaleInterval` | `number` | `5000` | Interval (ms) for auto-scaling checks in "smart" mode. |
+| `shutdownSignals` | `string[]` | `['SIGINT', 'SIGTERM', 'SIGQUIT']` | Signals to listen for to trigger graceful shutdown. |
+| `shutdownTimeout` | `number` | `10000` | Time (ms) to wait for workers to shutdown before forced exit. |
 
 ## Working with @ynode/autoshutdown
 

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@ynode/cluster",
-    "version": "1.1.0",
+    "version": "1.2.0",
     "description": "Smart, auto-scaling Node.js cluster manager that monitors event loop lag to optimize performance and resource usage.",
     "main": "src/cluster.js",
     "exports": {

package/src/cluster.js

@@ -67,6 +67,9 @@ export function run(startWorker, options = true, log = console) {
         mode = "smart", // 'smart' or 'max'
         scalingCooldown = 10000,
         scaleDownGrace = 30000,
+        autoScaleInterval = 5000,
+        shutdownSignals = ["SIGINT", "SIGTERM", "SIGQUIT"],
+        shutdownTimeout = 10000,
     } = typeof options === "object" ? options : {};
 
     if (minWorkers > maxWorkers) {
@@ -208,34 +211,30 @@ export function run(startWorker, options = true, log = console) {
 
                 return;
             }
-        }, 5000).unref();
+            return;
+        }, autoScaleInterval).unref();
     }
 
     // Graceful shutdown handling for Master
-    const signals = ["SIGINT", "SIGTERM", "SIGQUIT"];
-
-    signals.forEach((signal) => {
-        process.on(signal, () => {
-            log.info(`Master received ${signal}, shutting down workers...`);
-            isShuttingDown = true;
-            for (const worker of Object.values(cluster.workers)) {
-                if (worker && worker.isConnected()) {
-                    worker.send("shutdown");
+    if (Array.isArray(shutdownSignals) && shutdownSignals.length > 0) {
+        shutdownSignals.forEach((signal) => {
+            process.on(signal, () => {
+                log.info(`Master received ${signal}, shutting down workers...`);
+                isShuttingDown = true;
+                for (const worker of Object.values(cluster.workers)) {
+                    if (worker && worker.isConnected()) {
+                        worker.send("shutdown");
+                    }
                 }
-            }
 
-            // Allow some time for workers to clean up? 
-            // Ideally we wait for them to exit, but for now we just let the process exit eventually
-            // or rely on the fact that existing "shutdown" message logic in worker handles close.
-            // We can just exit the master after a timeout if we want to force it, 
-            // but usually letting workers exit will cause master to exit if all handles are closed.
-            // For safety in this template, we'll force exit after a timeout.
-            setTimeout(() => {
-                log.info("Master force exiting after timeout.");
-                process.exit(0);
-            }, 10000).unref();
+                // Allow some time for workers to clean up
+                if (shutdownTimeout > 0) {
+                    setTimeout(() => {
+                        log.warn(`Master force exiting after ${shutdownTimeout}s timeout.`);
+                        process.exit(0);
+                    }, shutdownTimeout * 1000).unref();
+                }
+            });
         });
-    });
-}
-
-
+    }
+}