diff --git a/src/main/java/com/cloudius/urchin/main/Main.java b/src/main/java/com/cloudius/urchin/main/Main.java index 11c2663..6730b3b 100644 --- a/src/main/java/com/cloudius/urchin/main/Main.java +++ b/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); } diff --git a/src/main/java/org/apache/cassandra/locator/EndpointSnitchInfo.java b/src/main/java/org/apache/cassandra/locator/EndpointSnitchInfo.java new file mode 100644 index 0000000..5050feb --- /dev/null +++ b/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(""); + } + +} diff --git a/src/main/java/org/apache/cassandra/locator/EndpointSnitchInfoMBean.java b/src/main/java/org/apache/cassandra/locator/EndpointSnitchInfoMBean.java new file mode 100644 index 0000000..84d2499 --- /dev/null +++ b/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(); + +}