[seam-commits] Seam SVN: r11656 - in modules/trunk/persistence/src/main/java: org and 2 other directories.

[seam-commits at lists.jboss.org]

Author: sboscarine
Date: 2009-11-23 21:15:55 -0500 (Mon, 23 Nov 2009)
New Revision: 11656

Added:
   modules/trunk/persistence/src/main/java/org/
   modules/trunk/persistence/src/main/java/org/seamframework/
   modules/trunk/persistence/src/main/java/org/seamframework/persistence/
   modules/trunk/persistence/src/main/java/org/seamframework/persistence/PersistenceContextExtension.java
Log:
initial commit

Added: modules/trunk/persistence/src/main/java/org/seamframework/persistence/PersistenceContextExtension.java
===================================================================
--- modules/trunk/persistence/src/main/java/org/seamframework/persistence/PersistenceContextExtension.java	                        (rev 0)
+++ modules/trunk/persistence/src/main/java/org/seamframework/persistence/PersistenceContextExtension.java	2009-11-24 02:15:55 UTC (rev 11656)
@@ -0,0 +1,202 @@
+package org.seamframework.persistence;
+
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Type;
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.Set;
+
+import javax.enterprise.context.ApplicationScoped;
+import javax.enterprise.context.spi.CreationalContext;
+import javax.enterprise.event.Observes;
+import javax.enterprise.inject.Alternative;
+import javax.enterprise.inject.Any;
+import javax.enterprise.inject.Default;
+import javax.enterprise.inject.spi.AfterBeanDiscovery;
+import javax.enterprise.inject.spi.AnnotatedField;
+import javax.enterprise.inject.spi.Bean;
+import javax.enterprise.inject.spi.BeanManager;
+import javax.enterprise.inject.spi.Extension;
+import javax.enterprise.inject.spi.InjectionPoint;
+import javax.enterprise.inject.spi.ProcessProducer;
+import javax.enterprise.inject.spi.Producer;
+import javax.enterprise.util.AnnotationLiteral;
+import javax.inject.Qualifier;
+import javax.persistence.EntityManager;
+import javax.persistence.EntityManagerFactory;
+import javax.persistence.Persistence;
+import javax.persistence.PersistenceContext;
+
+/**
+ * Support for managed persistence contexts in Java SE environment.
+ * 
+ * @author Gavin King
+ * 
+ */
+public class PersistenceContextExtension implements Extension
+{
+   public PersistenceContextExtension()
+   {
+      System.out.println("\n\n" + this.getClass() + " was constructed");
+   }
+
+   private Bean<EntityManagerFactory> emfBean;
+
+   /**
+    * For @PersistenceContext producer fields, make a bean for the EMF, then
+    * wrap the producer CDI creates, to get the EM from the EMF bean we made
+    * instead of trying to get it from the Java EE component environment.
+    */
+   void processProducer(@Observes ProcessProducer<?, EntityManager> pp, final BeanManager bm)
+   {
+      if (pp.getAnnotatedMember().isAnnotationPresent(PersistenceContext.class))
+      {
+         if (emfBean == null)
+         {
+            AnnotatedField<?> field = (AnnotatedField<?>) pp.getAnnotatedMember();
+            final String unitName = field.getAnnotation(PersistenceContext.class).unitName();
+            final Class<?> module = field.getJavaMember().getDeclaringClass();
+            final Set<Annotation> qualifiers = new HashSet<Annotation>();
+            for (Annotation ann : field.getAnnotations())
+            {
+               Class<? extends Annotation> annotationType = ann.annotationType();
+               // if ( bm.isQualifier(annotationType)) {
+               if (annotationType.isAnnotationPresent(Qualifier.class))
+               { // work around bug in Weld
+                  qualifiers.add(ann);
+               }
+            }
+            if (qualifiers.isEmpty())
+            {
+               qualifiers.add(new AnnotationLiteral<Default>()
+               {
+
+                  /** default value. Added only to suppress compiler warnings. */
+                  private static final long serialVersionUID = 1L;
+               });
+            }
+            qualifiers.add(new AnnotationLiteral<Any>()
+            {
+
+               /** default value. Added only to suppress compiler warnings. */
+               private static final long serialVersionUID = 1L;
+            });
+            final boolean alternative = field.isAnnotationPresent(Alternative.class);
+            final Set<Type> types = new HashSet<Type>()
+            {
+               /** default value. Added only to suppress compiler warnings. */
+               private static final long serialVersionUID = 1L;
+
+               {
+                  add(EntityManagerFactory.class);
+                  add(Object.class);
+               }
+            };
+
+            // create a bean for the EMF
+            emfBean = new Bean<EntityManagerFactory>()
+            {
+               public Set<Type> getTypes()
+               {
+                  return types;
+               }
+
+               public Class<? extends Annotation> getScope()
+               {
+                  return ApplicationScoped.class;
+               }
+
+               public EntityManagerFactory create(CreationalContext<EntityManagerFactory> ctx)
+               {
+                  return Persistence.createEntityManagerFactory(unitName);
+               }
+
+               public void destroy(EntityManagerFactory emf, CreationalContext<EntityManagerFactory> ctx)
+               {
+                  emf.close();
+                  ctx.release();
+               }
+
+               public Class<?> getBeanClass()
+               {
+                  return module;
+               }
+
+               public Set<InjectionPoint> getInjectionPoints()
+               {
+                  // return Collections.EMPTY_SET;
+                  return Collections.emptySet();
+               }
+
+               public String getName()
+               {
+                  return null;
+               }
+
+               public Set<Annotation> getQualifiers()
+               {
+                  return qualifiers;
+               }
+
+               public Set<Class<? extends Annotation>> getStereotypes()
+               {
+                  // return Collections.EMPTY_SET;
+                  return Collections.emptySet();
+
+               }
+
+               public boolean isAlternative()
+               {
+                  return alternative;
+               }
+
+               public boolean isNullable()
+               {
+                  return false;
+               }
+            };
+
+         }
+         else
+         {
+            throw new RuntimeException("Only one EMF per application is supported");
+         }
+
+         Producer<EntityManager> producer = new Producer<EntityManager>()
+         {
+
+            public Set<InjectionPoint> getInjectionPoints()
+            {
+               // return Collections.EMPTY_SET;
+               return Collections.emptySet();
+
+            }
+
+            public EntityManager produce(CreationalContext<EntityManager> ctx)
+            {
+               return getFactory(ctx).createEntityManager();
+            }
+
+            private EntityManagerFactory getFactory(CreationalContext<EntityManager> ctx)
+            {
+               return (EntityManagerFactory) bm.getReference(emfBean, EntityManagerFactory.class, ctx);
+            }
+
+            public void dispose(EntityManager em)
+            {
+               if (em.isOpen()) // work around what I suspect is a bug in Weld
+                  em.close();
+            }
+         };
+         pp.setProducer(producer);
+      }
+   }
+
+   /**
+    * Register the EMF bean with the container.
+    */
+   void afterBeanDiscovery(@Observes AfterBeanDiscovery abd)
+   {
+      abd.addBean(emfBean);
+   }
+}
\ No newline at end of file