CSE2410 Fall 2014 Bounce

// Copyright Joyent, Inc. and other Node contributors.

//

// Permission is hereby granted, free of charge, to any person obtaining a

// copy of this software and associated documentation files (the

// "Software"), to deal in the Software without restriction, including

// without limitation the rights to use, copy, modify, merge, publish,

// distribute, sublicense, and/or sell copies of the Software, and to permit

// persons to whom the Software is furnished to do so, subject to the

// following conditions:

//

// The above copyright notice and this permission notice shall be included

// in all copies or substantial portions of the Software.

//

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS

// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF

// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN

// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,

// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR

// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE

// USE OR OTHER DEALINGS IN THE SOFTWARE.

var spawn = require('child_process').spawn;

var cluster = require('cluster');

var http = require('http');

var options = {

  mode: 'master',

  host: '127.0.0.1',

  port: 22344,

  path: '/',

  servers: 1,

  clients: 1

};

for (var i = 2; i < process.argv.length; ++i) {

  var args = process.argv[i].split('=', 2);

  var key = args[0];

  var val = args[1];

  options[key] = val;

}

switch (options.mode) {

case 'master': startMaster(); break;

case 'server': startServer(); break;

case 'client': startClient(); break;

default: throw new Error('Bad mode: ' + options.mode);

}

process.title = 'http_bench[' + options.mode + ']';

// monkey-patch the log functions so they include name + pid

console.log = patch(console.log);

console.trace = patch(console.trace);

console.error = patch(console.error);

function patch(fun) {

  var prefix = process.title + '[' + process.pid + '] ';

  return function() {

    var args = Array.prototype.slice.call(arguments);

    args[0] = prefix + args[0];

    return fun.apply(console, args);

  };

}

function startMaster() {

  if (!cluster.isMaster) return startServer();

  for (var i = ~~options.servers; i > 0; --i) cluster.fork();

  for (var i = ~~options.clients; i > 0; --i) {

    var cp = spawn(process.execPath, [__filename, 'mode=client']);

    cp.stdout.pipe(process.stdout);

    cp.stderr.pipe(process.stderr);

  }

}

function startServer() {

  http.createServer(onRequest).listen(options.port, options.host);

  var body = Array(1024).join('x');

  var headers = {'Content-Length': '' + body.length};

  function onRequest(req, res) {

    req.on('error', onError);

    res.on('error', onError);

    res.writeHead(200, headers);

    res.end(body);

  }

  function onError(err) {

    console.error(err.stack);

  }

}

function startClient() {

  // send off a bunch of concurrent requests

  // TODO make configurable

  sendRequest();

  sendRequest();

  function sendRequest() {

    var req = http.request(options, onConnection);

    req.on('error', onError);

    req.end();

  }

  // add a little back-off to prevent EADDRNOTAVAIL errors, it's pretty easy

  // to exhaust the available port range

  function relaxedSendRequest() {

    setTimeout(sendRequest, 1);

  }

  function onConnection(res) {

    res.on('error', onError);

    res.on('data', onData);

    res.on('end', relaxedSendRequest);

  }

  function onError(err) {

    console.error(err.stack);

    relaxedSendRequest();

  }

  function onData(data) {

    // this space intentionally left blank

  }

}