Implement a control socket

Ref #1643
pump-io · Jun 19, 2018 · e9951de · e9951de
1 parent 3c2f8c5
commit e9951de
Show file tree

Hide file tree

Showing 4 changed files with 111 additions and 1 deletion.
diff --git a/bin/pump b/bin/pump
@@ -27,9 +27,11 @@ var cluster = require("cluster"),
     os = require("os"),
     fs = require("fs"),
     path = require("path"),
+    net = require("net"),
     Logger = require("bunyan"),
     _ = require("lodash"),
     importFresh = require("import-fresh"),
+    split = require("split"),
     configUtil = require("../lib/config"),
     restartEpoch = require("../lib/zerodowntime"),
     restartInflight = false,
@@ -143,6 +145,25 @@ var launchApps = function(configBase) {
             cluster.fork();
         });
 
+        var ctl = net.createServer(function(con) {
+            con.pipe(split()).on("data", function(line) {
+                switch (line.toLowerCase()) {
+                case "restart":
+                    triggerRestart(clusterLog, config);
+                    con.write("ACK\n");
+                    clusterLog.info("Received command over control socket.", line);
+                    break;
+                default:
+                    clusterLog.info("Received unknown command over control socket; reject.", line);
+                    con.write("NACK\n");
+                }
+            });
+        });
+
+        ctl.listen(config.controlSocket, function() {
+            clusterLog.info("Listening for control commands on " + config.controlSocket + ".");
+        });
+
         // Perform zero-downtime restart on SIGUSR2
         // TODO handle config changes
         process.on("SIGUSR2", triggerRestart.bind(null, clusterLog, config));

diff --git a/lib/defaults.js b/lib/defaults.js
@@ -81,5 +81,6 @@ module.exports = {driver: "memory",
                   cleanupSession: 1200000,
                   cleanupNonce: 1200000,
                   favicon: path.resolve(__dirname, "../public/images/favicon.ico"),
-                  users: []
+                  users: [],
+                  controlSocket: "/run/pump.socket"
                  };
diff --git a/npm-shrinkwrap.json b/npm-shrinkwrap.json
diff --git a/package.json b/package.json
@@ -56,6 +56,7 @@
     "serve-favicon": "^2.3.0",
     "showdown": "^1.4.1",
     "sockjs": "0.3.x",
+    "split": "^1.0.1",
     "ssl-config": "^1.0.0",
     "step": "^1.0.0",
     "ua-parser-js": "^0.7.10",