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

const cluster = require('cluster'),
  numCPUs = require('os').cpus().length;

// Koa Deps
const koa = require('koa'),
  route = require('koa-route'),
  handlebars = require('koa-handlebars'),
  bodyParser = require('koa-bodyparser'),
  override = require('koa-override');

// Monk MongoDB Driver Deps
const monk = require('monk'),
  wrap = require('co-monk'),
  db = monk('mongodb://TFB-database/hello_world'),
  worlds = wrap(db.get('world')),
  fortunes = wrap(db.get('fortune'));

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

  cluster.on('exit', (worker, code, signal) =>
    console.log('worker ' + worker.process.pid + ' died'));
} else {
  const app = module.exports = koa();
  app.use(bodyParser());
  app.use(override());
  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('/fortunes', fortuneHandler));
  app.use(route.get('/updates', updateHandler));
  app.use(route.get('/plaintext', textHandler));

  // Helpers
  const getRandomNumber = () => ~~(Math.random()*10000) + 1;

  const validateParam = (param) => {
    let numOfQueries = isNaN(param) ? 1 : param;
    if (numOfQueries > 500) {
      numOfQueries = 500;
    } else if (numOfQueries < 1) {
      numOfQueries = 1;
    }
    return numOfQueries;
  };

  // Query Helpers
  function *worldUpdateQuery() {
    const randomId = getRandomNumber();
    const randomNumber = getRandomNumber();
    const result = yield worlds.update(
      {id: randomId}, 
      {$set: {randomNumber: randomNumber}}
    );
    return {
      id: randomId,
      randomNumber: randomNumber
    }
  }

  function *worldQuery() {
    return yield worlds.findOne({id: getRandomNumber()}, '-_id');
  }

  function *fortunesQuery() {
    return yield fortunes.find({}, '-_id');
  }

  // Route handlers
  function *jsonHandler() {
    this.set('Server', 'Koa');
    this.body = {
      message: "Hello, world!"
    }
  }

  function *dbHandler() {
    this.set('Server', 'Koa');
    this.body = yield worldQuery;
  }

  function *queriesHandler() {
    this.set('Server', 'Koa');
    let numOfQueries = validateParam(this.query.queries);
    const queries = [];
    for (let i = 0; i < numOfQueries; i++) {
      queries.push(worldQuery);
    }
    this.body = yield queries;
  }

  function *fortuneHandler() {
    this.set('Server', 'Koa');
    const fortunes = yield fortunesQuery;
    fortunes.push({
      id: 0,
      message: 'Additional fortune added at request time.'
    });
    fortunes.sort((a, b) => (a.message < b.message) ? -1 : 1);
    yield this.render("fortunes", {fortunes: fortunes});
  }

  function *updateHandler() {
    this.set('Server', 'Koa');
    const numOfUpdates = validateParam(this.query.queries);
    const queries = [];
    for (let i = 0; i < numOfUpdates; i++) {
      queries.push(worldUpdateQuery);
    }
    this.body = yield queries;
  }

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

  app.listen(8080);
}