hello.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461
  1. var cluster = require('cluster')
  2. , numCPUs = require('os').cpus().length
  3. , http = require('http')
  4. , url = require('url')
  5. , Sequelize = require('sequelize')
  6. , mysql = require('mysql')
  7. , async = require('async')
  8. , Mongoose = require('mongoose')
  9. , conn = Mongoose.connect('mongodb://127.0.0.1/hello_world')
  10. , MongoClient = require('mongodb').MongoClient
  11. , Handlebars = require('handlebars');
  12. // MongoDB Raw Setup
  13. var collection = null;
  14. MongoClient.connect('mongodb://127.0.0.1/hello_world?maxPoolSize=5', function(err, db) {
  15. collection = db.collection('world');
  16. });
  17. // MySQL Raw Setup
  18. var connection = mysql.createConnection({
  19. host : '127.0.0.1',
  20. user : 'benchmarkdbuser',
  21. password : 'benchmarkdbpass',
  22. database : 'hello_world'
  23. });
  24. connection.connect();
  25. // Mongoose Setup
  26. var WorldSchema = new Mongoose.Schema({
  27. id : Number,
  28. randomNumber: Number
  29. }, {
  30. collection: 'world'
  31. });
  32. var FortuneSchema = new Mongoose.Schema({
  33. id: Number,
  34. message: String
  35. }, {
  36. collection: 'fortune'
  37. });
  38. var mongoose = {
  39. World: conn.model('World', WorldSchema),
  40. Fortune: conn.model('Fortune', FortuneSchema)
  41. }
  42. // Sequelize Setup
  43. var sequelize = new Sequelize('hello_world', 'benchmarkdbuser', 'benchmarkdbpass', {
  44. host: '127.0.0.1',
  45. dialect: 'mysql',
  46. logging: false
  47. });
  48. var World = sequelize.define('World', {
  49. id: {
  50. type: 'Sequelize.INTEGER'
  51. },
  52. randomNumber: {
  53. type: 'Sequelize.INTEGER'
  54. }
  55. }, {
  56. timestamps: false,
  57. freezeTableName: true
  58. });
  59. var Fortune = sequelize.define('Fortune', {
  60. id: {
  61. type: 'Sequelize.INTEGER'
  62. },
  63. message: {
  64. type: 'Sequelize.STRING'
  65. }
  66. }, {
  67. timestamps: false,
  68. freezeTableName: true
  69. });
  70. // Fortunes template via handlebars setup
  71. var fortunesTemplate = Handlebars.compile([
  72. "<!DOCTYPE html>",
  73. "<html>",
  74. "<head><title>Fortunes</title></head>",
  75. "<body>",
  76. "<table>",
  77. "<tr>",
  78. "<th>id</th>",
  79. "<th>message</th>",
  80. "</tr>",
  81. "{{#fortunes}}",
  82. "<tr>",
  83. "<td>{{id}}</td>",
  84. "<td>{{message}}</td>",
  85. "</tr>",
  86. "{{/fortunes}}",
  87. "</table>",
  88. "</body>",
  89. "</html>"
  90. ].join(''));
  91. // Helper functions
  92. function getRandomNumber() {
  93. return Math.floor(Math.random() * 10000) + 1;
  94. }
  95. function fillArray(value, len) {
  96. var arr = [];
  97. for (var i = 0; i < len; i++) {
  98. arr.push(value);
  99. }
  100. return arr;
  101. }
  102. function addTfbHeaders(res, headerType) {
  103. var headers = {
  104. 'Server': 'Node'
  105. }
  106. if (headerType === 'plain') {
  107. headers['Content-Type'] = 'text/plain; charset=UTF-8';
  108. } else if (headerType === 'json') {
  109. headers['Content-Type'] = 'application/json';
  110. } else if (headerType === 'html') {
  111. headers['Content-Type'] = 'text/html; cherset=UTF-8';
  112. }
  113. res.writeHead(200, headers);
  114. }
  115. // Mongoose Query Functions
  116. function mongooseRandomWorld(callback) {
  117. mongoose.World.findOne({
  118. id: getRandomNumber()
  119. }).exec(function (err, world) {
  120. callback(err, world);
  121. });
  122. }
  123. function mongooseGetAllFortunes(callback) {
  124. mongoose.Fortune.find({}).exec(function (err, fortunes) {
  125. callback(err, fortunes);
  126. });
  127. }
  128. // MongoDB-Raw Query Functions
  129. function mongodbDriverQuery(callback) {
  130. collection.findOne({
  131. id: getRandomNumber()
  132. }, function(err, world) {
  133. world._id = undefined; // remove _id from query response
  134. callback(err, world);
  135. });
  136. }
  137. function mongodbDriverUpdateQuery(callback) {
  138. collection.findAndModify({
  139. id: getRandomNumber()
  140. }, [['_id','asc']], {
  141. $set: {randomNumber: getRandomNumber()}
  142. }, {}, function(err, world) {
  143. world.value._id = undefined; // remove _id from query response
  144. callback(err, world.value);
  145. });
  146. }
  147. // Sequelize Query Functions
  148. function sequelizeQuery(callback) {
  149. World.findOne({
  150. where: {
  151. id: Math.floor(Math.random() * 10000) + 1}
  152. }
  153. ).complete(callback);
  154. }
  155. // MySQL-Raw Query Functions
  156. function mysqlQuery(callback) {
  157. connection.query("SELECT * FROM world WHERE id = " + getRandomNumber(), function (err, rows, fields) {
  158. if (err) {
  159. throw err;
  160. }
  161. callback(null, rows[0]);
  162. });
  163. }
  164. function mysqlUpdateQuery(callback) {
  165. connection.query("SELECT * FROM world WHERE id = " + getRandomNumber(), function (err, rows, fields) {
  166. if (err) {
  167. throw err;
  168. }
  169. rows[0].randomNumber = getRandomNumber();
  170. var updateQuery = "UPDATE world SET randomNumber = " + rows[0].randomNumber + " WHERE id = " + rows[0]['id'];
  171. connection.query(updateQuery, function (err, result) {
  172. if (err) {
  173. throw err;
  174. }
  175. callback(null, rows[0]);
  176. });
  177. });
  178. }
  179. var GREETING = "Hello, World!";
  180. var HELLO_OBJ = { message: GREETING };
  181. var ADDITIONAL_FORTUNE = {
  182. id: 0,
  183. message: 'Additional fortune added at request time.'
  184. };
  185. var responses = {
  186. jsonSerialization: function (req, res) {
  187. addTfbHeaders(res, 'json');
  188. res.end(JSON.stringify(HELLO_OBJ));
  189. },
  190. plaintext: function (req, res) {
  191. addTfbHeaders(res, 'plain');
  192. res.end(GREETING);
  193. },
  194. mongooseSingleQuery: function (req, res) {
  195. mongooseRandomWorld(function (err, result) {
  196. if (err) { throw err; }
  197. addTfbHeaders(res, 'json');
  198. res.end(JSON.stringify(result));
  199. })
  200. },
  201. mongooseMultipleQueries: function (queries, req, res) {
  202. var queryFunctions = fillArray(mongooseRandomWorld, queries)
  203. async.parallel(queryFunctions, function (err, results) {
  204. if (err) { throw err; }
  205. addTfbHeaders(res, 'json');
  206. res.end(JSON.stringify(results));
  207. });
  208. },
  209. mongooseFortunes: function (req, res) {
  210. mongooseGetAllFortunes(function (err, fortunes) {
  211. if (err) { throw err; }
  212. fortunes.push(ADDITIONAL_FORTUNE);
  213. fortunes.sort(function (a, b) {
  214. return a.message.localeCompare(b.message);
  215. });
  216. addTfbHeaders(res, 'html');
  217. res.end(fortunesTemplate({
  218. fortunes: fortunes
  219. }))
  220. });
  221. },
  222. mongooseUpdates: function (queries, req, res) {
  223. var selectFunctions = fillArray(mongooseRandomWorld, queries);
  224. async.parallel(selectFunctions, function (err, worlds) {
  225. var updateFunctions = [];
  226. for (var i = 0; i < queries; i++) {
  227. (function (i) {
  228. updateFunctions.push(function (callback) {
  229. worlds[i].randomNumber = getRandomNumber();
  230. mongoose.World.update({
  231. id: worlds[i].id
  232. }, {
  233. randomNumber: worlds[i].randomNumber
  234. }, callback);
  235. });
  236. })(i);
  237. }
  238. async.parallel(updateFunctions, function (err, results) {
  239. if (err) { throw err; }
  240. addTfbHeaders(res, 'json');
  241. // results does not have updated document information
  242. // if no err was found and thrown: all updates succeeded
  243. res.end(JSON.stringify(worlds));
  244. });
  245. });
  246. },
  247. mongodbSingleQuery: function (req, res) {
  248. mongodbDriverQuery(function (err, result) {
  249. if (err) { throw err; }
  250. addTfbHeaders(res, 'json');
  251. res.end(JSON.stringify(result));
  252. });
  253. },
  254. mongodbMultipleQueries: function (queries, req, res) {
  255. var queryFunctions = fillArray(mongodbDriverQuery, queries);
  256. async.parallel(queryFunctions, function (err, results) {
  257. if (err) { throw err; }
  258. addTfbHeaders(res, 'json');
  259. res.end(JSON.stringify(results));
  260. });
  261. },
  262. mongodbUpdates: function (queries, req, res) {
  263. var queryFunctions = fillArray(mongodbDriverUpdateQuery, queries);
  264. async.parallel(queryFunctions, function (err, results) {
  265. if (err) { throw err; }
  266. addTfbHeaders(res, 'json');
  267. res.end(JSON.stringify(results));
  268. });
  269. },
  270. sequelizeSingleQuery: function (req, res) {
  271. sequelizeQuery(function (err, result) {
  272. if (err) { throw err; }
  273. addTfbHeaders(res, 'json');
  274. res.end(JSON.stringify(result));
  275. });
  276. },
  277. sequelizeMultipleQueries: function (queries, req, res) {
  278. var queryFunctions = fillArray(sequelizeQuery, queries);
  279. async.parallel(queryFunctions, function (err, results) {
  280. if (err) { throw err; }
  281. addTfbHeaders(res, 'json');
  282. res.end(JSON.stringify(results));
  283. });
  284. },
  285. sequelizeFortunes: function (req, res) {
  286. Fortune.findAll().complete(function (err, fortunes) {
  287. fortunes.push(ADDITIONAL_FORTUNE);
  288. fortunes.sort(function (a, b) {
  289. return a.message.localeCompare(b.message);
  290. });
  291. addTfbHeaders(res, 'html');
  292. res.end(fortunesTemplate({
  293. fortunes: fortunes
  294. }));
  295. });
  296. },
  297. sequelizeUpdates: function (queries, req, res) {
  298. var selectFunctions = fillArray(sequelizeQuery, queries);
  299. async.parallel(selectFunctions, function (err, worlds) {
  300. var updateFunctions = [];
  301. for (var i = 0; i < queries; i++) {
  302. (function (i) {
  303. updateFunctions.push(function (callback) {
  304. worlds[i].randomNumber = getRandomNumber();
  305. worlds[i].save().complete(callback);
  306. });
  307. })(i);
  308. }
  309. async.parallel(updateFunctions, function (err, updates) {
  310. if (err) { throw err; }
  311. addTfbHeaders(res, 'json');
  312. res.end(JSON.stringify(updates));
  313. });
  314. });
  315. },
  316. mysqlSingleQuery: function (req, res) {
  317. mysqlQuery(function (err, result) {
  318. if (err) { throw err; }
  319. addTfbHeaders(res, 'json');
  320. res.end(JSON.stringify(result));
  321. });
  322. },
  323. mysqlMultipleQueries: function (queries, req, res) {
  324. var queryFunctions = fillArray(mysqlQuery, queries);
  325. async.parallel(queryFunctions, function (err, results) {
  326. if (err) { throw err; }
  327. addTfbHeaders(res, 'json');
  328. res.end(JSON.stringify(results));
  329. });
  330. },
  331. mysqlUpdates: function (queries, req, res) {
  332. var queryFunctions = fillArray(mysqlUpdateQuery, queries);
  333. async.parallel(queryFunctions, function (err, results) {
  334. if (err) { throw err; }
  335. addTfbHeaders(res, 'json');
  336. res.end(JSON.stringify(results));
  337. });
  338. },
  339. routeNotImplemented: function (req, res) {
  340. res.writeHead(501, {'Content-Type': 'text/plain; charset=UTF-8'});
  341. var reason = { reason: "`" + req.url + "` is not an implemented route" };
  342. res.end(JSON.stringify(reason));
  343. }
  344. };
  345. if (cluster.isMaster) {
  346. // Fork workers.
  347. for (var i = 0; i < numCPUs; i++) {
  348. cluster.fork();
  349. }
  350. cluster.on('exit', function (worker, code, signal) {
  351. console.log('worker ' + worker.pid + ' died');
  352. });
  353. } else {
  354. http.createServer(function (req, res) {
  355. var values = url.parse(req.url, true);
  356. var route = values.pathname;
  357. // Basic routes, no db required
  358. if (route === '/json') {
  359. return responses.jsonSerialization(req, res);
  360. } else if (route === '/plaintext') {
  361. return responses.plaintext(req, res);
  362. // No queries parameter required
  363. } else if (route === '/mongoose/db') {
  364. return responses.mongooseSingleQuery(req, res);
  365. } else if (route === '/mongoose/fortunes') {
  366. return responses.mongooseFortunes(req, res);
  367. } else if (route === '/mongodb/db') {
  368. return responses.mongodbSingleQuery(req, res);
  369. } else if (route === '/mysql-orm/db') {
  370. return responses.sequelizeSingleQuery(req, res);
  371. } else if (route === '/mysql-orm/fortunes') {
  372. return responses.sequelizeFortunes(req, res);
  373. } else if (route === '/mysql/db') {
  374. return responses.mysqlSingleQuery(req, res);
  375. }
  376. else {
  377. var queries = isNaN(values.query.queries) ? 1 : parseInt(values.query.queries, 10);
  378. queries = Math.min(Math.max(queries, 1), 500);
  379. if (route === '/mongoose/queries') {
  380. return responses.mongooseMultipleQueries(queries, req, res);
  381. } else if (route === '/mongoose/updates') {
  382. return responses.mongooseUpdates(queries, req, res);
  383. } else if (route === '/mongodb/queries') {
  384. return responses.mongodbMultipleQueries(queries, req, res);
  385. } else if (route === '/mongodb/updates') {
  386. return responses.mongodbUpdates(queries, req, res);
  387. } else if (route === '/mysql-orm/queries') {
  388. return responses.sequelizeMultipleQueries(queries, req, res);
  389. } else if (route === '/mysql-orm/updates') {
  390. return responses.sequelizeUpdates(queries, req, res);
  391. } else if (route === '/mysql/queries') {
  392. return responses.mysqlMultipleQueries(queries, req, res);
  393. } else if (route === '/mysql/updates') {
  394. return responses.mysqlUpdates(queries, req, res);
  395. } else {
  396. return responses.routeNotImplemented(req, res);
  397. }
  398. }
  399. }).listen(8080, function() {
  400. console.log("NodeJS worker listening on port 8080");
  401. });
  402. }