hello.js 13 KB

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