/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
*/
package org.apache.geronimo.commands;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.Iterator;
import java.io.IOException;
import javax.management.remote.JMXServiceURL;
import javax.management.remote.JMXConnector;
import javax.management.remote.JMXConnectorFactory;
import javax.management.MBeanServerConnection;
import javax.management.ObjectName;
//
// FIXME: It should be possible to query state with-out any Geronimo classes,
// just using JMX interfaces.
//
import org.apache.geronimo.gbean.AbstractName;
import org.apache.geronimo.gbean.AbstractNameQuery;
import org.apache.geronimo.kernel.Kernel;
import org.apache.geronimo.kernel.config.PersistentConfigurationList;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Helper to communicate with a remote server via JMX.
*
* @version $Rev: 636614 $ $Date: 2008-03-12 23:15:52 -0400 (Wed, 12 Mar 2008) $
*/
public class ServerProxy
{
private static final Logger log = LoggerFactory.getLogger(ServerProxy.class);
private JMXServiceURL url;
private Map environment;
private MBeanServerConnection mbeanConnection;
private Throwable lastError;
public ServerProxy(final JMXServiceURL url, final Map environment) throws Exception {
assert url != null;
assert environment != null;
this.url = url;
this.environment = environment;
log.debug("Initialized with URL: " + url + ", environment: " + environment);
}
public ServerProxy(final String hostname, final int port, final String username, final String password) throws Exception {
this("service:jmx:rmi://" + hostname + "/jndi/rmi://" + hostname + ":" + port + "/JMXConnector", username, password);
}
public ServerProxy(final String url, final String username, final String password) throws Exception {
assert url != null;
assert username != null;
assert password != null;
this.url = new JMXServiceURL(url);
this.environment = new HashMap();
this.environment.put("jmx.remote.credentials", new String[] {username, password});
log.debug("Initialized with URL: " + url + ", environment: " + environment);
}
public ServerProxy(JMXConnector connector) throws Exception {
this.mbeanConnection = connector.getMBeanServerConnection();
}
private MBeanServerConnection getConnection() throws IOException {
if (this.mbeanConnection == null) {
log.debug("Connecting to: " + url);
JMXConnector connector = JMXConnectorFactory.connect(url, environment);
this.mbeanConnection = connector.getMBeanServerConnection();
log.debug("Connected");
}
return mbeanConnection;
}
public boolean isFullyStarted() {
boolean fullyStarted = true;
try {
AbstractNameQuery query = new AbstractNameQuery(PersistentConfigurationList.class.getName());
Set result = listGBeans(query);
Iterator iter = result.iterator();
while (iter.hasNext()) {
AbstractName name = (AbstractName)iter.next();
boolean started = getBooleanAttribute(name, "kernelFullyStarted");
if (!started) {
fullyStarted = false;
break;
}
}
}
catch (IOException e) {
String msg = "Connection failure; ignoring";
if (log.isTraceEnabled()) {
log.trace(msg, e);
}
else if (log.isDebugEnabled()) {
log.debug(msg + ": " + e);
}
fullyStarted = false;
lastError = e;
}
catch (Exception e) {
String msg = "Unable to determine if the server is fully started; ignoring";
if (log.isTraceEnabled()) {
log.trace(msg, e);
}
else if (log.isDebugEnabled()) {
log.debug(msg + ": " + e);
}
fullyStarted = false;
lastError = e;
}
return fullyStarted;
}
public String getGeronimoHome() {
String home = null;
try {
ObjectName systemInfoQuery = new ObjectName("*:name=ServerInfo,j2eeType=GBean,*");
getConnection();
Set set = mbeanConnection.queryNames(systemInfoQuery, null);
if (set.size() > 0) {
ObjectName found = (ObjectName)set.iterator().next();
home = (String)mbeanConnection.getAttribute(found, "currentBaseDirectory");
}
}
catch (IOException e) {
String msg = "Connection failure; ignoring";
if (log.isTraceEnabled()) {
log.trace(msg, e);
}
else if (log.isDebugEnabled()) {
log.debug(msg + ": " + e);
}
lastError = e;
}
catch (Exception e) {
String msg = "Unable to determine if the server is fully started; ignoring";
if (log.isTraceEnabled()) {
log.trace(msg, e);
}
else if (log.isDebugEnabled()) {
log.debug(msg + ": " + e);
}
lastError = e;
}
return home;
}
public Throwable getLastError() {
return lastError;
}
public void shutdown() throws Exception {
invoke("shutdown");
}
//
// Kernel invocation helpers
//
private Object invoke(final String operation, final Object[] args, final String[] signature) throws Exception {
assert operation != null;
assert args != null;
assert signature != null;
return getConnection().invoke(Kernel.KERNEL, operation, args, signature);
}
private Object invoke(final String operation, final Object[] args) throws Exception {
assert args != null;
String[] signature = new String[args.length];
for (int i=0; i<args.length; i++) {
signature[i] = args[i].getClass().getName();
}
return invoke(operation, args, signature);
}
private Object invoke(final String operation) throws Exception {
return invoke(operation, new Object[0]);
}
private Set listGBeans(final AbstractNameQuery query) throws Exception {
return (Set)invoke("listGBeans", new Object[] { query });
}
private Object getAttribute(final AbstractName name, final String attribute) throws Exception {
assert name != null;
assert attribute != null;
return invoke("getAttribute", new Object[] { name, attribute });
}
private boolean getBooleanAttribute(final AbstractName name, final String attribute) throws Exception {
Object obj = getAttribute(name, attribute);
if (obj instanceof Boolean) {
return ((Boolean)obj).booleanValue();
}
else {
throw new RuntimeException("Attribute is not of type Boolean: " + attribute);
}
}
}