JMX API: Adding the EndpointSnitchInfoMBean stab

This adds a stab to the EndpointSnitchInfoMBean

Signed-off-by: Amnon Heiman <amnon@cloudius-systems.com>

src/main/java/com/cloudius/urchin/main/Main.java

@@ -7,6 +7,7 @@ import com.cloudius.urchin.api.APIClient;
 
 import org.apache.cassandra.db.commitlog.CommitLog;
 import org.apache.cassandra.gms.Gossiper;
+import org.apache.cassandra.locator.EndpointSnitchInfo;
 import org.apache.cassandra.net.MessagingService;
 import org.apache.cassandra.service.StorageService;
 
@@ -19,6 +20,7 @@ public class Main {
         MessagingService.getInstance();
         CommitLog.getInstance();
         Gossiper.getInstance();
+        EndpointSnitchInfo.getInstance();
         Thread.sleep(Long.MAX_VALUE);
     }
 

src/main/java/org/apache/cassandra/locator/EndpointSnitchInfo.java

@@ -0,0 +1,89 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 org.apache.cassandra.locator;
+
+import java.lang.management.ManagementFactory;
+import java.net.UnknownHostException;
+
+import javax.management.MBeanServer;
+import javax.management.ObjectName;
+
+import com.cloudius.urchin.api.APIClient;
+
+public class EndpointSnitchInfo implements EndpointSnitchInfoMBean {
+    private static final java.util.logging.Logger logger = java.util.logging.Logger
+            .getLogger(EndpointSnitchInfo.class.getName());
+
+    private APIClient c = new APIClient();
+
+    public void log(String str) {
+        System.out.println(str);
+        logger.info(str);
+    }
+
+    private static final EndpointSnitchInfo instance = new EndpointSnitchInfo();
+
+    public static EndpointSnitchInfo getInstance() {
+        return instance;
+    }
+
+    private EndpointSnitchInfo() {
+        MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
+        try {
+            mbs.registerMBean(new EndpointSnitchInfo(), new ObjectName(
+                    "org.apache.cassandra.db:type=EndpointSnitchInfo"));
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    /**
+     * Provides the Rack name depending on the respective snitch used, given the
+     * host name/ip
+     * 
+     * @param host
+     * @throws UnknownHostException
+     */
+    public String getRack(String host) throws UnknownHostException {
+        log(" getRack(String host) throws UnknownHostException");
+        return c.getStringValue("");
+    }
+
+    /**
+     * Provides the Datacenter name depending on the respective snitch used,
+     * given the hostname/ip
+     * 
+     * @param host
+     * @throws UnknownHostException
+     */
+    public String getDatacenter(String host) throws UnknownHostException {
+        log(" getDatacenter(String host) throws UnknownHostException");
+        return c.getStringValue("");
+    }
+
+    /**
+     * Provides the snitch name of the cluster
+     * 
+     * @return Snitch name
+     */
+    public String getSnitchName() {
+        log(" getSnitchName()");
+        return c.getStringValue("");
+    }
+
+}

src/main/java/org/apache/cassandra/locator/EndpointSnitchInfoMBean.java

@@ -0,0 +1,48 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 org.apache.cassandra.locator;
+
+import java.net.UnknownHostException;
+
+/**
+ * MBean exposing standard Snitch info
+ */
+public interface EndpointSnitchInfoMBean
+{
+    /**
+     * Provides the Rack name depending on the respective snitch used, given the host name/ip
+     * @param host
+     * @throws UnknownHostException
+     */
+    public String getRack(String host) throws UnknownHostException;
+
+    /**
+     * Provides the Datacenter name depending on the respective snitch used, given the hostname/ip
+     * @param host
+     * @throws UnknownHostException
+     */
+    public String getDatacenter(String host) throws UnknownHostException;
+
+
+    /**
+     * Provides the snitch name of the cluster
+     * @return Snitch name
+     */
+    public String getSnitchName();
+
+}