From b4544f79bb8efeb8680fd3a2b28dc9173f551eb9 Mon Sep 17 00:00:00 2001
From: Jesse Crocker <jesse@gaiagps.com>
Date: Thu, 19 Dec 2019 08:31:30 -0700
Subject: [PATCH] Dont keep reforking forever

---
 bin/squirrel.js | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/bin/squirrel.js b/bin/squirrel.js
index 6066eca..80a48a2 100755
--- a/bin/squirrel.js
+++ b/bin/squirrel.js
@@ -64,16 +64,21 @@ function run(opts) {
     numCPUs = require("os").cpus().length;
 
   var numworkers = process.env.WORKER_COUNT || numCPUs;
+  var failureCount = 0;
   if (cluster.isMaster) {
     // Fork workers.
     for (var i = 0; i < numworkers; i++) {
       cluster.fork();
     }
 
-    cluster.on("exit", (worker, code, signal) => {
-      // Restart the worker
-      var worker = cluster.fork();
-      console.log(`worker ${worker.process.pid} died and has been replaced by ${worker.process.pid}`);
+    cluster.on("exit", (oldWorker, code, signal) => {
+      failureCount += 1;
+      if (failureCount < numworkers * 3) { // Dont restart forever, only allow 3 restarts per worker.
+        var worker = cluster.fork();
+        console.log(`worker ${oldWorker.process.pid} died and has been replaced by ${worker.process.pid}`);
+      } else {
+        console.log(`worker ${oldWorker.process.pid} died. Not starting a new worker because of too many failures.`);
+      }
     });
   } else {
     console.log("Starting worker");