OGAgent.java revision 707
260N/A/*
260N/A * CDDL HEADER START
260N/A *
260N/A * The contents of this file are subject to the terms of the
260N/A * Common Development and Distribution License (the "License").
260N/A * You may not use this file except in compliance with the License.
260N/A *
260N/A * See LICENSE.txt included in this distribution for the specific
260N/A * language governing permissions and limitations under the License.
260N/A *
260N/A * When distributing Covered Code, include this CDDL HEADER in each
260N/A * file and include the License file at LICENSE.txt.
260N/A * If applicable, add the following below this CDDL HEADER, with the
260N/A * fields enclosed by brackets "[]" replaced with your own identifying
260N/A * information: Portions Copyright [yyyy] [name of copyright owner]
260N/A *
260N/A * CDDL HEADER END
260N/A */
260N/A
260N/A/*
260N/A * Copyright 2008 Sun Microsystems, Inc. All rights reserved.
260N/A * Use is subject to license terms.
260N/A */
260N/Apackage org.opensolaris.opengrok.management;
260N/A
260N/Aimport java.io.File;
260N/Aimport java.io.FileInputStream;
260N/Aimport java.io.IOException;
260N/Aimport java.io.InputStream;
260N/Aimport java.net.MalformedURLException;
260N/Aimport java.net.URI;
260N/Aimport java.net.URISyntaxException;
260N/Aimport java.util.ArrayList;
364N/Aimport java.util.Date;
260N/Aimport java.util.HashMap;
260N/Aimport java.util.Properties;
260N/Aimport java.util.logging.Level;
260N/Aimport java.util.logging.Logger;
465N/Aimport javax.management.JMException;
465N/Aimport javax.management.MBeanServer;
344N/Aimport javax.management.MBeanServerFactory;
465N/Aimport javax.management.NotificationFilter;
260N/Aimport javax.management.ObjectName;
260N/Aimport javax.management.remote.JMXConnectorServer;
260N/Aimport javax.management.remote.JMXServiceURL;
260N/Aimport javax.management.remote.jmxmp.JMXMPConnectorServer;
260N/Aimport javax.management.remote.rmi.RMIConnectorServer;
260N/Aimport javax.management.timer.Timer;
260N/Aimport org.opensolaris.opengrok.Info;
260N/Aimport org.opensolaris.opengrok.OpenGrokLogger;
260N/A
260N/A/**
260N/A * OG Agent main class.
260N/A * Class for starting the basic components:
260N/A * Monitor and JMX and HTTP Connectors.
260N/A * @author Jan S Berg
260N/A */
260N/Afinal public class OGAgent {
260N/A Properties props;
260N/A
260N/A private final static Logger log = Logger.getLogger("org.opensolaris.opengrok");
260N/A private MBeanServer server = null;
260N/A
260N/A
260N/A @SuppressWarnings("PMD.SystemPrintln")
260N/A private static boolean loadProperties(File file, InputStream in, Properties props) {
456N/A boolean ret = false;
260N/A InputStream stream = in;
260N/A try {
260N/A if (file != null) {
260N/A stream = new FileInputStream(file);
260N/A }
272N/A props.load(stream);
260N/A ret = true;
260N/A } catch (IOException e) {
260N/A System.err.println("Failed to read configuration");
260N/A e.printStackTrace(System.err);
260N/A ret = false;
894N/A } finally {
465N/A try {
1016N/A stream.close();
260N/A } catch (IOException e) {
260N/A System.err.println("Failed to close stream");
260N/A e.printStackTrace(System.err);
886N/A ret = false;
886N/A }
886N/A }
465N/A return ret;
260N/A }
465N/A
465N/A @SuppressWarnings("PMD.SystemPrintln")
465N/A public static void main(final String args[]) {
260N/A
955N/A Properties props = new Properties();
1026N/A // Load default values
260N/A boolean success = loadProperties(null, OGAgent.class.getResourceAsStream("oga.properties"), props);
260N/A
260N/A File file = new File("/etc/opengrok/opengrok.properties");
993N/A if (file.exists()) {
260N/A success = loadProperties(file, null, props);
260N/A }
993N/A
260N/A // System properties should override default properties
467N/A props.putAll(System.getProperties());
260N/A
260N/A // @todo Add support for longopts!!
260N/A for (int i = 0; success && i < args.length; i++) {
260N/A if (args[i].startsWith("--agent=")) {
260N/A props.setProperty("agent", args[i].substring("--agent=".length()));
260N/A } else if (args[i].startsWith("--config=")) {
260N/A file = new File(args[i].substring("--config=".length()));
260N/A if (file.exists()) {
260N/A success = loadProperties(file, null, props);
260N/A } else {
260N/A success = false;
260N/A System.err.println("Cannot load file \"" + file.getAbsolutePath() + "\": No such file");
260N/A }
260N/A }
260N/A }
260N/A
260N/A URI uri = null;
260N/A if (success) {
260N/A try {
260N/A uri = new URI(props.getProperty("agent"));
260N/A } catch (URISyntaxException ex) {
260N/A success = false;
260N/A System.err.println("Failed to decode agent url");
260N/A ex.printStackTrace(System.err);
260N/A }
260N/A }
260N/A
260N/A if (success) {
260N/A if (props.getProperty("org.opensolaris.opengrok.management.logging.path") == null) {
260N/A props.setProperty("org.opensolaris.opengrok.management.logging.path",
260N/A uri.getPath() + "/log");
260N/A }
260N/A
260N/A if (props.getProperty("org.opensolaris.opengrok.configuration.file") == null) {
260N/A props.setProperty("org.opensolaris.opengrok.configuration.file",
260N/A uri.getPath() + "/etc/configuration.xml");
260N/A }
260N/A
260N/A if (props.getProperty("org.opensolaris.opengrok.management.connection.host") == null) {
260N/A props.setProperty("org.opensolaris.opengrok.management.connection.host",
260N/A uri.getHost());
260N/A }
260N/A
260N/A if (props.getProperty("org.opensolaris.opengrok.management.connection.port") == null) {
260N/A props.setProperty("org.opensolaris.opengrok.management.connection.port",
260N/A Integer.toString(uri.getPort()));
260N/A }
260N/A
260N/A success = createLogger(props);
260N/A }
260N/A
260N/A if (success) {
260N/A
345N/A OGAgent oga = new OGAgent(props);
260N/A try {
260N/A oga.runOGA();
260N/A } catch (MalformedURLException e) {
260N/A log.log(Level.SEVERE, "Could not create connector server: " + e, e);
260N/A System.exit(1);
260N/A } catch (IOException e) {
260N/A log.log(Level.SEVERE, "Could not start connector server: " + e, e);
260N/A System.exit(2);
260N/A } catch (Exception ex) {
260N/A Logger.getLogger(OGAgent.class.getName()).log(Level.SEVERE, null, ex);
260N/A System.exit(1);
260N/A }
260N/A } else {
260N/A System.exit(1);
260N/A }
260N/A }
260N/A
260N/A private OGAgent(Properties props) {
260N/A this.props = props;
260N/A }
260N/A
260N/A public void runOGA() throws MalformedURLException, IOException, JMException {
260N/A String machinename = java.net.InetAddress.getLocalHost().getHostName();
260N/A String javaver = System.getProperty("java.version");
260N/A
260N/A
260N/A log.info("Starting " + Info.getFullVersion() +
260N/A " JMX Agent, with java version " + javaver);
260N/A //create mbeanserver
260N/A
260N/A String connprotocol = props.getProperty("org.opensolaris.opengrok.management.connection.protocol", "jmxmp");
260N/A int connectorport = Integer.parseInt(props.getProperty("org.opensolaris.opengrok.management.connection.port"));
260N/A log.fine("Using protocol " + connprotocol + ", port: " + connectorport);
260N/A
260N/A ArrayList mbservs = MBeanServerFactory.findMBeanServer(null);
364N/A log.fine("Finding MBeanservers, size " + mbservs.size());
364N/A if (mbservs.isEmpty()) {
364N/A server = MBeanServerFactory.createMBeanServer();
364N/A } else {
364N/A server = (MBeanServer) mbservs.get(0);
800N/A }
364N/A
364N/A //instantiate and register OGAManagement
540N/A ObjectName manager = new ObjectName("OGA:name=Management");
540N/A server.registerMBean(Management.getInstance(props), manager);
540N/A
540N/A //instantiate and register OGA:JMXConfiguration
540N/A ObjectName config = new ObjectName("OGA:name=JMXConfiguration");
540N/A JMXConfiguration jc = new JMXConfiguration();
540N/A server.registerMBean(jc, config);
540N/A
540N/A //instantiate and register Timer service and resource purger
540N/A createIndexTimer(props);
540N/A log.info("MBeans registered");
540N/A
540N/A // Create and start connector server
540N/A log.fine("Starting JMX connector");
540N/A HashMap<String, Object> env = new HashMap<String, Object>();
540N/A JMXServiceURL url = new JMXServiceURL(connprotocol, machinename, connectorport);
260N/A JMXConnectorServer connectorServer = null;
if ("jmxmp".equals(connprotocol)) {
connectorServer = new JMXMPConnectorServer(url, env, server);
} else if ("rmi".equals(connprotocol) || "iiop".equals(connprotocol)) {
connectorServer = new RMIConnectorServer(url, env, server);
} else {
throw new IOException("Unknown connector protocol");
}
connectorServer.start();
log.info("OGA is ready and running...");
}
private void createIndexTimer(Properties properties) throws IOException, JMException {
//instantiate, register and start the Timer service
ObjectName timer = new ObjectName("service:name=timer");
server.registerMBean(new Timer(), timer);
server.invoke(timer, "start", null, null);
log.info("Started timer service");
boolean enabled = Boolean.parseBoolean(properties.getProperty("org.opensolaris.opengrok.management.indexer.enabled"));
int period = Integer.parseInt(properties.getProperty("org.opensolaris.opengrok.management.indexer.sleeptime"));
log.fine("Indexer enabled: " + enabled);
log.fine("Indexer period: " + period + " seconds");
//instantiate and register resource purger
ObjectName indexRunner = new ObjectName("OGA:name=AgentIndexRunner," + "source=timer");
server.registerMBean(AgentIndexRunner.getInstance(enabled), indexRunner);
// Add index notification to timer (read from org.opensolaris.opengrok.management.indexer.sleeptime property).
Date date = new Date(System.currentTimeMillis() + Timer.ONE_SECOND * 5);
Long longPeriod = Long.valueOf(period * Timer.ONE_SECOND);
Integer id = (Integer) server.invoke(timer, "addNotification",
new Object[]{"timer.notification", // Type
"Time to index again", // Message
null, // user data
date, // Start time
longPeriod, // Period
},
new String[]{String.class.getName(),
String.class.getName(),
Object.class.getName(),
Date.class.getName(),
"long",});
// Add indexer as listener to index notifications
NotificationFilter filter = new TimerFilter(id);
server.addNotificationListener(timer, indexRunner, filter, null);
}
@SuppressWarnings("PMD.SystemPrintln")
private static boolean createLogger(Properties props) {
boolean ret = true;
String OGAlogpath = props.getProperty("org.opensolaris.opengrok.management.logging.path");
Level loglevel = null;
try {
loglevel = Level.parse(props.getProperty("org.opensolaris.opengrok.management.logging.filelevel"));
} catch (Exception exll) {
loglevel = Level.FINE;
}
Level consoleloglevel = null;
try {
consoleloglevel = Level.parse(props.getProperty("org.opensolaris.opengrok.management.logging.consolelevel"));
} catch (Exception excl) {
consoleloglevel = Level.INFO;
}
try {
OpenGrokLogger.setupLogger(OGAlogpath, loglevel, consoleloglevel);
} catch (IOException ex) {
System.err.println("OGAgent failed set up logging: " + ex);
ret = false;
}
return ret;
}
}