org.quartz.jobs.ee.jmx.JMXInvokerJob.java Source code

Java tutorial

Introduction

Here is the source code for org.quartz.jobs.ee.jmx.JMXInvokerJob.java

Source

/* 
 * Copyright 2001-2009 James House 
 * 
 * Licensed 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.
 * 
 */

/*
 * Previously Copyright (c) 2001-2004 James House
 */
package org.quartz.jobs.ee.jmx;

import java.util.LinkedList;
import java.util.StringTokenizer;

import javax.management.MBeanServer;
import javax.management.MBeanServerFactory;
import javax.management.ObjectName;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.quartz.Job;
import org.quartz.JobDataMap;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;

/**
 * Generic JMX invoker Job.  It supports any number or type of parameters
 * to the JMX bean.<p>
 * 
 * The required parameters are as follows (case doesn't matter):<p>
 * <dl>
 * <dt><strong>JMX_OBJECTNAME</strong>
 * <dd>This is the fully qualifed name of the object (ie in JBoss to lookup
 * the log4j jmx bean you would specify "jboss.system:type=Log4jService,service=Logging"
 * <dt><strong>JMX_METHOD</strong>
 * <dd>This is the method to invoke on the specified JMX Bean. (ie in JBoss to
 * change the log level you would specify "setLoggerLevel"
 * <dt><strong>JMX_PARAMDEFS</strong>
 * <dd>This is a definition of the parameters to be passed to the specified method
 * and their corresponding java types.  Each parameter definition is comma seperated
 * and has the following parts: <type>:<name>.  Type is the java type for the parameter.  
 * The following types are supported:<p>
 * <b>i</b> - is for int<p>
 * <b>l</b> - is for long<p>
 * <b>f</b> - is for float<p>
 * <b>d</b> - is for double<p>
 * <b>s</b> - is for String<p>
 * <b>b</b> - is for boolean<p>
 * For ilfdb use lower for native type and upper for object wrapper. The name portion
 * of the definition is the name of the parameter holding the string value. (ie
 * s:fname,s:lname would require 2 parameters of the name fname and lname and
 * would be passed in that order to the method.
 * 
 * @author James Nelson (jmn@provident-solutions.com) -- Provident Solutions LLC
 * 
 */
public class JMXInvokerJob implements Job {

    private final Log log = LogFactory.getLog(getClass());

    public void execute(JobExecutionContext context) throws JobExecutionException {
        try {
            Object[] params = null;
            String[] types = null;
            String objName = null;
            String objMethod = null;

            JobDataMap jobDataMap = context.getMergedJobDataMap();

            String[] keys = jobDataMap.getKeys();
            for (int i = 0; i < keys.length; i++) {
                String value = jobDataMap.getString(keys[i]);
                if ("JMX_OBJECTNAME".equalsIgnoreCase(keys[i])) {
                    objName = value;
                } else if ("JMX_METHOD".equalsIgnoreCase(keys[i])) {
                    objMethod = value;
                } else if ("JMX_PARAMDEFS".equalsIgnoreCase(keys[i])) {
                    String[] paramdefs = split(value, ",");
                    params = new Object[paramdefs.length];
                    types = new String[paramdefs.length];
                    for (int k = 0; k < paramdefs.length; k++) {
                        String parts[] = split(paramdefs[k], ":");
                        if (parts.length < 2) {
                            throw new Exception(
                                    "Invalid parameter definition: required parts missing " + paramdefs[k]);
                        }
                        switch (parts[0].charAt(0)) {
                        case 'i':
                            params[k] = new Integer(jobDataMap.getString(parts[1]));
                            types[k] = Integer.TYPE.getName();
                            break;
                        case 'I':
                            params[k] = new Integer(jobDataMap.getString(parts[1]));
                            types[k] = Integer.class.getName();
                            break;
                        case 'l':
                            params[k] = new Long(jobDataMap.getString(parts[1]));
                            types[k] = Long.TYPE.getName();
                            break;
                        case 'L':
                            params[k] = new Long(jobDataMap.getString(parts[1]));
                            types[k] = Long.class.getName();
                            break;
                        case 'f':
                            params[k] = new Float(jobDataMap.getString(parts[1]));
                            types[k] = Float.TYPE.getName();
                            break;
                        case 'F':
                            params[k] = new Float(jobDataMap.getString(parts[1]));
                            types[k] = Float.class.getName();
                            break;
                        case 'd':
                            params[k] = new Double(jobDataMap.getString(parts[1]));
                            types[k] = Double.TYPE.getName();
                            break;
                        case 'D':
                            params[k] = new Double(jobDataMap.getString(parts[1]));
                            types[k] = Double.class.getName();
                            break;
                        case 's':
                            params[k] = jobDataMap.getString(parts[1]);
                            types[k] = String.class.getName();
                            break;
                        case 'b':
                            params[k] = new Boolean(jobDataMap.getString(parts[1]));
                            types[k] = Boolean.TYPE.getName();
                            break;
                        case 'B':
                            params[k] = new Boolean(jobDataMap.getString(parts[1]));
                            types[k] = Boolean.class.getName();
                            break;
                        }
                    }
                }
            }

            if (objName == null || objMethod == null) {
                throw new Exception("Required parameters missing");
            }

            context.setResult(invoke(objName, objMethod, params, types));
        } catch (Exception e) {
            String m = "Caught a " + e.getClass().getName() + " exception : " + e.getMessage();
            getLog().error(m, e);
            throw new JobExecutionException(m, e, false);
        }
    }

    private String[] split(String str, String splitStr) // Same as String.split(.) in JDK 1.4
    {
        LinkedList l = new LinkedList();

        StringTokenizer strTok = new StringTokenizer(str, splitStr);
        while (strTok.hasMoreTokens()) {
            String tok = strTok.nextToken();
            l.add(tok);
        }

        return (String[]) l.toArray(new String[l.size()]);
    }

    private Object invoke(String objectName, String method, Object[] params, String[] types) throws Exception {
        MBeanServer server = (MBeanServer) MBeanServerFactory.findMBeanServer(null).get(0);
        ObjectName mbean = new ObjectName(objectName);

        if (server == null) {
            throw new Exception("Can't find mbean server");
        }

        getLog().info("invoking " + method);
        return server.invoke(mbean, method, params, types);
    }

    protected Log getLog() {
        return log;
    }

}