cluster-service/lib/http-client.js

Turns your single process code into a fault-resilient multi-process service with built-in REST & CLI support

var util = require("util"),
  http = require("http"),
  querystring = require("querystring"),
  options = null,
  cservice = require("../cluster-service");

exports.init = function(o) {
  options = o;

  cservice.log([
      "Service already running. Attached CLI to master service.",
      "Enter 'help [enter]' for help."
    ]
    .join(' ')
    .info
  );

  if (!options || options.silentMode !== true) {
    process.stdin.resume();
    process.stdin.setEncoding('utf8');
    process.stdin.on("data", onCommand);
    process.stdout.write("cservice> ".cservice);
  }
};

exports.execute = onCommand;

function onCommand(question, cb) {
  var split, qs, url, body, err;
  question = question.replace(/[\r\n]/g, "");
  split = question.split(" ");
  if (split[0] === "exit") {
    cservice.log("Exiting CLI ONLY.".yellow);
    process.kill(process.pid, "SIGKILL"); // exit by force
    return;
  }
  qs = querystring.stringify({
    cmd: question,
    accessKey: options.accessKey
  });
  url = "http://"
    + (options.host || "localhost")
    + ":"
    + (options.port || 11987)
    + "/cli"
    + "?"
    + qs;
  cservice.log(
    "Running remote command: ".warn
    + url.replace(/accessKey=.*/i, "accessKey={ACCESS_KEY}").data
  );
  body = "";
  http.request(
    {
      host: options.host || "localhost",
      port: options.port || 11987,
      path: "/cli?" + qs,
      method: "POST"
    }
    , function(res) {
      res.setEncoding('utf8');
      res.on("data", function(chunk) {
        body += chunk;
      });
      res.on("end", function() {
        if (res.statusCode !== 200 && body) {
          err = body;
        }
        onCallback(err, body, res, cb);
      });
    }
  ).on("error", function(err) {
      body = err;
      onCallback(err, body, null, cb);
    }
  ).end();
}

function onCallback(err, result, res, cb) {
  if (err) {
    cservice.error("Error: ", err);
    result = {statusCode: res ? res.statusCode : "unknown", error: err};
  } else if (result) {
    if (
      typeof result === "string"
      && (result.indexOf("{") === 0 || result.indexOf("[") === 0)
    ) {
      result = JSON.parse(result); // deserialize
    }
  }
  cservice.log(util.inspect(result, {depth: null, colors: true}));

  if (!options || options.silentMode !== true) {
    //cservice.log("");//newline
    process.stdout.write("cservice> ".cservice);
  }

  if(cb){
    cb(err, result);
  }
}