diff --git a/src/main/java/com/cloudius/main/Main.java b/src/main/java/com/cloudius/main/Main.java index 2404cc8..1b4994d 100644 --- a/src/main/java/com/cloudius/main/Main.java +++ b/src/main/java/com/cloudius/main/Main.java @@ -4,6 +4,7 @@ package com.cloudius.main; import com.cloudius.api.APIClient; +import org.apache.cassandra.net.MessagingService; import org.apache.cassandra.service.StorageService; public class Main { @@ -12,6 +13,7 @@ public class Main { System.out.println("Connecting to " + APIClient.getBaseUrl()); System.out.println("Starting the JMX server"); StorageService.getInstance(); + MessagingService.getInstance(); Thread.sleep(Long.MAX_VALUE); } diff --git a/src/main/java/org/apache/cassandra/net/MessagingService.java b/src/main/java/org/apache/cassandra/net/MessagingService.java new file mode 100644 index 0000000..e758487 --- /dev/null +++ b/src/main/java/org/apache/cassandra/net/MessagingService.java @@ -0,0 +1,161 @@ +/* + * 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. + */ +/* + * Copyright 2015 Cloudius Systems + * + * Modified by Cloudius Systems + */ +package org.apache.cassandra.net; + +import java.lang.management.ManagementFactory; +import java.net.*; +import java.util.*; + +import javax.management.MBeanServer; +import javax.management.ObjectName; + +import com.cloudius.api.APIClient; + +public final class MessagingService implements MessagingServiceMBean { + public static final String MBEAN_NAME = "org.apache.cassandra.net:type=MessagingService"; + private static final java.util.logging.Logger logger = java.util.logging.Logger + .getLogger(MessagingService.class.getName()); + + private APIClient c = new APIClient(); + + private final ObjectName jmxObjectName; + + public void log(String str) { + System.out.println(str); + logger.info(str); + } + + public MessagingService() { + MBeanServer mbs = ManagementFactory.getPlatformMBeanServer(); + try { + jmxObjectName = new ObjectName(MBEAN_NAME); + mbs.registerMBean(this, jmxObjectName); + // mbs.registerMBean(StreamManager.instance, new ObjectName( + // StreamManager.OBJECT_NAME)); + } catch (Exception e) { + throw new RuntimeException(e); + } + } + + static MessagingService instance = new MessagingService(); + + public static MessagingService getInstance() { + return instance; + } + + /** + * Pending tasks for Command(Mutations, Read etc) TCP Connections + */ + public Map getCommandPendingTasks() { + log(" getCommandPendingTasks()"); + return c.getMapStringIntegerValue("/messaging_service/messages/pending"); + } + + /** + * Completed tasks for Command(Mutations, Read etc) TCP Connections + */ + public Map getCommandCompletedTasks() { + System.out.println("getCommandCompletedTasks!"); + Map res = c + .getListMapStringLongValue("/messaging_service/messages/sent"); + return res; + } + + /** + * Dropped tasks for Command(Mutations, Read etc) TCP Connections + */ + public Map getCommandDroppedTasks() { + log(" getCommandDroppedTasks()"); + return c.getMapStringLongValue(""); + } + + /** + * Pending tasks for Response(GOSSIP & RESPONSE) TCP Connections + */ + public Map getResponsePendingTasks() { + log(" getResponsePendingTasks()"); + return c.getMapStringIntegerValue(""); + } + + /** + * Completed tasks for Response(GOSSIP & RESPONSE) TCP Connections + */ + public Map getResponseCompletedTasks() { + log(" getResponseCompletedTasks()"); + return c.getMapStringLongValue(""); + } + + /** + * dropped message counts for server lifetime + */ + public Map getDroppedMessages() { + log(" getDroppedMessages()"); + return c.getMapStringIntegerValue(""); + } + + /** + * dropped message counts since last called + */ + public Map getRecentlyDroppedMessages() { + log(" getRecentlyDroppedMessages()"); + return c.getMapStringIntegerValue(""); + } + + /** + * Total number of timeouts happened on this node + */ + public long getTotalTimeouts() { + log(" getTotalTimeouts()"); + return c.getLongValue(""); + } + + /** + * Number of timeouts per host + */ + public Map getTimeoutsPerHost() { + log(" getTimeoutsPerHost()"); + return c.getMapStringLongValue(""); + } + + /** + * Number of timeouts since last check. + */ + public long getRecentTotalTimouts() { + log(" getRecentTotalTimouts()"); + return c.getLongValue(""); + } + + /** + * Number of timeouts since last check per host. + */ + public Map getRecentTimeoutsPerHost() { + log(" getRecentTimeoutsPerHost()"); + return c.getMapStringLongValue(""); + } + + public int getVersion(String address) throws UnknownHostException { + log(" getVersion(String address) throws UnknownHostException"); + return c.getIntValue(""); + } + +} diff --git a/src/main/java/org/apache/cassandra/net/MessagingServiceMBean.java b/src/main/java/org/apache/cassandra/net/MessagingServiceMBean.java new file mode 100644 index 0000000..3fbd5c1 --- /dev/null +++ b/src/main/java/org/apache/cassandra/net/MessagingServiceMBean.java @@ -0,0 +1,84 @@ +/* + * 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.net; + +import java.net.UnknownHostException; +import java.util.Map; + +/** + * MBean exposing MessagingService metrics. - OutboundConnectionPools - + * Command/Response - Pending/Completed Tasks + */ +public interface MessagingServiceMBean { + /** + * Pending tasks for Command(Mutations, Read etc) TCP Connections + */ + public Map getCommandPendingTasks(); + + /** + * Completed tasks for Command(Mutations, Read etc) TCP Connections + */ + public Map getCommandCompletedTasks(); + + /** + * Dropped tasks for Command(Mutations, Read etc) TCP Connections + */ + public Map getCommandDroppedTasks(); + + /** + * Pending tasks for Response(GOSSIP & RESPONSE) TCP Connections + */ + public Map getResponsePendingTasks(); + + /** + * Completed tasks for Response(GOSSIP & RESPONSE) TCP Connections + */ + public Map getResponseCompletedTasks(); + + /** + * dropped message counts for server lifetime + */ + public Map getDroppedMessages(); + + /** + * dropped message counts since last called + */ + public Map getRecentlyDroppedMessages(); + + /** + * Total number of timeouts happened on this node + */ + public long getTotalTimeouts(); + + /** + * Number of timeouts per host + */ + public Map getTimeoutsPerHost(); + + /** + * Number of timeouts since last check. + */ + public long getRecentTotalTimouts(); + + /** + * Number of timeouts since last check per host. + */ + public Map getRecentTimeoutsPerHost(); + + public int getVersion(String address) throws UnknownHostException; +}