destroy sockets if reach wait threshold

DanielJDufour · Oct 26, 2021 · ca6cbd0 · ca6cbd0
1 parent 9914c37
commit ca6cbd0
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/bin/srvd.js b/bin/srvd.js
@@ -37,6 +37,7 @@ function serve(
 
   let last = Date.now();
   let server;
+  const sockets = [];
 
   let checkWaitTimeout, checkForCloseTimeout;
 
@@ -45,9 +46,20 @@ function serve(
     if (checkForCloseTimeout) clearTimeout(checkForCloseTimeout);
   }
 
+  function destroySockets() {
+    sockets.forEach(socket => {
+      try {
+        socket.destroy();
+      } catch (err) {
+        // pass
+      }
+    });
+  }
+
   function checkWait() {
     if (Date.now() - last > wait_ms) {
       if (debug) console.log(`[srvd] we haven't received a request in ${wait} seconds, so closing the server`);
+      destroySockets();
       server.close();
     }
   }
@@ -71,6 +83,10 @@ function serve(
     }
   });
 
+  server.on("connection", socket => {
+    sockets.push(socket);
+  });
+
   server.listen(port);
   if (debug) console.log("[srvd] serving on port " + port);
 

diff --git a/srvd.js b/srvd.js
@@ -35,6 +35,7 @@ function serve(
 
   let last = Date.now();
   let server;
+  const sockets = [];
 
   let checkWaitTimeout, checkForCloseTimeout;
 
@@ -43,9 +44,20 @@ function serve(
     if (checkForCloseTimeout) clearTimeout(checkForCloseTimeout);
   }
 
+  function destroySockets() {
+    sockets.forEach(socket => {
+      try {
+        socket.destroy();
+      } catch (err) {
+        // pass
+      }
+    });
+  }
+
   function checkWait() {
     if (Date.now() - last > wait_ms) {
       if (debug) console.log(`[srvd] we haven't received a request in ${wait} seconds, so closing the server`);
+      destroySockets();
       server.close();
     }
   }
@@ -69,6 +81,10 @@ function serve(
     }
   });
 
+  server.on("connection", socket => {
+    sockets.push(socket);
+  });
+
   server.listen(port);
   if (debug) console.log("[srvd] serving on port " + port);