* JBoss, Home of Professional Open Source.

* Copyright 2006, 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.

*/

* An EjbModule represents a collection of beans that are deployed as a unit.

*

* @author <a href="mailto:bill@jboss.org">Bill Burke</a>

* @author adrian@jboss.org

* @version $Revision: 76905 $

*/

  implements JavaEEModule, Ejb3DeploymentMBean

   private static final Logger log = Logger.getLogger(Ejb3Deployment.class);

   public static final String ACTUAL_ENTITY_MANAGER_FACTORY_CONTEXT = "java:/ActualEntityManagerFactories";

   public static final String MANAGED_ENTITY_FACTORY_CONTEXT = "java:/managedEntityFactories";

   private JBossMetaData metaData;

  

   private PersistenceMetaData persistenceUnitsMetaData;

   protected DeploymentUnit unit;

   protected LinkedHashMap<ObjectName, Container> ejbContainers = new LinkedHashMap<ObjectName, Container>();

   private boolean processPersistenceUnits = true;

   protected boolean hasEntities;

   protected List<String> explicitEntityClasses = new ArrayList<String>();

   protected List<PersistenceUnitDeployment> persistenceUnitDeployments = new ArrayList<PersistenceUnitDeployment>();

   protected String defaultSLSBDomain = "Stateless Bean";

   protected String defaultSFSBDomain = "Stateful Bean";

   protected String defaultMDBDomain = "Message Driven Bean";

   protected String defaultConsumerDomain = "Consumer Bean";

   protected String defaultServiceDomain = "Service Bean";

   protected InitialContext initialContext;

   protected KernelAbstraction kernelAbstraction;

   // used for @Management interfaces

   protected MBeanServer mbeanServer;

   protected DeploymentScope deploymentScope;

   protected EjbModulePersistenceUnitResolver persistenceUnitResolver;

   protected MessageDestinationResolver messageDestinationResolver;

   protected CacheFactoryRegistry cacheFactoryRegistry;

   protected RemoteProxyFactoryRegistry remoteProxyFactoryRegistry;

   protected PersistenceManagerFactoryRegistry persistenceManagerFactoryRegistry;

   protected PoolFactoryRegistry poolFactoryRegistry;

  

   protected ObjectName objectName;

  

   protected boolean reinitialize = false;

   private org.jboss.deployers.structure.spi.DeploymentUnit deploymentUnit;

  

   private PersistenceUnitDependencyResolver persistenceUnitDependencyResolver;

   @Deprecated

   public Ejb3Deployment(DeploymentUnit unit, DeploymentScope deploymentScope, JBossMetaData metaData, PersistenceMetaData persistenceUnitsMetaData)

   {

      assert unit != null : "unit is null";

     

      this.unit = unit;

      this.deploymentScope = deploymentScope;

      this.metaData = metaData;

      this.persistenceUnitsMetaData = persistenceUnitsMetaData;

      try

      {

         initialContext = InitialContextFactory.getInitialContext(unit.getJndiProperties());

      }

      catch (NamingException e)

      {

         throw new RuntimeException(e);

      }

      persistenceUnitResolver = new EjbModulePersistenceUnitResolver(persistenceUnitDeployments, deploymentScope, ejbContainers);

      MessageDestinationsMetaData destinations = null;

      if (metaData != null && metaData.getAssemblyDescriptor() != null)

         destinations = metaData.getAssemblyDescriptor().getMessageDestinations();

      messageDestinationResolver = new MessageDestinationResolver(deploymentScope, destinations);

   }

  

   /**

    * Do not deploy persistence unit anymore.

    *

    * @param deploymentUnit

    * @param unit

    * @param deploymentScope

    * @param metaData

    */

   public Ejb3Deployment(org.jboss.deployers.structure.spi.DeploymentUnit deploymentUnit, DeploymentUnit unit, DeploymentScope deploymentScope, JBossMetaData metaData)

   {

      this(unit, deploymentScope, metaData, null);

     

      assert deploymentUnit != null : "deploymentUnit is null";

     

      this.deploymentUnit = deploymentUnit;

   }

   public JavaEEApplication getApplication()

   {

      return deploymentScope;

   }

   public DeploymentScope getEar()

   {

      return deploymentScope;

   }

   public KernelAbstraction getKernelAbstraction()

   {

      return kernelAbstraction;

   }

   public MBeanServer getMbeanServer()

   {

      return mbeanServer;

   }

   public void setMbeanServer(MBeanServer mbeanServer)

   {

      this.mbeanServer = mbeanServer;

   }

   public DeploymentUnit getDeploymentUnit()

   {

      return unit;

   }

   public String getDefaultSLSBDomain()

   {

      return defaultSLSBDomain;

   }

   public CacheFactoryRegistry getCacheFactoryRegistry()

   {

      return cacheFactoryRegistry;

   }

   public void setCacheFactoryRegistry(CacheFactoryRegistry registry)

   {

      this.cacheFactoryRegistry = registry;

   }

   public RemoteProxyFactoryRegistry getRemoteProxyFactoryRegistry()

   {

      return remoteProxyFactoryRegistry;

   }

   public void setRemoteProxyFactoryRegistry(RemoteProxyFactoryRegistry registry)

   {

      this.remoteProxyFactoryRegistry = registry;

   }

   public PersistenceManagerFactoryRegistry getPersistenceManagerFactoryRegistry()

   {

      return persistenceManagerFactoryRegistry;

   }

   public void setPersistenceManagerFactoryRegistry(PersistenceManagerFactoryRegistry registry)

   {

      this.persistenceManagerFactoryRegistry = registry;

   }

  

   @Inject

   public void setPersistenceUnitDependencyResolver(PersistenceUnitDependencyResolver resolver)

   {

      this.persistenceUnitDependencyResolver = resolver;

   }

  

   public PoolFactoryRegistry getPoolFactoryRegistry()

   {

      return poolFactoryRegistry;

   }

   public void setPoolFactoryRegistry(PoolFactoryRegistry poolFactoryRegistry)

   {

      this.poolFactoryRegistry = poolFactoryRegistry;

   }

   public void setProcessPersistenceUnits(boolean b)

   {

      this.processPersistenceUnits = b;

   }

  

   /**

    * Returns a partial MBean attribute name of the form

    * ",ear=foo.ear,jar=foo.jar"

    *

    * @return

    */

   public String getScopeKernelName()

   {

      String scopedKernelName = "";

      if (deploymentScope != null)

         scopedKernelName += ",ear=" + deploymentScope.getShortName();

      scopedKernelName += ",jar=" + unit.getShortName();

      return scopedKernelName;

   }

   /**

    * The default AOP domain for stateless session beans

    *

    * @param defaultSLSBDomain

    */

   public void setDefaultSLSBDomain(String defaultSLSBDomain)

   {

      this.defaultSLSBDomain = defaultSLSBDomain;

   }

   public String getDefaultSFSBDomain()

   {

      return defaultSFSBDomain;

   }

   public String getDefaultConsumerDomain()

   {

      return defaultConsumerDomain;

   }

   /**

    * The default stateful session bean aspect domain

    *

    * @param defaultSFSBDomain

    */

   public void setDefaultSFSBDomain(String defaultSFSBDomain)

   {

      this.defaultSFSBDomain = defaultSFSBDomain;

   }

   public String getDefaultMDBDomain()

   {

      return defaultMDBDomain;

   }

   /**

    * The default AOP domain for message driven beans.

    *

    * @param defaultMDBDomain

    */

   public void setDefaultMDBDomain(String defaultMDBDomain)

   {

      this.defaultMDBDomain = defaultMDBDomain;

   }

   public String getDefaultServiceDomain()

   {

      return defaultServiceDomain;

   }

   /**

    * default AOP domain for service beans.

    *

    * @param defaultServiceDomain

    */

   public void setDefaultServiceDomain(String defaultServiceDomain)

   {

      this.defaultServiceDomain = defaultServiceDomain;

   }

   protected String getJaccContextId()

   {

      return unit.getShortName();

   }

   /**

    * Get the deployment ejb container for the given ejb name.

    *

    * @param ejbName the deployment unique ejb name

    * @return the ejb container if found, null otherwise

    * @throws IllegalStateException if the ejbName cannot be used to

    *    for the container name.

    */

   public EJBContainer getEjbContainerForEjbName(String ejbName)

   {

      String ejbObjectName = JavaEEComponentHelper.createObjectName(this, ejbName);

      EJBContainer container = null;

      ObjectName ejbON;

      try

      {

         ejbON = new ObjectName(ejbObjectName);

      }

      catch (Exception e)

      {

         throw new IllegalStateException("Failed to ", e);

      }

      container = (EJBContainer) ejbContainers.get(ejbON);

      return container;

   }

   public Container getContainer(ObjectName name)

   {

      return (Container) ejbContainers.get(name);

   }

   public java.util.Map getEjbContainers()

   {

      return ejbContainers;

   }

   public PersistenceUnitDeployment getPersistenceUnitDeployment(String unitName) throws NameNotFoundException

   {

      return persistenceUnitResolver.getPersistenceUnitDeployment(unitName);

   }

   public PersistenceUnitDeployment getPersistenceUnitDeploymentInternal(String unitName)

   {

      return persistenceUnitResolver.getPersistenceUnitDeploymentInternal(unitName);

   }

   public List<PersistenceUnitDeployment> getPersistenceUnitDeployments()

   {

      return persistenceUnitDeployments;

   }

   public EJBContainer getEjbContainer(String ejbLink, Class businessIntf)

   {

      String relativePath = unit.getRelativePath();

      EJBContainer container = deploymentScope.getEjbContainer(ejbLink, businessIntf, relativePath);

      return container;

   }

   public String getEjbJndiName(String ejbLink, Class businessIntf)

   {

      EJBContainer container = getEjbContainer(ejbLink, businessIntf);

      String jndiName = ProxyFactoryHelper.getJndiName(container, businessIntf);

      return jndiName;

   }

   public String getEjbJndiName(Class businessIntf)

      throws NameNotFoundException

   {

      EJBContainer container = getEjbContainer(businessIntf);

      String jndiName = ProxyFactoryHelper.getJndiName(container, businessIntf);

      return jndiName;

   }

   public EJBContainer getEjbContainer(Class businessIntf) throws NameNotFoundException

   {

      String relativePath = unit.getRelativePath();

      return deploymentScope.getEjbContainer(businessIntf, relativePath);

   }

   protected void processEJBContainerMetadata(Container container) throws Exception

   {

      log.trace("Process EJB container metadata " + container);

      ObjectName on = container.getObjectName();

      ejbContainers.put(on, container);

      container.processMetadata();

   }

  

   protected void registerDeployment() throws Exception

   {

      String on = "jboss.j2ee:jar=" + this.getName() + ",service=EJB3";

      if (metaData != null && metaData.getEnterpriseBeans() != null && metaData.getEnterpriseBeans().getEjbJarMetaData() != null)

      {

         String jmxName = metaData.getEnterpriseBeans().getEjbJarMetaData().getJmxName();

         if (jmxName != null && jmxName.trim().length() > 0)

            on = jmxName;

      }

    

      objectName = new ObjectName(on);

     

      mbeanServer.registerMBean(this, objectName);

   }

  

   protected void unregisterDeployment()

   {

      try

      {

         mbeanServer.unregisterMBean(objectName);

      }

      catch (Exception e)

      {

         log.debug("error trying to stop ejb deployment", e);

      }

   }

   protected void registerEJBContainer(Container container) throws Exception

   {

      ObjectName on = container.getObjectName();

      String name = on.getCanonicalName();

      DependencyPolicy dependsPolicy = container.getDependencyPolicy();

      dependsPolicy.addDependency("jboss.ejb:service=EJBTimerService");

      kernelAbstraction.install(name, dependsPolicy, unit, container);

      mbeanServer.registerMBean(container.getMBean(), on);

      log.debug("Bound ejb3 container " + name);

   }

   protected abstract PolicyConfiguration createPolicyConfiguration() throws Exception;

   protected abstract void putJaccInService(PolicyConfiguration pc, DeploymentUnit unit);

   /**

    * Return the container injection handler collection. If not specified(null)

    * a default handler collection will be created.

    * @return the injection handler collection to use, null if the container

    *    should use a default setup.

    */

   protected Collection<InjectionHandler<Environment>> getHandlers()

   {

      return null;

   }

   /**

    * Create all EJB containers and Persistence Units

    * The only things that should be initialized is metadata that does not need access to any

    * other deployment.  This is because we want the entire EAR to be initialized so that we do not

    * have to guess on dependencies MBean names.  This is because of the silly scoping rules for persistence units

    * and EJBs.

    *

    * @throws Exception

    */

   public void create() throws Exception

   {

      try

      {

         long start = System.currentTimeMillis();

         //pc = createPolicyConfiguration();

         deploy();

         initializePersistenceUnits();

        

         registerDeployment();

         log.debug("EJB3 deployment time took: " + (System.currentTimeMillis() - start));

      }

      catch (Exception e)

      {

         try

         {

            destroy();

         }

         catch (Exception ignored)

         {

            // ignore

         }

         throw e;

      }

   }

  

   protected void reinitialize() throws Exception

   {

      initializePersistenceUnits();

      reinitialize = false;

   }

   public void start() throws Exception

   {

      try

      {

         if (reinitialize)

            reinitialize();

        

         startPersistenceUnits();

         for (Object o : ejbContainers.values())

         {

            Container con = (Container) o;

            processEJBContainerMetadata(con);

         }

         for (Object o : ejbContainers.values())

         {

            Container con = (Container) o;

            registerEJBContainer(con);

         }

         //putJaccInService(pc, unit);

      }

      catch (Exception ex)

      {

         try

         {

            stop();

            destroy();

         }

         catch (Exception ignored)

         {

         }

         throw ex;

      }

   }

  

   public void stop() //throws Exception

   {

      for (ObjectName on : ejbContainers.keySet())

      {

         try

         {

            mbeanServer.unregisterMBean(on);

            kernelAbstraction.uninstall(on.getCanonicalName());

         }

         catch (Exception e)

         {

            log.debug("error trying to stop ejb container", e);

         }

      }

      stopPersistenceUnits();

     

      reinitialize = true;

   }

   protected void deploy() throws Exception

   {

      if(metaData == null || !metaData.isMetadataComplete())

      {

         Ejb3HandlerFactory factory = Ejb3HandlerFactory.getInstance(this);

         // Scan and deploy

         if (unit.getUrl() != null)

            deployUrl(factory);

  

         // Deploy any classes that have been explicitly marked

         if (unit.getClasses() != null)

         {

            for (Class explicit : unit.getClasses())

            {

               if (explicit.isAnnotationPresent(Entity.class))

               {

                  continue;

               }

               String name = explicit.getName().replace('.', '/') + ".class";

               InputStream stream = explicit.getClassLoader().getResourceAsStream(name);

               deployElement(stream, factory, initialContext);

            }

         }

      }

      // Deploy the beans from the descriptor

      deployBeansFromLib(initialContext);

   }

   protected void deployUrl(Ejb3HandlerFactory factory) throws Exception

   {

      // make sure we are not deploying ejbs from client jar

      List<VirtualFile> clientDescriptors = unit.getResources(new org.jboss.ejb3.ClientDescriptorFileFilter());

      if (clientDescriptors.size() > 0)

         return;

      InitialContext ctx = initialContext;

      // need to look into every entry in the archive to see if anybody has tags

      // defined.

      List<VirtualFile> classes = unit.getResources(new org.jboss.ejb3.ClassFileFilter());

      for (VirtualFile classFile : classes)

      {

         InputStream stream = classFile.openStream();

         deployElement(stream, factory, ctx);

      }

   }

   protected void deployElement(InputStream stream, Ejb3HandlerFactory factory, InitialContext ctx) throws Exception

   {

      DataInputStream dstream = new DataInputStream(new BufferedInputStream(stream));

      ClassFile cf = null;

      try

      {

         cf = new ClassFile(dstream);

      }

      finally

      {

         dstream.close();

         stream.close();

      }

      deployElement(factory, cf, ctx);

   }

   protected void deployBeansFromLib(InitialContext ctx) throws Exception

   {

      JBossMetaData dd = getMetaData();

      if (dd != null)

      {

         Ejb3DescriptorHandler handler = new Ejb3DescriptorHandler(this, dd);

         handler.setCtxProperties(unit.getJndiProperties());

         Map<String, Container> localContainers = new HashMap<String, Container>();

         Iterator<Container> containerIterator = ejbContainers.values().iterator();

         while (containerIterator.hasNext())

         {

            Container container = containerIterator.next();

            localContainers.put(container.getEjbName(), container);

         }

         List<Container> containers = handler.getContainers(this, localContainers);

         for (Container con : containers)

         {

            // EJBContainer has finished with all metadata initialization from XML files and such.

            // this is really a hook to do some processing after XML has been set up and before

            // and processing of dependencies and such.

            ((EJBContainer) con).instantiated();

            this.ejbContainers.put(con.getObjectName(), con);

            Ejb3Registry.register(con);

         }

      }

   }

   protected void deployElement(Ejb3HandlerFactory factory, ClassFile cf, InitialContext ctx) throws Exception

   {

      Ejb3Handler handler = factory.createHandler(cf);

      handler.setCtxProperties(unit.getJndiProperties());

      if (handler.isEjb() || handler.isJBossBeanType())

      {

         List<Container> containers = handler.getContainers(cf, this);

         for (Container con : containers)

         {

            // EJBContainer has finished with all metadata initialization from XML files and such.

            // this is really a hook to do some processing after XML has been set up and before

            // and processing of dependencies and such.

            try

            {

               ((EJBContainer) con).instantiated();

               this.ejbContainers.put(con.getObjectName(), con);

               Ejb3Registry.register(con);

            }

            catch (Throwable t)

            {

               throw new DeploymentException(

                     "Error creating ejb container " + con.getEjbName() + ": " + t.getMessage(), t);

            }

         }

      }

   }

   protected void initializePersistenceUnits() throws Exception

   {

      hasEntities = persistenceUnitsMetaData != null && processPersistenceUnits;

      if (!hasEntities)

         return;

      if (unit.getClasses() != null)

      {

         for (Class<?> explicit : unit.getClasses())

         {

            if (explicit.isAnnotationPresent(Entity.class))

            {

               explicitEntityClasses.add(explicit.getName());

            }

         }

      }

      // scope the unitName if this is an ejb archive

      // todo revert to this: List<PersistenceMetadata> persistenceMetadata = PersistenceXmlLoader.deploy(persistenceXmlUrl, new HashMap(), new EJB3DTDEntityResolver());

      List<PersistenceUnitMetaData> pumds = persistenceUnitsMetaData.getPersistenceUnits();

      for (PersistenceUnitMetaData metaData : pumds)

      {

         String earShortName = deploymentScope == null ? null : deploymentScope.getShortName();

         boolean isScoped = ejbContainers.size() > 0;

         Map<String, String> properties = metaData.getProperties();

         if (properties == null)

         {

            properties = new HashMap<String, String>();

            metaData.setProperties(properties);

         }

         // Ensure 2nd level cache entries are segregated from other deployments

         String cache_prefix = properties.get(SecondLevelCacheUtil.HIBERNATE_CACHE_REGION_PREFIX);

         if (cache_prefix == null)

         {

            // Create a region_prefix for the 2nd level cache to ensure

            // deployments are segregated

            String jarName = isScoped ? unit.getShortName() : null;

            cache_prefix = SecondLevelCacheUtil.createCacheRegionPrefix(earShortName, jarName, metaData.getName());

            properties.put(SecondLevelCacheUtil.HIBERNATE_CACHE_REGION_PREFIX, cache_prefix);

         }

         PersistenceUnitDeployment deployment = new PersistenceUnitDeployment(initialContext, this, explicitEntityClasses, metaData, earShortName, unit.getShortName(), isScoped);

         PersistenceUnitRegistry.register(deployment);

         persistenceUnitDeployments.add(deployment);

      }

   }

   public abstract DependencyPolicy createDependencyPolicy(JavaEEComponent component);

   protected void startPersistenceUnits()

   {

      if (persistenceUnitDeployments == null)

         return;

      for (PersistenceUnitDeployment entityDeployment : persistenceUnitDeployments)

      {

         if (entityDeployment != null)

         {

            DependencyPolicy policy = createDependencyPolicy(entityDeployment);

            entityDeployment.addDependencies(policy);

            kernelAbstraction.install(entityDeployment.getKernelName(), policy, unit, entityDeployment);

         }

      }

   }

   protected void stopPersistenceUnits()

   {

      if (persistenceUnitDeployments == null)

         return;

      for (PersistenceUnitDeployment entityDeployment : persistenceUnitDeployments)

      {

         try

         {

            PersistenceUnitRegistry.unregister(entityDeployment);

            if (entityDeployment != null)

            {

               kernelAbstraction.uninstall(entityDeployment.getKernelName());

            }

         }

         catch (Exception e)

         {

            log.debug("error trying to shut down persistence unit", e);

         }

      }

     

      persistenceUnitDeployments = new ArrayList<PersistenceUnitDeployment>();

   }

  

   public void destroy() //throws Exception

   {

      try

      {

         undeploy();

        

         unregisterDeployment();

      }

      catch (Exception e)

      {

         log.debug("error trying to destroy ejb deployment", e);

      }

   }

   private void undeploy()

   {

      for (Container container : ejbContainers.values())

      {

         Ejb3Registry.unregister(container);

      }

   }

   //   /**

   //    * Get the jndi name of a message destination.

   //    *

   //    * @param name   the name of the message destination

   //    * @return       the jndi name

   //    */

   //   private String getMessageDestination(String name)

   //   {

   //      EjbJarDD dd;

   //      // FIXME: Why isn't dd stored somewhere?

   //      try

   //      {

   //         dd = EjbJarDDObjectFactory.parse(getDeploymentUnit().getEjbJarXml());

   //         dd = JBossDDObjectFactory.parse(getDeploymentUnit().getJbossXml(), dd);

   //      }

   //      catch(IOException e)

   //      {

   //         throw new RuntimeException(e);

   //      }

   //      catch(JBossXBException e)

   //      {

   //         throw new RuntimeException(e);

   //      }

   //

   //      AssemblyDescriptor ad = dd.getAssemblyDescriptor();

   //      if(ad == null)

   //         throw new IllegalStateException("No assembly descriptor found in '" + getName() + "'");

   //      MessageDestination md = ad.findMessageDestination(name);

   //      if(md == null)

   //         throw new IllegalStateException("No message destination '" + name + "' found in '" + getName() + "'");

   //      String jndiName = md.getJndiName();

   //      if(jndiName == null)

   //         throw new IllegalStateException("No jndi name specified for message destination '" + name + "' in '" + getName() + "'");

   //      return jndiName;

   //   }

   public String resolveMessageDestination(String link)

   {

      //      // FIXME: this is a copy of DeploymentEjbResolver

      //      int hashIndex = link.indexOf('#');

      //      if (hashIndex != -1)

      //      {

      //         if (deploymentScope == null)

      //         {

      //            log.warn("ejb link '" + link + "' is relative, but no deployment scope found");

      //            return null;

      //         }

      //         String relativePath = link.substring(0, hashIndex);

      //         Ejb3Deployment dep = deploymentScope.findRelativeDeployment(relativePath);

      //         if (dep == null)

      //         {

      //            log.warn("can't find a deployment for path '" + relativePath + "' of ejb link '" + link + "'");

      //            return null;

      //         }

      //         String name = link.substring(hashIndex + 1);

      //         return dep.getMessageDestination(name);

      //      }

      //      return getMessageDestination(link);

      return messageDestinationResolver.resolveMessageDestination(link);

   }

   protected String resolvePersistenceUnitSupplier(String persistenceUnitName)

   {

      return persistenceUnitDependencyResolver.resolvePersistenceUnitSupplier(deploymentUnit, persistenceUnitName);

   }

  

   public MessageDestinationResolver getMessageDestinationResolver()

   {

      return messageDestinationResolver;

   }

   /**

    * Do not call, for use in Ejb3Handler.

    *

    * @param <B>

    * @param ejbName

    * @param enterpriseBeanMetaDataClass

    * @return

    */

   protected <B extends JBossEnterpriseBeanMetaData> B getEnterpriseBeanMetaData(String ejbName, Class<B> enterpriseBeanMetaDataClass)

   {

      if(metaData == null)

         return null;

     

      JBossEnterpriseBeanMetaData result = metaData.getEnterpriseBean(ejbName);

     

      // FIXME: EJBTHREE-1227: temporary workaround for JBCTS-756

      // see also org.jboss.ejb3.metadata.JBossSessionGenericWrapper

      if(result instanceof JBossGenericBeanMetaData)

      {

         log.warn("FIXME: EJBTHREE-1227: JBossGenericBeanMetaData found for '" + ejbName + "' instead of " + enterpriseBeanMetaDataClass);

         if(enterpriseBeanMetaDataClass.equals(JBossSessionBeanMetaData.class))

         {

            result = new JBossSessionGenericWrapper((JBossGenericBeanMetaData) result);

         }

         else if(enterpriseBeanMetaDataClass.equals(JBossMessageDrivenBeanMetaData.class))

         {

            result = new JBossMessageDrivenBeanGenericWrapper((JBossGenericBeanMetaData) result);

         }

         else

         {

            throw new IllegalStateException("Can't find a generic bean meta data wrapper for " + enterpriseBeanMetaDataClass);

         }

      }

     

      return ClassHelper.cast(enterpriseBeanMetaDataClass, result);

   }

  

   /**

    * Get the meta data associated with this deployment or null if none.

    *

    * @return   meta data or null

    */

   public JBossMetaData getMetaData()

   {

      return metaData;

   }

   public String getName()

   {

      return unit.getShortName();

   }

}