created hotfix for issue #4

nikfoundas · Jul 6, 2016 · e9c0db0 · e9c0db0
1 parent d4f3e74
commit e9c0db0
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/main/java/org/github/etcd/service/impl/ClusterManagerImpl.java b/src/main/java/org/github/etcd/service/impl/ClusterManagerImpl.java
@@ -45,10 +45,17 @@ public int compare(EtcdMember o1, EtcdMember o2) {
 
     private Map<String, EtcdCluster> clusters = Collections.synchronizedMap(new LinkedHashMap<String, EtcdCluster>());
 
+    private static final String DEFAULT_ETCD_CLIENT = "ETCD_CLIENT_URL";
 
     public ClusterManagerImpl() {
-        addCluster("default", "http://localhost:2379/");
-        addCluster("kvm", "http://192.168.122.101:4001/");
+
+        String etcdAddress = System.getenv(DEFAULT_ETCD_CLIENT);
+        if (etcdAddress == null) {
+            etcdAddress = System.getProperty(DEFAULT_ETCD_CLIENT, "http://localhost:2379/");
+        }
+
+        addCluster("default", etcdAddress);
+        // addCluster("kvm", "http://192.168.122.201:2379/");
     }
 
     @Override