org.hyperic.hq.plugin.weblogic.WeblogicServiceControlPlugin.java Source code

Introduction

Here is the source code for org.hyperic.hq.plugin.weblogic.WeblogicServiceControlPlugin.java

Source

/**
 * NOTE: This copyright does *not* cover user programs that use HQ
 * program services by normal system calls through the application
 * program interfaces provided as part of the Hyperic Plug-in Development
 * Kit or the Hyperic Client Development Kit - this is merely considered
 * normal use of the program, and does *not* fall under the heading of
 *  "derived work".
 *
 *  Copyright (C) [2009-2010], VMware, Inc.
 *  This file is part of HQ.
 *
 *  HQ is free software; you can redistribute it and/or modify
 *  it under the terms version 2 of the GNU General Public License as
 *  published by the Free Software Foundation. This program is distributed
 *  in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 *  even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 *  PARTICULAR PURPOSE. See the GNU General Public License for more
 *  details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 *  USA.
 *
 */

package org.hyperic.hq.plugin.weblogic;

import java.io.IOException;

import javax.management.Attribute;
import javax.management.AttributeNotFoundException;
import javax.management.MBeanInfo;
import javax.management.MBeanServerConnection;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;
import javax.management.remote.JMXConnector;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hyperic.hq.product.ControlPlugin;
import org.hyperic.hq.product.Metric;
import org.hyperic.hq.product.MetricNotFoundException;
import org.hyperic.hq.product.MetricUnreachableException;
import org.hyperic.hq.product.PluginException;
import org.hyperic.hq.product.jmx.MBeanUtil;

/**
 * Control plugin for dynamically discovered services in WebLogic. Makes JMX
 * invoke/getAttribute/setAttribute calls using the managed server's
 * RemoteMBeanServer, obtained through JNDI lookup
 * @author Jennifer Hickey
 * 
 */
public class WeblogicServiceControlPlugin extends ControlPlugin {
    private static final Log log = LogFactory.getLog(WeblogicServiceControlPlugin.class);

    public void doAction(String action, String[] args) throws PluginException {
        invokeMethod(action, args);
    }

    protected Object invoke(MBeanServerConnection mServer, String objectName, String method, Object[] args,
            String[] sig) throws MetricUnreachableException, MetricNotFoundException, PluginException {

        ObjectName obj;
        try {
            obj = new ObjectName(objectName);
        } catch (MalformedObjectNameException e1) {
            throw new PluginException("Unable to create an ObjectName from " + objectName);
        }

        MBeanInfo info;
        try {
            info = mServer.getMBeanInfo(obj);
        } catch (Exception e1) {
            throw new PluginException("Unable to obtain MBeanInfo from " + objectName);
        }

        if (sig.length == 0) {
            MBeanUtil.OperationParams params = MBeanUtil.getOperationParams(info, method, args);
            if (params.isAttribute) {
                if (method.startsWith("set")) {
                    try {
                        mServer.setAttribute(obj, new Attribute(method.substring(3), params.arguments[0]));
                    } catch (AttributeNotFoundException e) {
                        throw new MetricNotFoundException(e.getMessage(), e);
                    } catch (Exception e) {
                        throw new PluginException(e);
                    }
                    return null;
                } else {
                    try {
                        return mServer.getAttribute(obj, method.substring(3));
                    } catch (AttributeNotFoundException e) {
                        throw new MetricNotFoundException(e.getMessage(), e);
                    } catch (Exception e) {
                        throw new PluginException(e);
                    }
                }
            }
            sig = params.signature;
            args = params.arguments;
        }

        try {
            return mServer.invoke(obj, method, args, sig);
        } catch (Exception e) {
            throw new PluginException(e);
        }
    }

    protected void invokeMethod(ControlPlugin plugin, String objectName, String action, String[] args) {
        log.debug("invoking " + action + " " + MBeanUtil.anyToString(args));
        JMXConnector jmxConnector;
        try {
            jmxConnector = WeblogicUtil.getManagedServerConnection(plugin.getConfig().toProperties());
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            plugin.setMessage(e.getMessage());
            plugin.setResult(RESULT_FAILURE);
            return;
        }
        try {
            String result = null;
            Object obj = invoke(jmxConnector.getMBeanServerConnection(), objectName, action, args, new String[0]);
            if (obj != null) {
                result = MBeanUtil.anyToString(obj);
            }
            log.debug(objectName + "." + action + "() returned: " + obj);
            plugin.setResult(RESULT_SUCCESS);
            if (result != null) {
                plugin.setMessage(result);
            }
        } catch (PluginException e) {
            log.error(e.getMessage(), e);
            plugin.setMessage(e.getMessage());
            plugin.setResult(RESULT_FAILURE);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            // anything not explicitly thrown by invoke
            // needs to have the full stack trace logged for debugging.
            plugin.setMessage(e.getMessage());
            plugin.setResult(RESULT_FAILURE);
        } finally {
            try {
                jmxConnector.close();
            } catch (IOException e) {
                log.warn("Error closing remote JMX connection", e);
            }
        }
    }

    protected void invokeMethod(String action, String[] args) {
        String objectName = getTypeProperty("OBJECT_NAME");
        objectName = Metric.translate(objectName, getConfig());
        invokeMethod(this, objectName, action, args);
    }

}