From 940968bd96954ab295df3659816c1db36fb22542 Mon Sep 17 00:00:00 2001
From: Jan Steemann <jan@arangodb.com>
Date: Fri, 13 Mar 2015 18:26:50 +0100
Subject: [PATCH] when specifying the same dispatcher IP address multiple
 times, try to create non-overlapping port ranges

---
 .../modules/org/arangodb/cluster/planner.js   | 49 ++++++++++++++++++-
 1 file changed, 47 insertions(+), 2 deletions(-)

diff --git a/js/server/modules/org/arangodb/cluster/planner.js b/js/server/modules/org/arangodb/cluster/planner.js
index 1507c5f6e7..ef9d021703 100644
--- a/js/server/modules/org/arangodb/cluster/planner.js
+++ b/js/server/modules/org/arangodb/cluster/planner.js
@@ -133,9 +133,16 @@ PortFinder.prototype.next = function () {
   while (true) {   // will be left by return when port is found
     if (this.pos < this.list.length) {
       this.port = this.list[this.pos++];
+      if (this.dispatcher.hasOwnProperty("portOverlapIndex")) {
+        // add some value to the initial port if we had multiple 
+        // dispatchers on the same IP address. Otherwise, the
+        // dispatchers will try to bind the exact same ports, and
+        // this is prone to races
+        this.port += this.dispatcher.portOverlapIndex * 20;
+      }
     }
     else if (this.port === 0) {
-      this.port = Math.floor(Math.random()*(65536-1024))+1024;
+      this.port = Math.floor(Math.random() * (65536 - 1024)) + 1024;
     }
     else {
       this.port++;
@@ -151,7 +158,6 @@ PortFinder.prototype.next = function () {
         available = testPort("tcp://0.0.0.0:" + this.port);
       }
       else {
-	  require("internal").print(this.dispatcher.endpoint);
         var url = endpointToURL(this.dispatcher.endpoint) +
                   "/_admin/clusterCheckPort?port="+this.port;
         var hdrs = {};
@@ -230,6 +236,43 @@ function fillConfigWithDefaults (config, defaultConfig) {
   }
 }
 
+////////////////////////////////////////////////////////////////////////////////
+/// @brief check if IP addresses of dispatchers are non-unique, and if yes,
+/// then set a flag for each dispatcher with non-unique address
+/// we need this to prevent race conditions when multiple dispatchers on the
+/// same physical server start coordinators or dbservers and compete for the 
+/// same ports
+////////////////////////////////////////////////////////////////////////////////
+
+function checkDispatcherIps (config) {
+  if (typeof config.dispatchers !== "object") {
+    return;
+  }
+
+  var d, dispatcherIps = { };
+  for (d in config.dispatchers) {
+    if (config.dispatchers.hasOwnProperty(d)) {
+      var ip = config.dispatchers[d].endpoint.replace(/:\d+$/, '').replace(/^(ssl|tcp):\/\//, '');
+      if (! dispatcherIps.hasOwnProperty(ip)) {
+        dispatcherIps[ip] = [ ];
+      }
+      dispatcherIps[ip].push(d);
+    }
+  }
+
+  for (d in dispatcherIps) {
+    if (dispatcherIps.hasOwnProperty(d)) {
+      if (dispatcherIps[d].length > 1) {
+        // more than one dispatcher for an IP address
+        // need to ensure that the port ranges do not overlap
+        for (var i = 0; i < dispatcherIps[d].length; ++i) {
+          config.dispatchers[dispatcherIps[d][i]].portOverlapIndex = i;
+        }
+      }
+    }
+  }
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 /// @startDocuBlock JSF_Cluster_Planner_Constructor
 ///
@@ -395,6 +438,8 @@ function Planner (userConfig) {
     throw new Error("userConfig must be an object");
   }
   this.config = copy(userConfig);
+  checkDispatcherIps(this.config);
+
   fillConfigWithDefaults(this.config, PlannerLocalDefaults);
   this.commands = [];
   this.makePlan();