Add partition map reload

src/io/cluster/partitionclusterclient.js

@@ -6,9 +6,8 @@ class PartitionClusterClient {
     }
 
     getSocketConfig(channel) {
-        return Promise.resolve({
-            servers: this.partitionDecider.getPartitionForChannel(channel)
-        });
+        return Promise.resolve(
+                this.partitionDecider.getPartitionForChannel(channel));
     }
 }
 

src/partition/partitiondecider.js

@@ -2,27 +2,28 @@ import { murmurHash1 } from '../util/murmur';
 
 class PartitionDecider {
     constructor(config) {
-        this.identity = config.getIdentity();
-        this.partitionMap = config.getPartitionMap();
-        this.pool = config.getPool();
-        this.overrideMap = config.getOverrideMap();
+        this.config = config;
     }
 
     getPartitionForChannel(channel) {
-        return this.partitionMap[this.getPartitionIdentityForChannel(channel)];
+        const partitionMap = this.config.getPartitionMap();
+        return partitionMap[this.getPartitionIdentityForChannel(channel)];
     }
 
     getPartitionIdentityForChannel(channel) {
-        if (this.overrideMap.hasOwnProperty(channel)) {
-            return this.overrideMap[channel];
+        const overrideMap = this.config.getOverrideMap();
+        if (overrideMap.hasOwnProperty(channel)) {
+            return overrideMap[channel];
         } else {
-            const i = murmurHash1(channel) % this.pool.length;
-            return this.pool[i];
+            const pool = this.config.getPool();
+            const i = murmurHash1(channel) % pool.length;
+            return pool[i];
         }
     }
 
     isChannelOnThisPartition(channel) {
-        return this.getPartitionIdentityForChannel(channel) === this.identity;
+        return this.getPartitionIdentityForChannel(channel) ===
+                this.config.getIdentity();
     }
 }
 

src/server.js

@@ -68,12 +68,12 @@ var Server = function () {
             Switches.setActive(Switches.DUAL_BACKEND, true);
         }
         const BackendModule = require('./backend/backendmodule').BackendModule;
-        initModule = new BackendModule();
+        initModule = this.initModule = new BackendModule();
     } else if (Config.get('enable-partition')) {
-        initModule = new PartitionModule();
+        initModule = this.initModule = new PartitionModule();
         self.partitionDecider = initModule.getPartitionDecider();
     } else {
-        initModule = new LegacyModule();
+        initModule = this.initModule = new LegacyModule();
     }
 
     // database init ------------------------------------------------------
@@ -302,3 +302,43 @@ Server.prototype.shutdown = function () {
         process.exit(1);
     });
 };
+
+Server.prototype.reloadPartitionMap = function () {
+    if (!Config.get("enable-partition")) {
+        return;
+    }
+
+    var config;
+    try {
+        config = this.initModule.loadPartitionMap();
+    } catch (error) {
+        return;
+    }
+
+    this.initModule.partitionConfig.config = config.config;
+
+    const channels = Array.prototype.slice.call(this.channels);
+    Promise.reduce(channels, (_, channel) => {
+        if (channel.dead) {
+            return;
+        }
+
+        if (!this.partitionDecider.isChannelOnThisPartition(channel.uniqueName)) {
+            Logger.syslog.log("Partition changed for " + channel.uniqueName);
+            return channel.saveState().then(() => {
+                channel.broadcastAll("partitionChange",
+                        this.partitionDecider.getPartitionForChannel(channel.uniqueName));
+                const users = Array.prototype.slice.call(channel.users);
+                users.forEach(u => {
+                    try {
+                        u.socket.disconnect();
+                    } catch (error) {
+                    }
+                });
+                this.unloadChannel(channel);
+            });
+        }
+    }, 0).then(() => {
+        Logger.syslog.log("Partition reload complete");
+    });
+};