/*
*
* 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.qpid.server.jmx;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.ServiceLoader;
import javax.management.JMException;
import javax.management.StandardMBean;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.log4j.Logger;
import org.apache.qpid.AMQException;
import org.apache.qpid.server.jmx.mbeans.LoggingManagementMBean;
import org.apache.qpid.server.jmx.mbeans.UserManagementMBean;
import org.apache.qpid.server.jmx.mbeans.ConfigurationManagementMBean;
import org.apache.qpid.server.jmx.mbeans.ServerInformationMBean;
import org.apache.qpid.server.jmx.mbeans.Shutdown;
import org.apache.qpid.server.jmx.mbeans.VirtualHostMBean;
import org.apache.qpid.server.logging.log4j.LoggingFacade;
import org.apache.qpid.server.model.Broker;
import org.apache.qpid.server.model.ConfigurationChangeListener;
import org.apache.qpid.server.model.ConfiguredObject;
import org.apache.qpid.server.model.PasswordCredentialManagingAuthenticationProvider;
import org.apache.qpid.server.model.State;
import org.apache.qpid.server.model.VirtualHost;
import org.apache.qpid.server.registry.ApplicationRegistry;
public class JMXService implements ConfigurationChangeListener
{
private static final ClassLoader BUNDLE_CLASSLOADER = JMXService.class.getClassLoader();
private static final Logger LOGGER = Logger.getLogger(JMXService.class);
private final Broker _broker;
private final JMXManagedObjectRegistry _objectRegistry;
private final Shutdown _shutdown;
private final ServerInformationMBean _serverInfo;
private final ConfigurationManagementMBean _configManagement;
private final LoggingManagementMBean _loggingManagement;
private final Map<ConfiguredObject, AMQManagedObject> _children = new HashMap<ConfiguredObject, AMQManagedObject>();
public JMXService() throws AMQException, JMException
{
_broker = ApplicationRegistry.getInstance().getBroker();
_objectRegistry = new JMXManagedObjectRegistry();
_broker.addChangeListener(this);
synchronized (_children)
{
for(VirtualHost virtualHost : _broker.getVirtualHosts())
{
if(!_children.containsKey(virtualHost))
{
_children.put(virtualHost, new VirtualHostMBean(virtualHost, _objectRegistry));
}
}
}
_shutdown = new Shutdown(_objectRegistry);
_serverInfo = new ServerInformationMBean(_objectRegistry, _broker);
_configManagement = new ConfigurationManagementMBean(_objectRegistry);
_loggingManagement = new LoggingManagementMBean(LoggingFacade.getCurrentInstance(), _objectRegistry);
}
public void start() throws IOException, ConfigurationException
{
_objectRegistry.start();
}
public void close()
{
_broker.removeChangeListener(this);
_objectRegistry.close();
}
public void stateChanged(ConfiguredObject object, State oldState, State newState)
{
}
public void childAdded(ConfiguredObject object, ConfiguredObject child)
{
synchronized (_children)
{
try
{
AMQManagedObject mbean;
if(child instanceof VirtualHost)
{
VirtualHost vhostChild = (VirtualHost)child;
mbean = new VirtualHostMBean(vhostChild, _objectRegistry);
}
else if(child instanceof PasswordCredentialManagingAuthenticationProvider)
{
mbean = new UserManagementMBean((PasswordCredentialManagingAuthenticationProvider) child, _objectRegistry);
}
else
{
mbean = null;
}
if (mbean != null)
{
createAdditionalMBeansFromProviders(child, mbean);
}
}
catch(JMException e)
{
LOGGER.error("Error creating mbean", e);
// TODO - Implement error reporting on mbean creation
}
}
}
public void childRemoved(ConfiguredObject object, ConfiguredObject child)
{
// TODO - implement vhost removal (possibly just removing the instanceof check below)
synchronized (_children)
{
if(child instanceof PasswordCredentialManagingAuthenticationProvider)
{
AMQManagedObject mbean = _children.remove(child);
if(mbean != null)
{
try
{
mbean.unregister();
}
catch(JMException e)
{
LOGGER.error("Error creating mbean", e);
//TODO - report error on removing child MBean
}
}
}
}
}
private void createAdditionalMBeansFromProviders(ConfiguredObject child, AMQManagedObject mbean) throws JMException
{
_children.put(child, mbean);
for (Iterator<MBeanProvider> iterator = getMBeanProviderIterator(); iterator.hasNext();)
{
MBeanProvider provider = iterator.next();
LOGGER.debug("Consulting mbean provider : " + provider + " for child : " + child);
if (provider.isChildManageableByMBean(child))
{
LOGGER.debug("Provider will create mbean ");
StandardMBean bean = provider.createMBean(child, mbean);
// TODO track the mbeans that have been created on behalf of a child in a map, then
// if the child is ever removed, destroy these beans too.
}
}
}
/**
* Finds all classes implementing the {@link MBeanProvider} interface. This will find
* <b>only</b> those classes which are visible to the classloader of this OSGI bundle.
*/
private Iterator<MBeanProvider> getMBeanProviderIterator()
{
return ServiceLoader.load(MBeanProvider.class, BUNDLE_CLASSLOADER).iterator();
}
}