Java tutorial
/** * Helios, OpenSource Monitoring * Brought to you by the Helios Development Group * * Copyright 2007, Helios Development Group and individual contributors * as indicated by the @author tags. See the copyright.txt file in the * distribution for a full listing of individual contributors. * * This is free software; you can redistribute it and/or modify it * under the terms of the GNU Lesser General Public License as * published by the Free Software Foundation; either version 2.1 of * the License, or (at your option) any later version. * * This software 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this software; if not, write to the Free * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA * 02110-1301 USA, or see the FSF site: http://www.fsf.org. * */ package org.helios.jzab.agent.commands.impl.jmx; import java.util.Collections; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.Set; import java.util.regex.Matcher; import java.util.regex.Pattern; import javax.management.MBeanServerConnection; import javax.management.ObjectName; import javax.management.remote.JMXServiceURL; import org.helios.jzab.util.JMXHelper; import org.json.JSONObject; /** * <p>Title: JMXDiscoveryCommandProcessor</p> * <p>Description: JMX discovery command processor. Receives requests with ObjectName patterns, locates all matching instances, and returns the extracted values.</p> * <p>Company: Helios Development Group LLC</p> * @author Whitehead (nwhitehead AT heliosdev DOT org) * <p><code>org.helios.jzab.agent.commands.impl.jmx.JMXDiscoveryCommandProcessor</code></p> */ public class JMXDiscoveryCommandProcessor extends BaseJMXCommandProcessor { /** This processors command keys */ public static final String COMMAND_KEY = "jmxd"; /** The regex pattern to extract tokens */ // public static final Pattern TOKEN = Pattern.compile("(\\[\\{\\$(.*?)\\}\\])"); public static final Pattern TOKEN = Pattern.compile("(\\{#(.*?)\\})"); /** * {@inheritDoc} * @see org.helios.jzab.agent.commands.ICommandProcessor#getLocatorKey() */ @Override public String getLocatorKey() { return COMMAND_KEY; } /** * Var parameters:<ol> * <li><b>JMX Object Name</b>: The ObjectName pattern query.</li> * <li><b>Domain</b>: (Optional) Defines the MBeanServer domain in which the target MBeans are registered. Can also be interpreted as a {@link JMXServiceURL} in which case a remote connection will be used to retrieve the attribute values.</li> * </ol> * {@inheritDoc} * @see org.helios.jzab.agent.commands.AbstractCommandProcessor#doExecute(java.lang.String, java.lang.String[]) */ @Override protected Object doExecute(String commandName, String... args) throws Exception { if (commandName == null || commandName.trim().isEmpty()) throw new IllegalArgumentException("Null or empty command name", new Throwable()); if (args == null || args.length < 1) throw new IllegalArgumentException("Invalid argument count for command [" + commandName + "] with args [" + (args == null ? 0 : args.length) + "]", new Throwable()); commandName = commandName.trim(); Map<String, String> tokens = new HashMap<String, String>(); ObjectName objectName = null; try { objectName = extractTokens(args[0].trim(), tokens); } catch (Exception e) { log.error("Invalid ObjectName Requested [{}], Error:[{}]", args[0], e.getMessage()); throw new RuntimeException(e); } String domain = null; if (args.length > 3) { domain = args[3]; } MBeanServerConnection server = null; Set<JSONObject> results = new HashSet<JSONObject>(); try { server = getServerForDomain(domain); //return JMXHelper.getAttribute(server, compoundDelimiter, on, attrName); for (ObjectName on : server.queryNames(objectName, null)) { for (Map.Entry<String, String> entry : tokens.entrySet()) { String resolvedValue = resolveValue(on.toString(), entry.getKey(), args[0].trim()); log.debug("Resolved Value [{}] for Token [{}]", resolvedValue, entry.getKey()); //result.append("\n").append(entry.getValue()).append("--->").append(resolvedValue); results.add(new JSONObject(Collections.singletonMap(entry.getKey(), resolvedValue))); } } return results.toArray(new JSONObject[results.size()]); } catch (Exception e) { log.debug("Failed to get MBeanServerConnection for domain [{}]", domain, e); log.error("Failed to get MBeanServerConnection for domain [{}]", domain); throw new RuntimeException(e); } } /** * Resolves a discovery token * @param objectName The JMX ObjectName string value containing the subst tokens * @param token The token to replace * @param original The token replacing value * @return the resolved and replaced ObjectName string */ protected String resolveValue(String objectName, String token, String original) { StringBuilder b = new StringBuilder(original); int tokLen = token.length(); int startIndex = original.indexOf(token); b.delete(startIndex, startIndex + tokLen); String ePart = b.toString().substring(startIndex); String replacement = null; if (ePart.isEmpty()) { replacement = objectName.substring(startIndex); } else { replacement = objectName.substring(startIndex, objectName.indexOf(ePart)); } return replacement; } /** * {@inheritDoc} * @see org.helios.jzab.agent.commands.AbstractCommandProcessor#isDiscovery() */ @Override public boolean isDiscovery() { return true; } // public static void main(String[] args) { // log("JMX Discovery Test"); // JMXDiscoveryCommandProcessor processor = new JMXDiscoveryCommandProcessor(); // log(processor.execute("java.lang:type=GarbageCollector,name={#GCName}")); // } // // public static void log(Object msg) { // System.out.println(msg); // } /** * Extracts the discovery tokens into a map * @param on The ObjectName to extract from * @param tokenMap The token map * @return a map of tokens */ protected ObjectName extractTokens(String on, final Map<String, String> tokenMap) { Matcher m = TOKEN.matcher(on.toString()); String str = on.toString(); while (m.find()) { String key = m.group(1); tokenMap.put(key, m.group(2)); str = str.replace(key, "*"); } return JMXHelper.objectName(str); } }