All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.uima.ducc.ws.broker.BrokerHelper Maven / Gradle / Ivy
/*
* 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.uima.ducc.ws.broker;
import java.lang.management.ManagementFactory;
import java.lang.management.OperatingSystemMXBean;
import java.lang.management.ThreadMXBean;
import java.util.ArrayList;
import java.util.Set;
import java.util.TreeMap;
import java.util.TreeSet;
import javax.management.Attribute;
import javax.management.AttributeList;
import javax.management.MBeanServerConnection;
import javax.management.ObjectName;
import javax.management.openmbean.CompositeData;
import javax.management.remote.JMXConnector;
import javax.management.remote.JMXConnectorFactory;
import javax.management.remote.JMXServiceURL;
import org.apache.uima.ducc.common.utils.DuccLogger;
import org.apache.uima.ducc.common.utils.DuccLoggerComponents;
import org.apache.uima.ducc.common.utils.DuccPropertiesResolver;
import org.apache.uima.ducc.common.utils.id.DuccId;
public class BrokerHelper {
private static DuccLogger duccLogger = DuccLoggerComponents.getWsLogger(BrokerHelper.class.getName());
private static DuccId jobid = null;
private static BrokerHelper instance = new BrokerHelper();
public static BrokerHelper getInstance() {
return instance;
}
public enum BrokerAttribute { BrokerVersion, MemoryPercentUsage, Uptime };
public enum FrameworkAttribute { ConsumerCount, QueueSize, MaxEnqueueTime, AverageEnqueueTime, MemoryPercentUsage };
private String host = "?";
private int port = 1100;
private JMXServiceURL url;
private JMXConnector jmxc;
private MBeanServerConnection mbsc;
private OperatingSystemMXBean remoteOperatingSystem;
private ThreadMXBean remoteThread;
private BrokerHelper() {
init();
}
private void init() {
String location = "init";
try {
DuccPropertiesResolver duccPropertiesResolver = DuccPropertiesResolver.getInstance();
String key;
String value;
key = "ducc.broker.hostname";
value = duccPropertiesResolver.getCachedProperty(key);
setHost(value);
key = "ducc.broker.jmx.port";
value = duccPropertiesResolver.getCachedProperty(key);
setPort(value);
url = new JMXServiceURL("service:jmx:rmi:///jndi/rmi://"+getHost()+":"+getPort()+"/jmxrmi");
jmxc = JMXConnectorFactory.connect(url, null);
mbsc = jmxc.getMBeanServerConnection();
remoteOperatingSystem =
ManagementFactory.newPlatformMXBeanProxy(
mbsc,
ManagementFactory.OPERATING_SYSTEM_MXBEAN_NAME,
OperatingSystemMXBean.class);
remoteThread =
ManagementFactory.newPlatformMXBeanProxy(
mbsc,
ManagementFactory.THREAD_MXBEAN_NAME,
ThreadMXBean.class);
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
}
private void setHost(String value) {
host = value;
}
public String getHost() {
return host;
}
private void setPort(String value) {
String location = "setPort";
try {
port = Integer.parseInt(value);
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
}
public int getPort() {
return port;
}
// Memory Info //
public Long getMemoryUsed() {
String location = "getMemoryUsed";
Long retVal = new Long(0);
try {
Object o = mbsc.getAttribute(new ObjectName("java.lang:type=Memory"), "HeapMemoryUsage");
CompositeData cd = (CompositeData) o;
retVal = (Long) cd.get("used");
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
public Long getMemoryMax() {
String location = "getMemoryMax";
Long retVal = new Long(0);
try {
Object o = mbsc.getAttribute(new ObjectName("java.lang:type=Memory"), "HeapMemoryUsage");
CompositeData cd = (CompositeData) o;
retVal = (Long) cd.get("max");
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
// Threads Info //
public int getThreadsLive() {
String location = "getThreadsLive";
int retVal = 0;
try {
retVal = remoteThread.getThreadCount();
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
public int getThreadsPeak() {
String location = "getThreadsPeak";
int retVal = 0;
try {
retVal = remoteThread.getPeakThreadCount();
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
// Operating System Info //
public double getSystemLoadAverage() {
String location = "getSystemLoadAverage";
double retVal = 0;
try {
retVal = remoteOperatingSystem.getSystemLoadAverage();
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
/////
private boolean isFrameworkEntity(ObjectName objectName) {
String location = "isFrameworkEntity";
boolean retVal = false;
String key = "Destination";
String value = objectName.getKeyProperty(key);
if(value != null) {
if(value.startsWith("ducc.")) {
retVal = true;
duccLogger.debug(location, jobid, key+"="+value);
}
}
return retVal;
}
private String getName(ObjectName objectName) {
String retVal = "";
String key = "Destination";
String value = objectName.getKeyProperty(key);
retVal = value;
return retVal;
}
private String getType(ObjectName objectName) {
String retVal = "";
String key = "Type";
String value = objectName.getKeyProperty(key);
retVal = value;
return retVal;
}
private boolean isBrokerInfo(ObjectName objectName) {
boolean retVal = false;
String key = "Type";
String value = objectName.getKeyProperty(key);
if(value != null) {
if(value.startsWith("Broker")) {
retVal = true;
}
}
return retVal;
}
public String getAttribute(BrokerAttribute qa) {
String location = "getAttribute";
String retVal = "";
try {
Set objectNames = new TreeSet(mbsc.queryNames(null, null));
for (ObjectName objectName : objectNames) {
if(isBrokerInfo(objectName)) {
String[] attrNames = { qa.name() };
AttributeList attributeList = mbsc.getAttributes(objectName, attrNames);
for(Object object : attributeList) {
Attribute attribute = (Attribute) object;
retVal = ""+attribute.getValue();
}
break;
}
}
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
public TreeMap getAttributes(String name, String[] attrNames) {
String location = "getAttributes";
TreeMap retVal = new TreeMap();
try {
Set objectNames = new TreeSet(mbsc.queryNames(null, null));
for (ObjectName objectName : objectNames) {
String topicName = getName(objectName);
if(topicName != null) {
if(topicName.equals(name)) {
AttributeList attributeList = mbsc.getAttributes(objectName, attrNames);
for(Object object : attributeList) {
Attribute attribute = (Attribute) object;
String attrName = attribute.getName();
String attrValue = ""+attribute.getValue();
retVal.put(attrName, attrValue);
duccLogger.debug(location, jobid, attrName+"="+attrValue);
}
break;
}
}
}
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
public ArrayList getFrameworkEntities() {
String location = "getFrameworkTopicNames";
ArrayList list = new ArrayList();
ArrayList retVal = new ArrayList();
try {
Set objectNames = new TreeSet(mbsc.queryNames(null, null));
for (ObjectName objectName : objectNames) {
if(isFrameworkEntity(objectName)) {
String name = getName(objectName);
String type = getType(objectName);
EntityInfo entityInfo = new EntityInfo(name,type);
String key = entityInfo.getKey();
if(!list.contains(key)) {
retVal.add(entityInfo);
}
}
}
}
catch(Exception e) {
duccLogger.error(location, jobid, e);
}
return retVal;
}
}
