From 58a73b52b50de926d20689ea972d6a3c0eb8dc16 Mon Sep 17 00:00:00 2001
From: Julian Lam <julian@designcreateplay.com>
Date: Wed, 3 Sep 2014 01:24:26 -0400
Subject: [PATCH] work-in-progress commit to get cluster module working with
 loader.js

---
 loader.js               | 238 +++++++++++++++++++++++-----------------
 src/middleware/index.js |   9 +-
 src/routes/index.js     |   7 --
 src/webserver.js        |  11 +-
 4 files changed, 155 insertions(+), 110 deletions(-)

diff --git a/loader.js b/loader.js
index 48d4929238..6d65b203b5 100644
--- a/loader.js
+++ b/loader.js
@@ -2,109 +2,147 @@
 
 var	nconf = require('nconf'),
 	fs = require('fs'),
-	pidFilePath = __dirname + '/pidfile',
-	output = fs.openSync(__dirname + '/logs/output.log', 'a'),
-	start = function() {
-		var	nbb_start = function(callback) {
-				if (timesStarted > 3) {
-					console.log('\n[loader] Experienced three start attempts in 10 seconds, most likely an error on startup. Halting.');
-					return nbb_stop();
-				}
-
-				timesStarted++;
-				if (startTimer) {
-					clearTimeout(startTimer);
-				}
-				startTimer = setTimeout(resetTimer, 1000*10);
-
-				if (nbb) {
-					nbbOld = nbb;
-				}
-
-				nbb = require('child_process').fork('./app', process.argv.slice(2), {
-					env: process.env
-				});
-
-				nbb.on('message', function(message) {
-					if (message && typeof message === 'object' && message.action) {
-						switch (message.action) {
-							case 'ready':
-								if (!callback) return nbb.send('bind');
-								callback();
-							break;
-							case 'restart':
-								nbb_restart();
-							break;
-						}
-					}
-				});
-
-				nbb.on('exit', function(code, signal) {
-					if (code) {
-						nbb_start();
-					} else {
-						nbb_stop();
-					}
-				});
-			},
-			nbb_stop = function() {
-				if (startTimer) {
-					clearTimeout(startTimer);
-				}
-
-				nbb.kill();
-				if (fs.existsSync(pidFilePath)) {
-					var	pid = parseInt(fs.readFileSync(pidFilePath, { encoding: 'utf-8' }), 10);
-					if (process.pid === pid) {
-						fs.unlinkSync(pidFilePath);
-					}
-				}
-			},
-			nbb_restart = function() {
-				nbb_start(function() {
-					nbbOld.removeAllListeners('exit').on('exit', function() {
-						nbb.send('bind');
-					});
-					nbbOld.kill();
-				});
-			},
-			resetTimer = function() {
-				clearTimeout(startTimer);
-				timesStarted = 0;
-			},
-			timesStarted = 0,
-			startTimer;
-
-		process.on('SIGINT', nbb_stop);
-		process.on('SIGTERM', nbb_stop);
-		process.on('SIGHUP', nbb_restart);
-
-		nbb_start();
-	},
-	nbb, nbbOld;
+	cluster = require('cluster'),
+	numCPUs = require('os').cpus().length;
+
+	/* TODO
+	    * pidFile and reset timer
+	    * logging
+	    * handling SIGHUP
+	    * restart signal from child
+	*/
+
+	// pidFilePath = __dirname + '/pidfile',
+	// output = fs.openSync(__dirname + '/logs/output.log', 'a'),
+	// start = function() {
+	// 	var	nbb_start = function(callback) {
+	// 			if (timesStarted > 3) {
+	// 				console.log('\n[loader] Experienced three start attempts in 10 seconds, most likely an error on startup. Halting.');
+	// 				return nbb_stop();
+	// 			}
+
+	// 			timesStarted++;
+	// 			if (startTimer) {
+	// 				clearTimeout(startTimer);
+	// 			}
+	// 			startTimer = setTimeout(resetTimer, 1000*10);
+
+	// 			if (nbb) {
+	// 				nbbOld = nbb;
+	// 			}
+
+	// 			nbb = require('child_process').fork('./app', process.argv.slice(2), {
+	// 				env: process.env
+	// 			});
+
+	// 			nbb.on('message', function(message) {
+	// 				if (message && typeof message === 'object' && message.action) {
+	// 					switch (message.action) {
+	// 						case 'ready':
+	// 							if (!callback) return nbb.send('bind');
+	// 							callback();
+	// 						break;
+	// 						case 'restart':
+	// 							nbb_restart();
+	// 						break;
+	// 					}
+	// 				}
+	// 			});
+
+	// 			nbb.on('exit', function(code, signal) {
+	// 				if (code) {
+	// 					nbb_start();
+	// 				} else {
+	// 					nbb_stop();
+	// 				}
+	// 			});
+	// 		},
+	// 		nbb_stop = function() {
+	// 			if (startTimer) {
+	// 				clearTimeout(startTimer);
+	// 			}
+
+	// 			nbb.kill();
+	// 			if (fs.existsSync(pidFilePath)) {
+	// 				var	pid = parseInt(fs.readFileSync(pidFilePath, { encoding: 'utf-8' }), 10);
+	// 				if (process.pid === pid) {
+	// 					fs.unlinkSync(pidFilePath);
+	// 				}
+	// 			}
+	// 		},
+	// 		nbb_restart = function() {
+	// 			nbb_start(function() {
+	// 				nbbOld.removeAllListeners('exit').on('exit', function() {
+	// 					nbb.send('bind');
+	// 				});
+	// 				nbbOld.kill();
+	// 			});
+	// 		},
+	// 		resetTimer = function() {
+	// 			clearTimeout(startTimer);
+	// 			timesStarted = 0;
+	// 		},
+	// 		timesStarted = 0,
+	// 		startTimer;
+
+	// 	process.on('SIGINT', nbb_stop);
+	// 	process.on('SIGTERM', nbb_stop);
+	// 	process.on('SIGHUP', nbb_restart);
+
+	// 	nbb_start();
+	// },
+	// nbb, nbbOld;
 
 nconf.argv();
 
