TechEmpower.FrameworkBenchmarks/frameworks/JavaScript/koa/app.js

var cluster = require('cluster')
  , numCPUs = require('os').cpus().length
  , koa = require('koa')
  , route = require('koa-route')
  , handlebars = require('koa-handlebars')
  , bodyParser = require('koa-bodyparser')
  , override = require('koa-override')
  , mongo = require('koa-mongo');

if (cluster.isMaster) {
  // Fork workers.
  for (var i = 0; i < numCPUs; i++) {
    cluster.fork();
  }

  cluster.on('exit', function(worker, code, signal) {
    console.log('worker ' + worker.process.pid + ' died');
  });
} else {
  var app = module.exports = koa();
  app.use(bodyParser());
  app.use(override());
  app.use(mongo({
    uri: "mongodb://localhost/hello_world"
  }));
  app.use(handlebars({
    // needed, otherwise missing dir err
    partialsDir: "views"
  }));

  // routes
  app.use(route.get('/json', jsonHandler));
  app.use(route.get('/db', dbHandler));
  app.use(route.get('/queries', queriesHandler));
  app.use(route.get('/fortune', fortuneHandler));
  app.use(route.get('/update', updateHandler));
  app.use(route.get('/plaintext', textHandler));

  // Helper
  function getRandomNumber() {return Math.floor(Math.random()*10000) + 1;};

  // Query Helpers
  function *worldUpdateQuery() {
    var randomId = getRandomNumber();
    var randomNumber = getRandomNumber();
    var result = yield function(callback) {
        this.mongo.collection('world').update(
          {id: randomId},
          {randomNumber: randomNumber}, 
          callback
        );
    }
    return {
      id: randomId,
      randomNumber: randomNumber
    }
  }

  function *worldQuery() {
    return yield function(callback) {
        var randomId = {id: Math.floor(Math.random()*10000) + 1};
        this.mongo.collection('world').findOne(randomId, {_id: 0}, callback);
    }
  }

  function *fortunesQuery() {
    return yield function(callback) {
        this.mongo.collection('fortune').find({}, {_id: 0}).toArray(callback);
    }
  }

  // Route handlers

  function *jsonHandler() {
    this.body = {
      message: "Hello, world!"
    }
  }

  function *dbHandler() {
    this.body = yield worldQuery;
  }

  function *queriesHandler() {
    var numOfQueries = isNaN(this.query.queries) ? 1 : this.query.queries,
        queries = [];
    if (numOfQueries > 500) {
      numOfQueries = 500;
    } else if (numOfQueries < 1) {
      numOfQueries = 1;
    }
    for (var i = 0; i < numOfQueries; i++) {
      queries.push(worldQuery);
    }
    this.body = yield queries;
  }

  function *fortuneHandler() {
    var fortunes = yield fortunesQuery;
    fortunes.push({
      id: 0,
      message: 'Additional fortune added at request time.'
    });
    fortunes.sort(function(a, b) {
      return a.id - b.id;
    });
    yield this.render("fortunes", {
      fortunes: fortunes
    });
  }

  function *updateHandler() {
    var numOfUpdates = isNaN(this.query.queries) ? 1 : this.query.queries,
    queries = [];
    if (numOfUpdates > 500) {
      numOfUpdates = 500;
    } else if (numOfUpdates < 1) {
      numOfUpdates = 1;
    }
    for (var i = 0; i < numOfUpdates; i++) {
      queries.push(worldUpdateQuery);
    }
    this.body = yield queries;
  }

  function *textHandler() {
    this.body = 'Hello, world!'
  }

  app.listen(3000); //used for local testing
}