Merge pull request #322 from Nicholi/workerclusterInitFix

Workercluster init fix

Author: jondubois

index.js

@@ -109,6 +109,7 @@ SocketCluster.prototype._init = function (options) {
     brokerController: null,
     brokerConnectRetryErrorThreshold: null,
     initController: null,
+    initWorkerClusterController: null,
     rebootOnSignal: true,
     downgradeToUser: false,
     path: '/socketcluster/',
@@ -182,6 +183,12 @@ SocketCluster.prototype._init = function (options) {
       self._paths.appInitControllerPath = null;
   }
 
+  if (self.options.initWorkerClusterController) {
+    self._paths.appInitWorkerClusterControllerPath = path.resolve(self.options.initWorkerClusterController);
+  } else {
+    self._paths.appInitWorkerClusterControllerPath = null;
+  }
+
   if (/\.js$/.test(self.options.wsEngine)) {
     self._paths.wsEnginePath = path.resolve(self.options.wsEngine);
   } else {

lib/workercluster.js

@@ -45,6 +45,12 @@ if (cluster.isMaster) {
         processTermTimeout = m.data.processTermTimeout;
       }
 
+      // Run an initController to initialize anything for the master process of all SCWorkers
+      if (m.data.paths.appInitWorkerClusterControllerPath != null) {
+        var _initWorkerClusterController = require(m.data.paths.appInitWorkerClusterControllerPath);
+        _initWorkerClusterController.run();
+      }
+
       var workerCount = m.data.workerCount;
       var readyCount = 0;
       var isReady = false;

sample/server.js

@@ -8,6 +8,7 @@ var SocketCluster = require('socketcluster').SocketCluster;
 var workerControllerPath = argv.wc || process.env.SOCKETCLUSTER_WORKER_CONTROLLER;
 var brokerControllerPath = argv.bc || process.env.SOCKETCLUSTER_BROKER_CONTROLLER;
 var initControllerPath = argv.ic || process.env.SOCKETCLUSTER_INIT_CONTROLLER;
+var initWorkerClusterControllerPath = argv.iwc || process.env.SOCKETCLUSTER_INIT_WORKERCLUSTER_CONTROLLER;
 var environment = process.env.ENV || 'dev';
 
 var options = {
@@ -20,6 +21,7 @@ var options = {
   workerController: workerControllerPath || __dirname + '/worker.js',
   brokerController: brokerControllerPath || __dirname + '/broker.js',
   initController: initControllerPath || null,
+  initWorkerClusterController: initWorkerClusterControllerPath || null,
   socketChannelLimit: Number(process.env.SOCKETCLUSTER_SOCKET_CHANNEL_LIMIT) || 1000,
   clusterStateServerHost: argv.cssh || process.env.SCC_STATE_SERVER_HOST || null,
   clusterStateServerPort: process.env.SCC_STATE_SERVER_PORT || null,