hello.js 11 KB

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