Add partition map reload

index.js

@@ -73,5 +73,7 @@ function handleLine(line) {
             Logger.syslog.log("Switch " + args[0] + " is now " +
                     (Switches.isActive(args[0]) ? "ON" : "OFF"));
         }
+    } else if (line.indexOf("/reload-partitions") === 0) {
+        sv.reloadPartitionMap();
     }
 }

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");
+    });
+};

www/js/callbacks.js

@@ -1030,6 +1030,12 @@ Callbacks = {
             "unneeded playlist items, filters, and/or emotes.  Changes to the channel " +
             "will not be saved until the size is reduced to under the limit.")
             .attr("id", "chandumptoobig");
+    },
+
+    partitionChange: function (socketConfig) {
+        window.socket.disconnect();
+        ioServerConnect(socketConfig);
+        setupCallbacks();
     }
 }
 
@@ -1065,25 +1071,7 @@ setupCallbacks = function() {
     });
 };
 
-(function () {
-    if (typeof io === "undefined") {
-        var script = document.getElementById("socketio-js");
-        var source = "unknown";
-        if (script) {
-            source = script.src;
-        }
-
-        var message = "The socket.io library could not be loaded from <code>" +
-                      source + "</code>.  Ensure that it is not being blocked " +
-                      "by a script blocking extension or firewall and try again.";
-        makeAlert("Error", message, "alert-danger")
-            .appendTo($("#announcements"));
-        Callbacks.disconnect();
-        return;
-    }
-
-    $.getJSON("/socketconfig/" + CHANNEL.name + ".json")
-        .done(function (socketConfig) {
+function ioServerConnect(socketConfig) {
     if (socketConfig.error) {
         makeAlert("Error", "Socket.io configuration returned error: " +
                 socketConfig.error, "alert-danger")
@@ -1124,7 +1112,29 @@ setupCallbacks = function() {
         secure: chosenServer.secure
     };
 
-            socket = io(chosenServer.url, opts);
+    window.socket = io(chosenServer.url, opts);
+}
+
+(function () {
+    if (typeof io === "undefined") {
+        var script = document.getElementById("socketio-js");
+        var source = "unknown";
+        if (script) {
+            source = script.src;
+        }
+
+        var message = "The socket.io library could not be loaded from <code>" +
+                      source + "</code>.  Ensure that it is not being blocked " +
+                      "by a script blocking extension or firewall and try again.";
+        makeAlert("Error", message, "alert-danger")
+            .appendTo($("#announcements"));
+        Callbacks.disconnect();
+        return;
+    }
+
+    $.getJSON("/socketconfig/" + CHANNEL.name + ".json")
+        .done(function (socketConfig) {
+            ioServerConnect(socketConfig);
             setupCallbacks();
         }).fail(function () {
             makeAlert("Error", "Failed to retrieve socket.io configuration",