From 4dff67e03ba61f1489066561ea59055119150fb4 Mon Sep 17 00:00:00 2001
From: Geoff Bourne <itzgeoff@gmail.com>
Date: Mon, 16 Feb 2015 14:50:03 +0000
Subject: [PATCH] [titan-gremlin] Automatically use 'itzg/cassandra' image's
 publish-as address

---
 titan-gremlin/start-gremlin.sh | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/titan-gremlin/start-gremlin.sh b/titan-gremlin/start-gremlin.sh
index 521bd4fa..4408c2ec 100755
--- a/titan-gremlin/start-gremlin.sh
+++ b/titan-gremlin/start-gremlin.sh
@@ -9,14 +9,27 @@ fi
 rm -f /tmp/titan.properties
 
 if [ -n "$CASS_PORT_9160_TCP_ADDR" ]; then
+
   shortcut=/tmp/titan.properties
   cat >> /tmp/titan.properties <<END
 storage.backend=cassandra
 storage.hostname=$CASS_PORT_9160_TCP_ADDR
 END
+
+elif [ -n "$CASS_ADDR" ]; then
+
+  shortcut=/tmp/titan.properties
+  cat >> /tmp/titan.properties <<END
+storage.backend=cassandra
+storage.hostname=$CASS_ADDR
+END
+
 fi
 
-if [ -n "$ES_CLUSTER" -o -n "$ES_PORT_9300_TCP_PORT" ]; then
+
+esAddr=${ES_ENV_PUBLISH_AS:-ES_PORT_9300_TCP_ADDR}
+
+if [ -n "$ES_CLUSTER" -o -n "$esAddr" ]; then
   shortcut=/tmp/titan.properties
   cat >> /tmp/titan.properties <<END
 index.search.backend=elasticsearch
@@ -28,9 +41,11 @@ END
 index.search.elasticsearch.ext.cluster.name=$ES_CLUSTER
 END
   fi
-  if [ -n "$ES_PORT_9300_TCP_PORT" ]; then
+  if [ -n "$esAddr" ]; then
+    # strip off the port spec, if present
+    esAddr=$(echo $esAddr | cut -d: -f1)
     cat >> /tmp/titan.properties <<END
-index.search.hostname=$ES_PORT_9300_TCP_ADDR
+index.search.hostname=$esAddr
 END
   fi