node-query/node_modules/mysql2/benchmarks/FB/hello.js

293 lines
7.8 KiB
JavaScript

var cluster = require('cluster');
var numCPUs = require('os').cpus().length;
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.pid + ' died');
});
return;
}
var http = require('http');
var url = require('url');
var libmysql = require('mysql-libmysqlclient').createConnectionSync();
var mysql2 = require('../..');
var mysql = require('mysql');
var mariasql = require('mariasql');
var async = require('async');
var Mapper = require('mapper');
var jade = require('jade');
var fs = require('fs');
var connMap = { user: 'root', password: '', database: 'hello_world', host: 'localhost' };
Mapper.connect(connMap, {verbose: false, strict: false});
var World = Mapper.map("World", "id", "randomNumber");
var template = jade.compile(fs.readFileSync('./fortunes.jade'));
libmysql.connectSync('localhost', 'root', '', 'hello_world');
pool2 = mysql2.createPool(connMap);
pool1 = mysql.createPool(connMap);
mysql2conn = mysql2.createConnection(connMap);
mysql1conn = mysql.createConnection(connMap);
mariaconn = new mariasql();
mariaconn.connect({
host: connMap.host,
user: connMap.user,
password: connMap.password,
db: connMap.database
});
function getRandomNumber() {
return Math.floor(Math.random() * 10000) + 1;
}
function sequelizeQuery(callback) {
World.findById(getRandomNumber(), function (err, world) {
callback(null, world);
});
}
function handlePrepared(req, res) {
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
var results = [];
for (var i=0; i < queries; ++i) {
mysql2conn.execute("SELECT * FROM world WHERE id = ?", [getRandomNumber()], function (err, rows) {
results.push(rows[0]);
if (results.length == queries)
res.end(JSON.stringify(results));
});
}
}
function handleMysqlIsh(conn, req, res) {
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
//pool.getConnection(function(err, conn) {
//console.log(conn, conn.query, '===============');
var results = [];
for (var i=0; i < queries; ++i) {
mysql2conn.query("SELECT * FROM world WHERE id = ?", [getRandomNumber()], function (err, rows) {
results.push(rows[0]);
if (results.length == queries)
res.end(JSON.stringify(results));
});
}
//});
}
function handleMysqlIshPool(pool, req, res) {
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
var results = [];
for (var i=0; i < queries; ++i) {
pool.getConnection(function(err, conn) {
mysql2conn.query("SELECT * FROM world WHERE id = " + getRandomNumber(), function (err, rows) {
results.push(rows[0]);
if (results.length == queries)
res.end(JSON.stringify(results));
});
});
}
}
function handleMaria(req, res)
{
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
var results = [];
for (var i=0; i < queries; ++i) {
mariaconn.query("SELECT * FROM world WHERE id = :id", { id: getRandomNumber() } )
.on('result', function (dbres) {
dbres.on('row', function(row) {
results.push(row);
if (results.length == queries)
res.end(JSON.stringify(results));
});
});
}
}
function fortuneMysql(conn, res) {
var fortunes = [];
res.writeHead(200, {'Content-Type': 'text/html'});
conn.query('select * from Fortune', function(err, fortunes) {
fortunes.push({id: 0, message: "Additional fortune added at request time."});
fortunes.sort(sortFortunes);
res.end(template({fortunes: fortunes}));
});
}
function fortuneMaria(res) {
fortunes = [];
res.writeHead(200, {'Content-Type': 'text/html'});
mariaconn.query("SELECT * from Fortune")
.on('result', function (dbres) {
dbres.on('row', function(row) { fortunes.push(row); });
dbres.on('end', function() {
fortunes.push({id: 0, message: "Additional fortune added at request time."});
fortunes.sort(sortFortunes);
res.end(template({fortunes: fortunes}));
});
});
}
function sortFortunes(a, b) {
return (a.message < b.message) ? -1 : (a.message > b.message) ? 1 : 0;
}
http.createServer(function (req, res) {
// JSON response object
var hello = {message: "Hello, world"};
var path = url.parse(req.url).pathname;
switch (path) {
case '/json':
// JSON Response Test
res.writeHead(200, {'Content-Type': 'application/json; charset=UTF-8'});
// Write JSON object to response
res.end(JSON.stringify(hello));
break;
case '/mysql-orm':
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
var queryFunctions = new Array(queries);
for (var i = 0; i < queries; i += 1) {
queryFunctions[i] = sequelizeQuery;
}
res.writeHead(200, {'Content-Type': 'application/json'});
async.parallel(queryFunctions, function(err, results) {
res.end(JSON.stringify(results));
});
break;
case '/mysql':
res.writeHead(200, {'Content-Type': 'application/json'});
function libmysqlQuery(callback) {
libmysql.query("SELECT * FROM world WHERE id = " + getRandomNumber(), function (err, res) {
if (err) {
throw err;
}
res.fetchAll(function(err, rows) {
if (err) {
throw err;
}
res.freeSync();
callback(null, rows[0]);
});
});
}
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
var queryFunctions = new Array(queries);
for (var i = 0; i < queries; i += 1) {
queryFunctions[i] = libmysqlQuery;
}
async.parallel(queryFunctions, function(err, results) {
if (err) {
res.writeHead(500);
return res.end('MYSQL CONNECTION ERROR.');
}
res.end(JSON.stringify(results));
});
break;
case '/mysql2':
handleMysqlIsh(mysql2conn, req, res);
break;
case '/fortunes-mysql2':
fortuneMysql(mysql2conn, res);
break;
case '/fortunes-mysql1':
fortuneMysql(mysql1conn, res);
break;
case '/fortunes-maria':
fortuneMaria(res);
break;
case '/mysql2pool':
handleMysqlIshPool(pool2, req, res);
break;
case '/mysql2ps':
handlePrepared(req, res);
break;
case '/mysql1':
handleMysqlIsh(mysql1conn, req, res);
break;
case '/maria':
handleMaria(req, res);
break;
case '/update':
res.writeHead(200, {'Content-Type': 'application/json'});
function libmysqlQuery(callback) {
libmysql.query("SELECT * FROM world WHERE id = " + getRandomNumber(), function (err, res) {
if (err) {
throw err;
}
res.fetchAll(function(err, rows) {
if (err) {
throw err;
}
res.freeSync();
rows[0].randomNumber = getRandomNumber();
libmysql.query("UPDATE World SET randomNumber = " + rows[0].randomNumber + " WHERE id = " + rows[0]['id'], function (err, res) {
if (err) {
throw err;
}
});
callback(null, rows[0]);
});
});
}
var values = url.parse(req.url, true);
var queries = values.query.queries || 1;
var queryFunctions = new Array(queries);
for (var i = 0; i < queries; i += 1) {
queryFunctions[i] = libmysqlQuery;
}
async.parallel(queryFunctions, function(err, results) {
if (err) {
res.writeHead(500);
return res.end('MYSQL CONNECTION ERROR.');
}
res.end(JSON.stringify(results));
});
break;
default:
// File not found handler
res.writeHead(404, {'Content-Type': 'text/html; charset=UTF-8'});
res.end("NOT IMPLEMENTED");
}
}).listen(8080);