uima-ducc-web/src/main/java/org/apache/uima/ducc/ws/helper/JmxHelper.java

/*
 * 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.helper;

import java.io.IOException;

import javax.management.MBeanServerConnection;
import javax.management.ObjectName;
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.id.DuccId;

public abstract class JmxHelper {
	
	private static DuccLogger logger = DuccLoggerComponents.getWsLogger(JmxHelper.class.getName());
	private static DuccId jobid = null;
	
	private String jmxHost = "localhost";
	private int jmxPort = -1;
	
	private JMXServiceURL url;
	private JMXConnector jmxc;
	private MBeanServerConnection mbsc;
	
	protected void setJmxHost(String value) {
		jmxHost = value;
	}
	
	public String getJmxHost() {
		return jmxHost;
	}
	
	protected void setJmxPort(int value) {
		jmxPort = value;
	}
	
	public int getJmxPort() {
		return jmxPort;
	}
	
	public String getJmxUrl() {
		return "service:jmx:rmi:///jndi/rmi://"+getJmxHost()+":"+getJmxPort()+"/jmxrmi";
	}
	
	public MBeanServerConnection getMBSC() {
		return mbsc;
	}
	
	protected void jmxConnect() throws IOException {
		url = new JMXServiceURL(getJmxUrl());
		jmxc = JMXConnectorFactory.connect(url, null);
		mbsc = jmxc.getMBeanServerConnection();
	}
	
	protected String getJmxData() throws Exception {
		Object o = null;
		MBeanServerConnection mbsc = null;
		try {
			mbsc = getMBSC();
			o = mbsc.getAttribute(new ObjectName("java.lang:type=Runtime"), "Name");
		} 
		catch(Exception e) {
			reconnect();
			mbsc = getMBSC();
			o = mbsc.getAttribute(new ObjectName("java.lang:type=Runtime"), "Name");
		}
		String data = (String) o;
		return data;
	}
	
	public Long getPID() {
		String location = "getPID";
		Long retVal = new Long(0);
		try {
			String data = getJmxData();
			String[] address = data.split("@");
			Long pid = Long.parseLong(address[0]);
			retVal = pid;
		}
		catch(Exception e) {
			logger.error(location, jobid, e);
		}
		return retVal;
	}

	protected abstract void reconnect();
}