Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.jboss.as.arquillian.container.ManagementClient Maven / Gradle / Ivy
/*
* Copyright 2015 Red Hat, Inc.
*
* 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.
*/
package org.jboss.as.arquillian.container;
import static org.jboss.as.controller.client.helpers.ClientConstants.CHILD_TYPE;
import static org.jboss.as.controller.client.helpers.ClientConstants.CONTROLLER_PROCESS_STATE_STARTING;
import static org.jboss.as.controller.client.helpers.ClientConstants.CONTROLLER_PROCESS_STATE_STOPPING;
import static org.jboss.as.controller.client.helpers.ClientConstants.DEPLOYMENT;
import static org.jboss.as.controller.client.helpers.ClientConstants.FAILURE_DESCRIPTION;
import static org.jboss.as.controller.client.helpers.ClientConstants.INCLUDE_RUNTIME;
import static org.jboss.as.controller.client.helpers.ClientConstants.OP;
import static org.jboss.as.controller.client.helpers.ClientConstants.OP_ADDR;
import static org.jboss.as.controller.client.helpers.ClientConstants.OUTCOME;
import static org.jboss.as.controller.client.helpers.ClientConstants.READ_ATTRIBUTE_OPERATION;
import static org.jboss.as.controller.client.helpers.ClientConstants.READ_CHILDREN_NAMES_OPERATION;
import static org.jboss.as.controller.client.helpers.ClientConstants.READ_RESOURCE_OPERATION;
import static org.jboss.as.controller.client.helpers.ClientConstants.RECURSIVE_DEPTH;
import static org.jboss.as.controller.client.helpers.ClientConstants.RESULT;
import static org.jboss.as.controller.client.helpers.ClientConstants.SUBSYSTEM;
import static org.jboss.as.controller.client.helpers.ClientConstants.SUCCESS;
import java.io.Closeable;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import javax.management.Attribute;
import javax.management.AttributeList;
import javax.management.AttributeNotFoundException;
import javax.management.InstanceAlreadyExistsException;
import javax.management.InstanceNotFoundException;
import javax.management.IntrospectionException;
import javax.management.InvalidAttributeValueException;
import javax.management.ListenerNotFoundException;
import javax.management.MBeanException;
import javax.management.MBeanInfo;
import javax.management.MBeanRegistrationException;
import javax.management.MBeanServerConnection;
import javax.management.NotCompliantMBeanException;
import javax.management.NotificationFilter;
import javax.management.NotificationListener;
import javax.management.ObjectInstance;
import javax.management.ObjectName;
import javax.management.QueryExp;
import javax.management.ReflectionException;
import javax.management.remote.JMXConnector;
import javax.management.remote.JMXConnectorFactory;
import javax.management.remote.JMXServiceURL;
import javax.security.auth.callback.CallbackHandler;
import org.jboss.arquillian.container.spi.client.protocol.metadata.HTTPContext;
import org.jboss.arquillian.container.spi.client.protocol.metadata.JMXContext;
import org.jboss.arquillian.container.spi.client.protocol.metadata.ProtocolMetaData;
import org.jboss.arquillian.container.spi.client.protocol.metadata.Servlet;
import org.jboss.as.controller.client.ModelControllerClient;
import org.jboss.as.controller.client.helpers.Operations;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;
import org.jboss.dmr.Property;
import org.jboss.logging.Logger;
import org.wildfly.common.Assert;
/**
* A helper class to join management related operations, like extract sub system ip/port (web/jmx)
* and deployment introspection.
*
*
* Instances of this type are not thread-safe.
*
* @author Aslak Knutsen
*/
public class ManagementClient implements Closeable {
private static final Logger logger = Logger.getLogger(ManagementClient.class);
private static final String SUBDEPLOYMENT = "subdeployment";
private static final String UNDERTOW = "undertow";
private static final String NAME = "name";
private static final String SERVLET = "servlet";
private static final String POSTFIX_WEB = ".war";
private static final String POSTFIX_EAR = ".ear";
private static final ModelNode UNDERTOW_SUBSYSTEM_ADDRESS = new ModelNode().add("subsystem", UNDERTOW);
private final String mgmtAddress;
private final int mgmtPort;
private final String mgmtProtocol;
private final ModelControllerClient client;
private final CommonContainerConfiguration config;
private boolean initialized = false;
private URI webUri;
private URI ejbUri;
private ModelNode undertowSubsystem = null;
private MBeanServerConnection connection;
private JMXConnector connector;
private boolean undertowSubsystemPresent = false;
private boolean jmxSubsystemPresent = false;
private boolean closed = false;
public ManagementClient(ModelControllerClient client, final String mgmtAddress, final int managementPort, final String protocol) {
if (client == null) {
throw new IllegalArgumentException("Client must be specified");
}
this.client = client;
this.mgmtAddress = mgmtAddress;
this.mgmtPort = managementPort;
this.mgmtProtocol = protocol;
this.config = null;
}
public ManagementClient(ModelControllerClient client, final CommonContainerConfiguration config) {
if (client == null) {
throw new IllegalArgumentException("Client must be specified");
}
Assert.checkNotNullParam("config", config);
this.client = client;
this.mgmtAddress = config.getManagementAddress();
this.mgmtPort = config.getManagementPort();
this.mgmtProtocol = config.getManagementProtocol();
this.config = config;
}
//-------------------------------------------------------------------------------------||
// Public API -------------------------------------------------------------------------||
//-------------------------------------------------------------------------------------||
/**
* Returns the client used to connect to the server.
*
* @return the client
*
* @throws IllegalStateException if this has been {@linkplain #close() closed}
*/
public ModelControllerClient getControllerClient() {
checkState();
return client;
}
/**
* Resets the client. {@linkplain #close() Closes} open resources and resets flags so that the client itself can be
* reinitialized later if desired.
*/
void reset() {
close();
initialized = false;
closed = false;
}
/**
* Checks whether or not the Undertow subsystem is present and sets the internal state if it is. An invocation of
* this should happen after the server has been started.
*
* @throws IllegalStateException if this has been {@linkplain #close() closed}
*/
private void init() {
AccessController.doPrivileged(new PrivilegedAction() {
@Override
public Object run() {
checkState();
if (!initialized) {
initialized = true;
try {
ModelNode op = Operations.createReadResourceOperation(UNDERTOW_SUBSYSTEM_ADDRESS, true);
ModelNode result = client.execute(op);
undertowSubsystemPresent = Operations.isSuccessfulOutcome(result);
if (undertowSubsystemPresent) {
undertowSubsystem = Operations.readResult(result);
}
URI webUri;
try {
webUri = new URI("http://localhost:8080");
} catch (URISyntaxException e) {
throw new RuntimeException(e);
}
if (config != null && config.getSocketBindingName() != null) {
webUri = getBinding(config.getSocketBindingName());
} else {
if (undertowSubsystem != null && undertowSubsystem.hasDefined("server")) {
List vhosts = undertowSubsystem.get("server").asPropertyList();
ModelNode socketBinding = new ModelNode();
if (!vhosts.isEmpty()) {//if empty no virtual hosts defined
socketBinding = vhosts.get(0)
.getValue()
.get("http-listener", "default")
.get("socket-binding");
}
if (socketBinding.isDefined()) {
webUri = getBinding(socketBinding.asString());
}
}
}
ManagementClient.this.webUri = webUri;
try {
ejbUri = new URI("http-remoting", webUri.getUserInfo(), webUri.getHost(), webUri.getPort(), null, null, null);
} catch (URISyntaxException e) {
throw new RuntimeException(e);
}
// Determine if JMX is available
op = Operations.createOperation(READ_CHILDREN_NAMES_OPERATION);
op.get(CHILD_TYPE).set(SUBSYSTEM);
result = client.execute(op);
if (!Operations.isSuccessfulOutcome(result)) {
throw new RuntimeException("Failed to determine if the JMX subsystem is present: "
+ Operations.getFailureDescription(result).asString());
}
jmxSubsystemPresent = Operations.readResult(result)
.asList()
.stream()
.map(ModelNode::asString)
.anyMatch("jmx"::equals);
} catch (Exception e) {
throw new RuntimeException("Could not init arquillian protocol", e);
}
}
return null;
}
});
}
/**
* @return The base URI or the web susbsystem. Usually http://localhost:8080
*
* @throws IllegalStateException if this has been {@linkplain #close() closed}
*/
public URI getWebUri() {
init();
return webUri;
}
/**
* Gets the meta-data.
*
* @return the meta-data
*
* @throws IllegalStateException if this has been {@linkplain #close() closed}
*/
public ProtocolMetaData getProtocolMetaData(String deploymentName) {
init();
ProtocolMetaData metaData = new ProtocolMetaData();
if (jmxSubsystemPresent) {
metaData.addContext(new JMXContext(getConnection()));
}
if (undertowSubsystemPresent) {
URI webURI = getWebUri();
HTTPContext context = new HTTPContext(webURI.getHost(), webURI.getPort(), "https".equalsIgnoreCase(webURI.getScheme()));
metaData.addContext(context);
try {
final ModelNode deploymentNode = readDeploymentNode(deploymentName);
if (isWebArchive(deploymentName)) {
extractWebArchiveContexts(context, deploymentNode);
} else if (isEnterpriseArchive(deploymentName)) {
extractEnterpriseArchiveContexts(context, deploymentNode);
}
} catch (Exception e) {
throw new RuntimeException(e);
}
}
return metaData;
}
/**
* Checks whether or not the server is running.
*
* Note that if this client has been {@linkplain #close() closed} the state of the server cannot be checked.
*
*
* @return {@code true} if the server is running, otherwise {@code false}
*
* @throws IllegalStateException if this has been {@linkplain #close() closed}
*/
public boolean isServerInRunningState() {
checkState();
try {
ModelNode op = new ModelNode();
op.get(OP).set(READ_ATTRIBUTE_OPERATION);
op.get(OP_ADDR).setEmptyList();
op.get(NAME).set("server-state");
ModelNode rsp = client.execute(op);
return SUCCESS.equals(rsp.get(OUTCOME).asString())
&& !CONTROLLER_PROCESS_STATE_STARTING.equals(rsp.get(RESULT).asString())
&& !CONTROLLER_PROCESS_STATE_STOPPING.equals(rsp.get(RESULT).asString());
} catch (RuntimeException rte) {
throw rte;
} catch (IOException ex) {
return false;
}
}
/**
* Checks whether or not the client has been closed.
*
* @return {@code true} if the client has been closed otherwise {@code false}
*/
public boolean isClosed() {
return closed;
}
@Override
public void close() {
if (!closed) {
try {
client.close();
closed = true;
} catch (IOException e) {
throw new RuntimeException("Could not close connection", e);
} finally {
if (connector != null) {
try {
connector.close();
} catch (IOException e) {
throw new RuntimeException("Could not close JMX connection", e);
}
}
}
}
}
private static ModelNode defined(final ModelNode node, final String message) {
if (!node.isDefined()) { throw new IllegalStateException(message); }
return node;
}
private URI getBinding(final String socketBinding) {
String protocol = "http";
String host = null;
int port = -1;
if (config != null) {
host = config.getHost();
port = config.getPort();
protocol = config.getProtocol();
}
try {
if (host == null || port < 0) {
ModelNode sbgOp = new ModelNode();
sbgOp.get(OP).set(READ_CHILDREN_NAMES_OPERATION);
sbgOp.get(CHILD_TYPE).set("socket-binding-group");
final ModelNode socketBindingGroups = executeForResult(sbgOp);
final String socketBindingGroupName = socketBindingGroups.asList().get(0).asString();
final ModelNode operation = new ModelNode();
operation.get(OP_ADDR).get("socket-binding-group").set(socketBindingGroupName);
operation.get(OP_ADDR).get("socket-binding").set(socketBinding);
operation.get(OP).set(READ_RESOURCE_OPERATION);
operation.get("include-runtime").set(true);
ModelNode binding = executeForResult(operation);
if (host == null) {
host = formatIP(binding.get("bound-address").asString());
}
// Check if we need to look up the port
if (port < 0) {
port = defined(binding.get("bound-port"),
socketBindingGroupName + " -> " + socketBinding + " -> bound-port is undefined").asInt();
}
}
return URI.create(protocol + "://" + NetworkUtils.formatPossibleIpv6Address(host) + ":" + port);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
static String formatIP(String ip) {
//it appears some system can return a binding with the zone specifier on the end
if (ip.contains(":") && ip.contains("%")) {
ip = ip.split("%")[0];
}
if (ip.equals("0.0.0.0")) {
logger.debug("WildFly is bound to 0.0.0.0 which is correct, setting client to 127.0.0.1");
ip = "127.0.0.1";
}
return ip;
}
//-------------------------------------------------------------------------------------||
// Metadata Extraction Operations -----------------------------------------------------||
//-------------------------------------------------------------------------------------||
private boolean isEnterpriseArchive(String deploymentName) {
return deploymentName.endsWith(POSTFIX_EAR);
}
private boolean isWebArchive(String deploymentName) {
return deploymentName.endsWith(POSTFIX_WEB);
}
/**
* Expects the {@code deploymentNode} to bit a list of addresses which contain a result of the subsystem model. The
* deployment node model should be the equivalent of
*
* /deployment=someEar.ear/subdeployment=*/subsystem=undertow:read-resource(recursive-depth=2, include-runtime=true)
*
*
* Example Output :
*
*
*
* [{
* "address" => [
* ("deployment" => "example-app.ear"),
* ("subdeployment" => "example.war"),
* ("subsystem" => "undertow")
* ],
* "outcome" => "success",
* "result" => {
* "active-sessions" => 0,
* "context-root" => "/example",
* "expired-sessions" => 0,
* "max-active-sessions" => -1,
* "rejected-sessions" => 0,
* "server" => "default-server",
* "session-avg-alive-time" => 0,
* "session-max-alive-time" => 0,
* "sessions-created" => 0,
* "virtual-host" => "default-host",
* "servlet" => {"ExapleService" => {
* "mappings" => 0,
* "max-request-time" => 0,
* "min-request-time" => 0,
* "request-count" => 0,
* "servlet-class" => "org.jboss.example.ExampleServlet",
* "servlet-name" => "ExampleService",
* "total-request-time" => 0
* }},
* "websocket" => undefined
* }
* }]
*
*
*
address = subdeployment.get(OP_ADDR).asPropertyList();
if (address.size() > 1) {
// This should be the subdeployment name
final String deploymentName = address.get(1).getValue().asString();
if (isWebArchive(deploymentName)) {
extractWebArchiveContexts(context, Operations.readResult(subdeployment));
}
}
}
}
}
}
private void extractWebArchiveContexts(HTTPContext context, ModelNode deploymentNode) {
if (deploymentNode.isDefined() && deploymentNode.hasDefined("context-root")) {
final String contextName = toContextName(deploymentNode.get("context-root").asString());
if (deploymentNode.hasDefined(SERVLET)) {
for (final ModelNode servletNode : deploymentNode.get(SERVLET).asList()) {
for (final String servletName : servletNode.keys()) {
context.add(new Servlet(servletName, contextName));
}
}
}
/*
* This is a WebApp, it has some form of webcontext whether it has a
* Servlet or not. AS7 does not expose jsp / default servlet in mgm api
*/
context.add(new Servlet("default", contextName));
}
}
private String toContextName(String deploymentName) {
String correctedName = deploymentName;
if (correctedName.startsWith("/")) {
correctedName = correctedName.substring(1);
}
return correctedName;
}
//-------------------------------------------------------------------------------------||
// Common Management API Operations ---------------------------------------------------||
//-------------------------------------------------------------------------------------||
private ModelNode executeForResult(final ModelNode operation) throws Exception {
checkState();
final ModelNode result = client.execute(operation);
checkSuccessful(result, operation);
return result.get(RESULT);
}
private ModelNode readDeploymentNode(final String deploymentName) throws IOException, UnSuccessfulOperationException {
final ModelNode address;
if (isWebArchive(deploymentName)) {
address = Operations.createAddress(DEPLOYMENT, deploymentName, SUBSYSTEM, UNDERTOW);
} else if (isEnterpriseArchive(deploymentName)) {
address = Operations.createAddress(DEPLOYMENT, deploymentName, SUBDEPLOYMENT, "*", SUBSYSTEM, UNDERTOW);
} else {
// We don't have a web context so just return the meta data
return new ModelNode();
}
final ModelNode operation = Operations.createReadResourceOperation(address);
operation.get(RECURSIVE_DEPTH).set(2);
operation.get(INCLUDE_RUNTIME).set(true);
// Normally this could just use the executeForResult(). However singleton deployments may not be on a
// node that has been started and therefore the deployment may not have an undertow subsystem resource.
final ModelNode result = client.execute(operation);
if (Operations.isSuccessfulOutcome(result)) {
return Operations.readResult(result);
} else {
// This check is a bit heavy, however we need to determine whether or not the error was due to not
// having the undertow resource on the deployment or the failure was a different type of failure.
// It's possible we could get the failure description and just check the message id. However that
// could be fragile for future usage. We can just check if the subsystem resource is defined and
// has an undertow subsystem. If it does, there was a different type of failure and we should just
// throw an exception.
final ModelNode checkOp = Operations.createOperation(READ_CHILDREN_NAMES_OPERATION,
Operations.createAddress(DEPLOYMENT, deploymentName));
checkOp.get(CHILD_TYPE).set(SUBSYSTEM);
final ModelNode r = client.execute(checkOp);
if (Operations.isSuccessfulOutcome(r)) {
final ModelNode children = Operations.readResult(r);
if (children.isDefined()) {
for (ModelNode child : children.asList()) {
if (UNDERTOW.equals(child.asString())) {
throw new UnSuccessfulOperationException(Operations.getFailureDescription(result).asString());
}
}
}
} else {
// The read-children-names failed, but we'll just throw the original failure
throw new UnSuccessfulOperationException(Operations.getFailureDescription(result).asString());
}
}
return new ModelNode();
}
private void checkSuccessful(final ModelNode result,
final ModelNode operation) throws UnSuccessfulOperationException {
if (!SUCCESS.equals(result.get(OUTCOME).asString())) {
logger.error("Operation " + operation + " did not succeed. Result was " + result);
throw new UnSuccessfulOperationException(result.get(
FAILURE_DESCRIPTION).toString());
}
}
private MBeanServerConnection getConnection() {
MBeanServerConnection connection = this.connection;
if (connection == null) {
try {
final Map env = new HashMap<>();
if (Authentication.username != null && !Authentication.username.isEmpty()) {
// Only set this is there is a username as it disabled local authentication.
env.put(CallbackHandler.class.getName(), Authentication.getCallbackHandler());
}
if (config.getAuthenticationConfig() != null) {
env.put("wildfly.config.url", config.getAuthenticationConfig());
}
final JMXServiceURL serviceURL = getRemoteJMXURL();
final JMXConnector connector = this.connector = JMXConnectorFactory.connect(serviceURL, env);
connection = this.connection = new MBeanConnectionProxy(connector.getMBeanServerConnection());
} catch (IOException e) {
throw new RuntimeException(e);
}
}
return connection;
}
public JMXServiceURL getRemoteJMXURL() {
try {
if ("http-remoting".equals(mgmtProtocol) || "remote+http".equals(mgmtProtocol)) {
return new JMXServiceURL("service:jmx:remote+http://" + NetworkUtils.formatPossibleIpv6Address(mgmtAddress) + ":" + mgmtPort);
} else if (mgmtProtocol.equals("https-remoting")) {
return new JMXServiceURL("service:jmx:remote+https://" + NetworkUtils.formatPossibleIpv6Address(mgmtAddress) + ":" + mgmtPort);
} else {
return new JMXServiceURL("service:jmx:remoting-jmx://" + NetworkUtils.formatPossibleIpv6Address(mgmtAddress) + ":" + mgmtPort);
}
} catch (Exception e) {
throw new RuntimeException("Could not create JMXServiceURL:" + this, e);
}
}
public int getMgmtPort() {
return mgmtPort;
}
public String getMgmtAddress() {
return NetworkUtils.formatPossibleIpv6Address(mgmtAddress);
}
public String getMgmtProtocol() {
return mgmtProtocol;
}
/**
* Returns the URI for EJB's.
*
* @return the resolved EJB URI
*
* @throws IllegalStateException if this has been {@linkplain #close() closed}
*/
public URI getRemoteEjbURL() {
init();
return ejbUri;
}
private void checkState() {
if (closed) {
throw new IllegalStateException("The client connection has been closed.");
}
}
//-------------------------------------------------------------------------------------||
// Helper classes ---------------------------------------------------------------------||
//-------------------------------------------------------------------------------------||
private static class UnSuccessfulOperationException extends Exception {
private static final long serialVersionUID = 1L;
UnSuccessfulOperationException(String message) {
super(message);
}
}
private class MBeanConnectionProxy implements MBeanServerConnection {
private MBeanServerConnection connection;
/**
* @param connection connection to delegate to
*/
private MBeanConnectionProxy(MBeanServerConnection connection) {
super();
this.connection = connection;
}
@Override
public ObjectInstance createMBean(String className, ObjectName name) throws ReflectionException,
InstanceAlreadyExistsException, MBeanException, NotCompliantMBeanException,
IOException {
checkConnection();
return connection.createMBean(className, name);
}
@Override
public ObjectInstance createMBean(String className, ObjectName name, ObjectName loaderName) throws ReflectionException,
InstanceAlreadyExistsException, MBeanException, NotCompliantMBeanException,
InstanceNotFoundException, IOException {
checkConnection();
return connection.createMBean(className, name, loaderName);
}
@Override
public ObjectInstance createMBean(String className, ObjectName name, Object[] params, String[] signature)
throws ReflectionException, InstanceAlreadyExistsException, MBeanException,
NotCompliantMBeanException, IOException {
checkConnection();
return connection.createMBean(className, name, params, signature);
}
@Override
public ObjectInstance createMBean(String className, ObjectName name, ObjectName loaderName, Object[] params,
String[] signature) throws ReflectionException, InstanceAlreadyExistsException,
MBeanException, NotCompliantMBeanException, InstanceNotFoundException, IOException {
checkConnection();
return connection.createMBean(className, name, loaderName, params, signature);
}
@Override
public void unregisterMBean(ObjectName name) throws InstanceNotFoundException, MBeanRegistrationException, IOException {
checkConnection();
connection.unregisterMBean(name);
}
@Override
public ObjectInstance getObjectInstance(ObjectName name) throws InstanceNotFoundException, IOException {
try {
return connection.getObjectInstance(name);
} catch (IOException e) {
checkConnection();
return connection.getObjectInstance(name);
}
}
@Override
public Set queryMBeans(ObjectName name, QueryExp query) throws IOException {
try {
return connection.queryMBeans(name, query);
} catch (IOException e) {
checkConnection();
return connection.queryMBeans(name, query);
}
}
@Override
public Set queryNames(ObjectName name, QueryExp query) throws IOException {
try {
return connection.queryNames(name, query);
} catch (IOException e) {
checkConnection();
return connection.queryNames(name, query);
}
}
@Override
public boolean isRegistered(ObjectName name) throws IOException {
try {
return connection.isRegistered(name);
} catch (IOException e) {
checkConnection();
return connection.isRegistered(name);
}
}
@Override
public Integer getMBeanCount() throws IOException {
try {
return connection.getMBeanCount();
} catch (IOException e) {
checkConnection();
return connection.getMBeanCount();
}
}
@Override
public Object getAttribute(ObjectName name, String attribute) throws MBeanException, AttributeNotFoundException,
InstanceNotFoundException, ReflectionException, IOException {
try {
return connection.getAttribute(name, attribute);
} catch (IOException e) {
checkConnection();
return connection.getAttribute(name, attribute);
}
}
@Override
public AttributeList getAttributes(ObjectName name, String[] attributes) throws InstanceNotFoundException,
ReflectionException, IOException {
try {
return connection.getAttributes(name, attributes);
} catch (IOException e) {
checkConnection();
return connection.getAttributes(name, attributes);
}
}
@Override
public void setAttribute(ObjectName name, Attribute attribute) throws InstanceNotFoundException,
AttributeNotFoundException, InvalidAttributeValueException, MBeanException, ReflectionException, IOException {
checkConnection();
connection.setAttribute(name, attribute);
}
@Override
public AttributeList setAttributes(ObjectName name, AttributeList attributes) throws InstanceNotFoundException,
ReflectionException, IOException {
checkConnection();
return connection.setAttributes(name, attributes);
}
@Override
public Object invoke(ObjectName name, String operationName, Object[] params, String[] signature)
throws InstanceNotFoundException, MBeanException, ReflectionException, IOException {
checkConnection();
return connection.invoke(name, operationName, params, signature);
}
@Override
public String getDefaultDomain() throws IOException {
try {
return connection.getDefaultDomain();
} catch (IOException e) {
checkConnection();
return connection.getDefaultDomain();
}
}
@Override
public String[] getDomains() throws IOException {
try {
return connection.getDomains();
} catch (IOException e) {
checkConnection();
return connection.getDomains();
}
}
@Override
public void addNotificationListener(ObjectName name, NotificationListener listener, NotificationFilter filter,
Object handback) throws InstanceNotFoundException, IOException {
try {
connection.addNotificationListener(name, listener, filter, handback);
} catch (IOException e) {
if (!checkConnection()) {
connection.addNotificationListener(name, listener, filter, handback);
} else {
throw e;
}
}
}
@Override
public void addNotificationListener(ObjectName name, ObjectName listener, NotificationFilter filter, Object handback)
throws InstanceNotFoundException, IOException {
try {
connection.addNotificationListener(name, listener, filter, handback);
} catch (IOException e) {
if (!checkConnection()) {
connection.addNotificationListener(name, listener, filter, handback);
} else {
throw e;
}
}
}
@Override
public void removeNotificationListener(ObjectName name, ObjectName listener) throws InstanceNotFoundException,
ListenerNotFoundException, IOException {
try {
connection.removeNotificationListener(name, listener);
} catch (IOException e) {
if (!checkConnection()) {
connection.removeNotificationListener(name, listener);
} else {
throw e;
}
}
}
@Override
public void removeNotificationListener(ObjectName name, ObjectName listener, NotificationFilter filter, Object handback)
throws InstanceNotFoundException, ListenerNotFoundException, IOException {
try {
connection.removeNotificationListener(name, listener, filter, handback);
} catch (IOException e) {
if (!checkConnection()) {
connection.removeNotificationListener(name, listener, filter, handback);
} else {
throw e;
}
}
}
@Override
public void removeNotificationListener(ObjectName name, NotificationListener listener)
throws InstanceNotFoundException, ListenerNotFoundException, IOException {
try {
connection.removeNotificationListener(name, listener);
} catch (IOException e) {
if (!checkConnection()) {
connection.removeNotificationListener(name, listener);
} else {
throw e;
}
}
}
@Override
public void removeNotificationListener(ObjectName name, NotificationListener listener, NotificationFilter filter,
Object handback) throws InstanceNotFoundException, ListenerNotFoundException, IOException {
try {
connection.removeNotificationListener(name, listener, filter, handback);
} catch (IOException e) {
if (!checkConnection()) {
connection.removeNotificationListener(name, listener, filter, handback);
} else {
throw e;
}
}
}
@Override
public MBeanInfo getMBeanInfo(ObjectName name) throws InstanceNotFoundException, IntrospectionException,
ReflectionException, IOException {
try {
return connection.getMBeanInfo(name);
} catch (IOException e) {
checkConnection();
return connection.getMBeanInfo(name);
}
}
@Override
public boolean isInstanceOf(ObjectName name, String className) throws InstanceNotFoundException, IOException {
try {
return connection.isInstanceOf(name, className);
} catch (IOException e) {
checkConnection();
return connection.isInstanceOf(name, className);
}
}
private boolean checkConnection() {
try {
this.connection.getDefaultDomain();
return true;
} catch (IOException ioe) {
logger.debug("JMX connection error.", ioe);
}
this.connection = reconnect();
return false;
}
private MBeanServerConnection reconnect() {
try {
final Map env = new HashMap<>();
env.put(CallbackHandler.class.getName(), Authentication.getCallbackHandler());
final JMXConnector connector = ManagementClient.this.connector = JMXConnectorFactory.connect(getRemoteJMXURL(), env);
connection = connector.getMBeanServerConnection();
} catch (IOException e) {
throw new RuntimeException(e);
}
return connection;
}
}
}
