How to use

JBossJpaServices

in

org.jboss.weld.integration.persistence

private PersistenceUnitDeployment lookupPersistenceUnitDeployment(String unitName)
{
 if (unitName == null)
 {
   throw new IllegalArgumentException("unitName is null");
 }
 String beanName = getPersistenceUnitSupplier(topLevelDeploymentUnit, persistenceUnitDependencyResolver, unitName);
 if (beanName == null)
 {
   throw new IllegalStateException("No persistence unit available for " + unitName);
 }
 return jbossEjb.lookupPersistenceUnitDeployment(beanName);
}

@SuppressWarnings({"deprecation"})
private EntityManagerFactory resolvePersistenceUnit(String unitName)
{
 PersistenceUnitDeployment deployment = lookupPersistenceUnitDeployment(unitName);
 ManagedEntityManagerFactory managedFactory = deployment.getManagedFactory();
 return new InjectedEntityManagerFactory(managedFactory);
}

public EntityManager resolvePersistenceContext(InjectionPoint injectionPoint)
{
 if (!injectionPoint.getAnnotated().isAnnotationPresent(PersistenceContext.class))
 {
   throw new IllegalArgumentException("No @PersistenceContext annotation found on injection point " + injectionPoint);
 }
 if (injectionPoint.getMember() instanceof Method && ((Method) injectionPoint.getMember()).getParameterTypes().length != 1)
 {
   throw new IllegalArgumentException("Injection point represents a method which doesn't follow JavaBean conventions (must have exactly one parameter) " + injectionPoint);
 }
 try
 {
   return resolvePersistenceUnit((injectionPoint.getAnnotated().getAnnotation(PersistenceContext.class).unitName())).createEntityManager();
 }
 catch (IllegalStateException e)
 {
   throw new IllegalStateException("Unable to resolve persistence context for " + injectionPoint);
 }
}

public EntityManagerFactory resolvePersistenceUnit(InjectionPoint injectionPoint)
{
 if (!injectionPoint.getAnnotated().isAnnotationPresent(PersistenceUnit.class))
 {
   throw new IllegalArgumentException("No @PersistenceUnit annotation found on injection point " + injectionPoint);
 }
 if (injectionPoint.getMember() instanceof Method && ((Method) injectionPoint.getMember()).getParameterTypes().length != 1)
 {
   throw new IllegalArgumentException("Injection point represents a method which doesn't follow JavaBean conventions (must have exactly one parameter) " + injectionPoint);
 }
 try
 {
   return resolvePersistenceUnit((injectionPoint.getAnnotated().getAnnotation(PersistenceUnit.class).unitName()));
 }
 catch (IllegalStateException e)
 {
   throw new IllegalStateException("Unable to resolve persistence context for " + injectionPoint);
 }
}

private EntityManagerFactory resolvePersistenceUnit(String unitName)
{
 if (unitName == null)
 {
   throw new IllegalArgumentException("unitName is null");
 }
 String beanName = getPersistenceUnitSupplier(topLevelDeploymentUnit, persistenceUnitDependencyResolver, unitName);
 if (beanName == null)
 {
   throw new IllegalStateException("No persistence unit available for " + unitName);
 }
 PersistenceUnitDeployment deployment = jbossEjb.lookupPersistenceUnitDeployment(beanName);
 ManagedEntityManagerFactory managedFactory = deployment.getManagedFactory();
 return new InjectedEntityManagerFactory(managedFactory);
}

public EntityManagerFactory resolvePersistenceUnit(InjectionPoint injectionPoint)
{
 if (!injectionPoint.getAnnotated().isAnnotationPresent(PersistenceUnit.class))
 {
   throw new IllegalArgumentException("No @PersistenceUnit annotation found on injection point " + injectionPoint);
 }
 if (injectionPoint.getMember() instanceof Method && ((Method) injectionPoint.getMember()).getParameterTypes().length != 1)
 {
   throw new IllegalArgumentException("Injection point represents a method which doesn't follow JavaBean conventions (must have exactly one parameter) " + injectionPoint);
 }
 try
 {
   return resolvePersistenceUnit((injectionPoint.getAnnotated().getAnnotation(PersistenceUnit.class).unitName()));
 }
 catch (IllegalStateException e)
 {
   throw new IllegalStateException("Unable to resolve persistence context for " + injectionPoint);
 }
}

@SuppressWarnings({"deprecation"})
public EntityManager resolvePersistenceContext(InjectionPoint injectionPoint)
{
 if (!injectionPoint.getAnnotated().isAnnotationPresent(PersistenceContext.class))
 {
   throw new IllegalArgumentException("No @PersistenceContext annotation found on injection point " + injectionPoint);
 }
 if (injectionPoint.getMember() instanceof Method && ((Method) injectionPoint.getMember()).getParameterTypes().length != 1)
 {
   throw new IllegalArgumentException("Injection point represents a method which doesn't follow JavaBean conventions (must have exactly one parameter) " + injectionPoint);
 }
 try
 {
   String persistenceUnitName = injectionPoint.getAnnotated().getAnnotation(PersistenceContext.class).unitName();
   return new TransactionScopedEntityManager(lookupPersistenceUnitDeployment(persistenceUnitName).getManagedFactory());
 }
 catch (IllegalStateException e)
 {
   throw new IllegalStateException("Unable to resolve persistence context for " + injectionPoint);
 }
}

  private static String getPersistenceUnitSupplier(DeploymentUnit deploymentUnit, PersistenceUnitDependencyResolver persistenceUnitDependencyResolver, String persistenceUnitName)
  {
   if ((deploymentUnit.getAttachment(AttachmentNames.PROCESSED_METADATA, JBossMetaData.class) != null && deploymentUnit.getAttachment(JBossMetaData.class).isEJB3x()) || (deploymentUnit.getAttachment(JBossWebMetaData.class) != null))
   {
     try
     {
      return persistenceUnitDependencyResolver.resolvePersistenceUnitSupplier(deploymentUnit, persistenceUnitName);
     }
     catch (IllegalArgumentException e)
     {
      // No-op, means we can't find the PU in this DU
     }
   }
   for (DeploymentUnit child : deploymentUnit.getChildren())
   {
     String beanName = getPersistenceUnitSupplier(child, persistenceUnitDependencyResolver, persistenceUnitName);
     if (beanName != null)
     {
      return beanName;
     }
   }
   return null;
  }
}

private static String getPersistenceUnitSupplier(DeploymentUnit deploymentUnit, PersistenceUnitDependencyResolver persistenceUnitDependencyResolver, String persistenceUnitName)
{
 if ((deploymentUnit.getAttachment(AttachmentNames.PROCESSED_METADATA, JBossMetaData.class) != null && deploymentUnit.getAttachment(JBossMetaData.class).isEJB3x()) || (deploymentUnit.getAttachment(JBossWebMetaData.class) != null))
 {
   try
   {
    return persistenceUnitDependencyResolver.resolvePersistenceUnitSupplier(deploymentUnit, persistenceUnitName);
   }
   catch (IllegalArgumentException e)
   {
    // No-op, means we can't find the PU in this DU
   }
 }
 for (DeploymentUnit child : deploymentUnit.getChildren())
 {
   String beanName = getPersistenceUnitSupplier(child, persistenceUnitDependencyResolver, persistenceUnitName);
   if (beanName != null)
   {
    return beanName;
   }
 }
 return null;
}