/*
* JBoss, Home of Professional Open Source.
* Copyright 2008, Red Hat Middleware LLC, 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.jboss.test.system.controller.integration.support;
import javax.management.MBeanServer;
import javax.management.ObjectName;
import javax.management.StandardMBean;
import org.jboss.aop.microcontainer.aspects.jmx.JMX;
import org.jboss.dependency.spi.ControllerContext;
import org.jboss.logging.Logger;
import org.jboss.metadata.spi.MetaData;
import org.jboss.system.ServiceControllerMBean;
/**
* @author <a href="mailto:ales.justin@jboss.com">Ales Justin</a>
*/
public class TestServiceControllerLifecycleCallback
{
/** The log */
private static final Logger log = Logger.getLogger(TestServiceControllerLifecycleCallback.class);
/** The mbean server */
private MBeanServer mbeanServer;
/** The service controller */
private ServiceControllerMBean serviceController;
public void setMbeanServer(MBeanServer mbeanServer)
{
this.mbeanServer = mbeanServer;
}
/**
* Set the serviceController.
*
* @param serviceController the serviceController.
*/
public void setServiceController(ServiceControllerMBean serviceController)
{
this.serviceController = serviceController;
}
public void create() throws Exception
{
if (mbeanServer == null)
throw new IllegalArgumentException("No mbean server.");
if (serviceController == null)
throw new IllegalStateException("No service controller configured");
}
public void install(ControllerContext context) throws Exception
{
JMX jmx = readJmxAnnotation(context);
ObjectName objectName = createObjectName(context, jmx);
@SuppressWarnings("unchecked")
Class intfClass = null;
boolean registerDirectly = false;
if (jmx != null)
{
intfClass = jmx.exposedInterface();
registerDirectly = jmx.registerDirectly();
}
@SuppressWarnings("unchecked")
Object mbean = (registerDirectly ? context.getTarget() : new StandardMBean(context.getTarget(), intfClass));
mbeanServer.registerMBean(mbean, objectName);
try
{
serviceController.start(objectName);
}
catch (Exception e)
{
try
{
mbeanServer.unregisterMBean(objectName);
}
catch (Exception t)
{
log.debug("Error unregistering mbean", t);
}
throw e;
}
log.debug("Registered MBean " + objectName);
}
public void uninstall(ControllerContext context) throws Exception
{
JMX jmx = readJmxAnnotation(context);
ObjectName objectName = createObjectName(context, jmx);
log.debug("Unregistering MBean " + objectName);
serviceController.destroy(objectName);
}
private JMX readJmxAnnotation(ControllerContext context) throws Exception
{
MetaData metaData = context.getScopeInfo().getMetaData();
if (metaData != null)
return metaData.getAnnotation(JMX.class);
return null;
}
private ObjectName createObjectName(ControllerContext context, JMX jmx) throws Exception
{
ObjectName objectName = null;
if (jmx != null)
{
String jmxName = jmx.name();
if (jmxName != null && jmxName.length() > 0)
objectName = new ObjectName(jmxName);
}
if (objectName == null)
{
// try to build one from the bean name
String name = (String) context.getName();
if (name.contains(":"))
{
objectName = new ObjectName(name);
}
else
{
objectName = new ObjectName("test:name='" + name + "'");
}
}
return objectName;
}
}