BrokerAdmin.java

package com.trendmicro.tme.broker;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.management.Attribute;
import javax.management.AttributeList;
import javax.management.MBeanAttributeInfo;
import javax.management.MBeanInfo;
import javax.management.MBeanServerConnection;
import javax.management.ObjectName;
import javax.management.QueryExp;
import javax.management.remote.JMXConnector;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.sun.messaging.AdminConnectionConfiguration;
import com.sun.messaging.AdminConnectionFactory;
import com.trendmicro.tme.mfr.Exchange;

public class BrokerAdmin {
    private static Logger logger = LoggerFactory.getLogger(BrokerAdmin.class);
    private String brokerHost;
    private JMXConnector connector;
    private MBeanServerConnection connection;

    ////////////////////////////////////////////////////////////////////////////////
    
	public BrokerAdmin(String host) {
	    brokerHost = host;
	}

    public String getExchangeAttribs(boolean isQueue, String exchangeName, String attrib) {
        String pattern = String.format("com.sun.messaging.jms.server:type=Destination,subtype=Monitor,desttype=%s,name=\"%s\"", isQueue ? "q": "t", exchangeName);
        try {
            jmxConnectServer();
            Map<String, String> map = getMBeanAttributesMap(pattern, null);
            return map.get(attrib);
        }
        catch(Exception e) {
            logger.error(e.getMessage(),e );
        }
        finally {
            jmxCloseServer();
        }
        return null;
    }
    
    public Map<String,String> getExchangeAttribMap(Exchange exchange) {
        String pattern = String.format("com.sun.messaging.jms.server:type=Destination,subtype=Monitor,desttype=%s,name=\"%s\"", exchange.isQueue() ? "q": "t", exchange.getName());
        return getMBeanAttributesMap(pattern, null);
    }
    
    public void setExchangeAttrib(Exchange exchange, String attribName, Object attribValue) {
        try {
            String pattern = String.format("com.sun.messaging.jms.server:type=Destination,subtype=Config,desttype=%s,name=\"%s\"", exchange.isQueue() ? "q": "t", exchange.getName());
            ObjectName name = connection.queryNames(new ObjectName(pattern), null).iterator().next();
            connection.setAttribute(name, new Attribute(attribName, attribValue));
        }
        catch(Exception e) {
            logger.error(e.getMessage(), e);
        }
    }
	
	public void jmxConnectServer() throws Exception {
        Exception e = null;
        for(int i = 0; i <= 60; i++) {
            try {
                AdminConnectionFactory acf;
                acf = new AdminConnectionFactory();
                acf.setProperty(AdminConnectionConfiguration.imqAddress, brokerHost);
                connector = acf.createConnection();
                connection = connector.getMBeanServerConnection();
                return;
            }
            catch(Exception e2) {
                logger.warn("jmx connect error: retry #" + i);
                e = e2;
                Thread.sleep(1000);
            }
        }
        logger.error("jmxConnectServer gave up retrying");
        throw e;
    }

    public void jmxCloseServer() {
        try {
            if(connector != null)
                connector.close();
        }
        catch(IOException e) {
            logger.error(e.getMessage(), e);
        }
    }

	public String[] getMBeanAttributeNames(ObjectName name) {
        try {
            MBeanInfo info = connection.getMBeanInfo(name);
            MBeanAttributeInfo[] attrinfo = info.getAttributes();
            String[] attrnames = new String[attrinfo.length];
            int i = 0;
            for(MBeanAttributeInfo a : attrinfo)
                attrnames[i++] = a.getName();
            return attrnames;
        }
        catch(Exception e) {
            logger.error(e.getMessage(), e);
        }
        return null;
    }

    public Map<String, AttributeList> getMBeanAttributes(String name, QueryExp query) {
        try {
            Set<ObjectName> names = connection.queryNames(new ObjectName(name), query);
            HashMap<String, AttributeList> map = new HashMap<String, AttributeList>();
            for(ObjectName n : names) {
                String objname = n.getCanonicalName();
                String[] attrnames = getMBeanAttributeNames(n);
                AttributeList attrs = connection.getAttributes(n, attrnames);
                map.put(objname, attrs);
            }
            return map;
        }
        catch(Exception e) {
            logger.error(e.getMessage(), e);
        }
        return null;
    }

    public Map<String, String> getMBeanAttributesMap(String name, QueryExp query) {
        Map<String, AttributeList> map = getMBeanAttributes(name, query);
        Map<String, String> result = new HashMap<String, String>();
        Set<String> keys = map.keySet();
        for(String k : keys) {
            List<Attribute> attrs = map.get(k).asList();
            for(Attribute a : attrs)
                result.put(a.getName(), (a.getValue() == null) ? "": a.getValue().toString());
        }
        return result;
    }

    public String getBrokerHost() {
        return brokerHost;
    }

    public ArrayList<Exchange> getAllExchangeMetadata() throws Exception{        
        try {
            jmxConnectServer();
            ArrayList<Exchange> result = new ArrayList<Exchange>();

            ObjectName objname = new ObjectName("com.sun.messaging.jms.server:type=Destination,subtype=Monitor,desttype=q,name=*");
            for(ObjectName n : connection.queryNames(objname, null)) {
                Exchange excQ = new Exchange(connection.getAttribute(n, "Name").toString());
                excQ.setQueue();
                result.add(excQ);
            }
            objname = new ObjectName("com.sun.messaging.jms.server:type=Destination,subtype=Monitor,desttype=t,name=*");
            for(ObjectName n : connection.queryNames(objname, null)) {
                Exchange excT = new Exchange(connection.getAttribute(n, "Name").toString());
                excT.setTopic();
                result.add(excT);
            }
            Exchange excAdmin = new Exchange("mq.sys.dmq");
            excAdmin.setQueue();
            result.remove(excAdmin);
            return result;
        } 
        catch(Exception e) {
            throw new Exception(e.getMessage());
        }
        finally {
            jmxCloseServer();
        }        
    }
    
    public Object invokeMBeanMethod(String name, String op_name, Object[] params) throws Exception {
        Object result = null;
        try {
            jmxConnectServer();
            Set<ObjectName> names = connection.queryNames(new ObjectName(name), null);
            for(ObjectName n : names) {
                result = connection.invoke(n, op_name, params, null);
                break;
            }            
        }catch(Exception e) {
            throw new Exception(e.getMessage());
        }
        finally {
            jmxCloseServer();
        }     

        return result;
    }
}