-// Start the daemon!
-if (nconf.get('daemon') !== false) {
-	// Check for a still-active NodeBB process
-	if (fs.existsSync(pidFilePath)) {
-		try {
-			var	pid = fs.readFileSync(pidFilePath, { encoding: 'utf-8' });
-			process.kill(pid, 0);
-			process.exit();
-		} catch (e) {
-			fs.unlinkSync(pidFilePath);
-		}
-	}
+cluster.setupMaster({
+	exec: "app.js",
+	silent: true
+});
+
+for(var x=0;x<numCPUs;x++) {
+	cluster.fork();
+}
+
+cluster.on('exit', function(worker, code, signal) {
+	console.log('worker ' + worker.process.pid + ' died');
+});
 
-	// Daemonize and record new pid
-	require('daemon')({
-		stdout: output
+Object.keys(cluster.workers).forEach(function(id) {
+	cluster.workers[id].on('message', function(message) {
+		if (message && typeof message === 'object' && message.action) {
+			switch (message.action) {
+				case 'ready':
+					cluster.workers[id].send('bind');
+				break;
+				// case 'restart':
+				// 	nbb_restart();
+				// break;
+			}
+		}
 	});
-	fs.writeFile(__dirname + '/pidfile', process.pid);
+});
 
-	start();
-} else {
-	start();
-}
+// Start the daemon!
+// if (nconf.get('daemon') !== false) {
+// 	// Check for a still-active NodeBB process
+// 	if (fs.existsSync(pidFilePath)) {
+// 		try {
+// 			var	pid = fs.readFileSync(pidFilePath, { encoding: 'utf-8' });
+// 			process.kill(pid, 0);
+// 			process.exit();
+// 		} catch (e) {
+// 			fs.unlinkSync(pidFilePath);
+// 		}
+// 	}
+
+// 	// Daemonize and record new pid
+// 	require('daemon')({
+// 		stdout: output
+// 	});
+// 	fs.writeFile(__dirname + '/pidfile', process.pid);
+
+// 	start();
+// } else {
+// 	start();
+// }
\ No newline at end of file
diff --git a/src/middleware/index.js b/src/middleware/index.js
index 5ea711b46c..140393a1ee 100644
--- a/src/middleware/index.js
+++ b/src/middleware/index.js
@@ -22,6 +22,7 @@ var utils = require('./../../public/src/utils'),
 	multipart = require('connect-multiparty'),
 	csrf = require('csurf'),
 	session = require('express-session'),
+	cluster = require('cluster'),
 
 	relativePath,
 	themesPath;
@@ -128,7 +129,13 @@ module.exports = function(app, data) {
 	routeCurrentTheme(app, data.currentThemeId, data.themesData);
 	routeThemeScreenshots(app, data.themesData);
 
-	meta.templates.compile();
+	if (cluster.worker.id === 1) {
+		meta.templates.compile();
+	} else {
+		setTimeout(function() {
+			emitter.emit('templates:compiled');
+		}, 1000);
+	}
 
 	return middleware;
 };
diff --git a/src/routes/index.js b/src/routes/index.js
index dc2e0f0d94..c74a350fd2 100644
--- a/src/routes/index.js
+++ b/src/routes/index.js
@@ -198,13 +198,6 @@ module.exports = function(app, middleware) {
 
 	// Add plugin routes
 	plugins.reloadRoutes();
-	// plugins.ready(function() {
-	// 	plugins.fireHook('static:app.load', pluginRouter, middleware, controllers, function() {
-
-
-
-	// 	});
-	// });
 };
 
 function handleErrors(err, req, res, next) {
diff --git a/src/webserver.js b/src/webserver.js
index 3278713dd1..9e336d55d9 100644
--- a/src/webserver.js
+++ b/src/webserver.js
@@ -6,6 +6,7 @@ var path = require('path'),
 	server,
 	winston = require('winston'),
 	async = require('async'),
+	cluster = require('cluster'),
 
 	emailer = require('./emailer'),
 	db = require('./database'),
@@ -39,13 +40,19 @@ if(nconf.get('ssl')) {
 	auth.registerApp(app);
 	emailer.registerApp(app);
 	notifications.init();
-	user.startJobs();
+	
+	if (cluster.worker.id === 1) {
+		user.startJobs();
+	}
 
 	// Preparation dependent on plugins
 	plugins.ready(function() {
 		meta.js.minify(app.enabled('minification'));
 		meta.css.minify();
-		meta.sounds.init();
+		
+		if (cluster.worker.id === 1) {
+			meta.sounds.init();
+		}
 	});
 
 	async.parallel({