Dynomite api support

dyno-core/src/main/java/com/netflix/dyno/connectionpool/impl/lb/AbstractTokenMapSupplier.java

@@ -123,6 +123,12 @@ public AbstractTokenMapSupplier(int port) {
         localDatacenter = ConfigUtils.getDataCenter();
         unsuppliedPort = port;
     }
+
+    public AbstractTokenMapSupplier(int port,String localZone,String localDatacenter) {
+        this.localZone = localZone;
+        this.localDatacenter = localDatacenter;
+        unsuppliedPort = port;
+    }
 
     public abstract String getTopologyJsonPayload(Set<Host> activeHosts);
 
@@ -238,4 +244,4 @@ List<HostToken> parseTokenListFromJson(String json) {
 
         return hostTokens;
     }
-}
+}

dyno-core/src/main/java/com/netflix/dyno/connectionpool/impl/lb/DynomiteHttpEndpointBasedTokenMapSupplier.java

@@ -0,0 +1,135 @@
+/*******************************************************************************
+ * Copyright 2017 Netflix
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ ******************************************************************************/
+package com.netflix.dyno.connectionpool.impl.lb;
+
+import java.util.Iterator;
+import java.util.Set;
+
+import org.json.simple.JSONArray;
+import org.json.simple.JSONObject;
+import org.json.simple.parser.JSONParser;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.netflix.dyno.connectionpool.Host;
+import com.netflix.dyno.connectionpool.exception.DynoException;
+import com.netflix.dyno.connectionpool.impl.lb.HttpEndpointBasedTokenMapSupplier;
+
+/**
+ * 
+ * Class to dynomite api REST
+ * 
+ */
+public class DynomiteHttpEndpointBasedTokenMapSupplier extends HttpEndpointBasedTokenMapSupplier {
+
+    private static final Logger Logger = LoggerFactory.getLogger(DynomiteHttpEndpointBasedTokenMapSupplier.class);
+
+    private static final String DEFAULT_URL = "http://{hostname}:{port}/cluster_describe";
+    private static final Integer DEFAULT_PORT = 22222;
+
+    public DynomiteHttpEndpointBasedTokenMapSupplier() {
+        super(DEFAULT_URL,DEFAULT_PORT);
+    }
+
+    public DynomiteHttpEndpointBasedTokenMapSupplier(String localDatacenter, String localRack) {
+        super(DEFAULT_URL,DEFAULT_PORT,localDatacenter,localRack);
+    }
+
+    public DynomiteHttpEndpointBasedTokenMapSupplier(int port) {
+        super(port);
+    }
+
+    public DynomiteHttpEndpointBasedTokenMapSupplier(String url, int port) {
+        super(url, port);
+    }
+
+    public DynomiteHttpEndpointBasedTokenMapSupplier(String url, int port, String localDatacenter, String localRack) {
+        super(url, port,localDatacenter,localRack);
+    }
+
+    @Override
+    public String getTopologyJsonPayload(String hostname) {
+        return dynomiteTrans(super.getTopologyJsonPayload(hostname));
+    }
+
+    @Override
+    public String getTopologyJsonPayload(Set<Host> activeHosts) {
+        return dynomiteTrans(super.getTopologyJsonPayload(activeHosts));
+    }
+
+    @SuppressWarnings("unchecked")
+    public String dynomiteTrans(String json) {
+        JSONArray ja = new JSONArray();
+
+        JSONParser parser = new JSONParser();
+        try {
+            JSONObject jitem = (JSONObject) parser.parse(json);
+            JSONArray arr = (JSONArray) jitem.get("dcs");
+            Iterator<?> iter = arr.iterator();
+            while (iter.hasNext()) {
+                Object item = iter.next();
+                if (!(item instanceof JSONObject)) {
+                    continue;
+                }
+                JSONObject jItem = (JSONObject) item;
+                String datacenter = (String) jItem.get("name");
+                JSONArray racks = (JSONArray) jItem.get("racks");
+
+                Iterator<?> racksI = racks.iterator();
+                while (racksI.hasNext()) {
+                    Object rackItem = racksI.next();
+                    if (!(rackItem instanceof JSONObject)) {
+                        continue;
+                    }
+                    JSONObject jRackItem = (JSONObject) rackItem;
+                    String rackName = (String) jRackItem.get("name");
+                    JSONArray servers = (JSONArray) jRackItem.get("servers");
+
+                    Iterator<?> serversI = servers.iterator();
+                    while (serversI.hasNext()) {
+                        Object serverItem = serversI.next();
+                        if (!(serverItem instanceof JSONObject)) {
+                            continue;
+                        }
+                        JSONObject JServerItem = (JSONObject) serverItem;
+                        String serverName = (String) JServerItem.get("name");
+                        String serverHost = (String) JServerItem.get("host");
+                        String serverPort = String.valueOf((Long) JServerItem.get("port"));
+                        String serverToken = String.valueOf((Long) JServerItem.get("token"));
+
+                        JSONObject jnode = new JSONObject();
+                        jnode.put("token", serverToken);
+                        jnode.put("serverName", serverName);
+                        jnode.put("serverPort", serverPort);
+                        jnode.put("hostname", serverHost);
+                        jnode.put("rack", rackName);
+                        jnode.put("ip", serverHost);
+                        jnode.put("zone", rackName);
+                        jnode.put("dc", datacenter);
+                        ja.add(jnode);
+                    }
+
+                }
+
+            }
+        } catch (Exception lastEx) {
+            Logger.error("Process error on dynomite rest json = [ " + json + " ]", lastEx);
+            throw new DynoException(lastEx);
+        }
+        return ja.toJSONString();
+    }
+
+}

dyno-core/src/main/java/com/netflix/dyno/connectionpool/impl/lb/HttpEndpointBasedTokenMapSupplier.java

@@ -55,16 +55,24 @@ public HttpEndpointBasedTokenMapSupplier() {
     public HttpEndpointBasedTokenMapSupplier(int port) {
 	this(DefaultServerUrl, port);
     }
+
+    public HttpEndpointBasedTokenMapSupplier(String url, int port,String localDatacenter,String localRack) {
+    super(port,localDatacenter,localRack);
+    serverUrl = transformUrl(url,port);
+    }
 
     public HttpEndpointBasedTokenMapSupplier(String url, int port) {
 	super(port);
-
-	/**
-	 * If no port is passed means -1 then we will substitute to defaultPort
-	 * else the passed one.
-	 */
-	url = url.replace("{port}", (port > -1) ? Integer.toString(port) : Integer.toString(defaultPort));
-	serverUrl = url;
+	serverUrl = transformUrl(url,port);
+    }
+
+    /**
+     * If no port is passed means -1 then we will substitute to defaultPort
+     * else the passed one.
+    */
+    private static String transformUrl(String url,int port){
+     url = url.replace("{port}", (port > -1) ? Integer.toString(port) : Integer.toString(defaultPort));
+     return url;
     }
 
     @Override
@@ -75,7 +83,7 @@ public String getTopologyJsonPayload(String hostname) {
 	    throw new RuntimeException(e);
 	}
     }
-
+    
     /**
      * Tries to get topology information by randomly trying across nodes.
      */
@@ -209,4 +217,4 @@ private String getTopologyWithNodeRetry(Set<Host> activeHosts) {
 
     }
 
-}
+}