[webbeans-commits] Webbeans SVN: r2956 - in ri/trunk/impl/src/main/java/org/jboss/webbeans: bootstrap and 1 other directory.

[webbeans-commits at lists.jboss.org]

Author: pete.muir at jboss.org
Date: 2009-07-02 15:14:00 -0400 (Thu, 02 Jul 2009)
New Revision: 2956

Added:
   ri/trunk/impl/src/main/java/org/jboss/webbeans/BeanIdStore.java
Modified:
   ri/trunk/impl/src/main/java/org/jboss/webbeans/bootstrap/WebBeansBootstrap.java
Log:
Add BeanIdStore service

Added: ri/trunk/impl/src/main/java/org/jboss/webbeans/BeanIdStore.java
===================================================================
--- ri/trunk/impl/src/main/java/org/jboss/webbeans/BeanIdStore.java	                        (rev 0)
+++ ri/trunk/impl/src/main/java/org/jboss/webbeans/BeanIdStore.java	2009-07-02 19:14:00 UTC (rev 2956)
@@ -0,0 +1,119 @@
+/*
+ * JBoss, Home of Professional Open Source
+ * Copyright 2008, Red Hat Middleware LLC, and individual contributors
+ * by the @authors tag. See the copyright.txt in the distribution for a
+ * full listing of individual contributors.
+ *
+ * Licensed 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.jboss.webbeans;
+
+import java.util.concurrent.atomic.AtomicInteger;
+
+import javax.enterprise.inject.spi.Bean;
+
+import org.jboss.webbeans.bootstrap.api.Service;
+
+import com.google.common.collect.BiMap;
+import com.google.common.collect.HashBiMap;
+
+/**
+ * Application wide bean identifier service which allows a serializable
+ * reference to a bean to be obtained, and the bean object to be got for a given
+ * id. Note that this allows a Bean object to be loaded regardless of the bean's
+ * accessiblity from the current module, and should not be abused as a way to
+ * ignore accessibility rules enforced during resolution.
+ * 
+ * @author Pete Muir
+ * 
+ */
+public class BeanIdStore implements Service
+{
+   
+   private static class BeanHolder<T>
+   {
+      public static <T>  BeanHolder<T> of(Bean<T> bean, BeanManagerImpl manager)
+      {
+         return new BeanHolder<T>(bean, manager);
+      }
+      
+      private final Bean<T> bean;
+      private final BeanManagerImpl manager;
+      
+      public BeanHolder(Bean<T> bean, BeanManagerImpl manager)
+      {
+         this.bean = bean;
+         this.manager = manager;
+      }
+
+      public Bean<T> getBean()
+      {
+         return bean;
+      }
+      
+      public BeanManagerImpl getManager()
+      {
+         return manager;
+      }
+      
+      @Override
+      public int hashCode()
+      {
+         return bean.hashCode();
+      }
+      
+      @Override
+      public boolean equals(Object obj)
+      {
+         if (obj instanceof BeanHolder)
+         {
+            return (this.bean.equals(((BeanHolder<?>) obj).getBean()));
+         }
+         else
+         {
+            return false;
+         }
+      }
+   }
+   
+   private final BiMap<Integer, BeanHolder<?>> beans;
+   private final AtomicInteger idGenerator;
+   
+   public BeanIdStore()
+   {
+      this.idGenerator = new AtomicInteger(0);
+      this.beans = HashBiMap.create();
+   }
+   
+   public <T> Bean<T> get(Integer id)
+   {
+      return (Bean<T>) beans.get(id).getBean();
+   }
+   
+   public boolean contains(Integer id)
+   {
+      return beans.containsKey(id);
+   }
+   
+   public Integer get(Bean<?> bean, BeanManagerImpl manager)
+   {
+      if (beans.inverse().containsKey(bean))
+      {
+         return beans.inverse().get(bean);
+      }
+      else
+      {
+         Integer id = idGenerator.incrementAndGet();
+         beans.put(id, BeanHolder.of(bean, manager));
+         return id;
+      }
+   }
+}


Property changes on: ri/trunk/impl/src/main/java/org/jboss/webbeans/BeanIdStore.java
___________________________________________________________________
Name: svn:mime-type
   + text/plain

Modified: ri/trunk/impl/src/main/java/org/jboss/webbeans/bootstrap/WebBeansBootstrap.java
===================================================================
--- ri/trunk/impl/src/main/java/org/jboss/webbeans/bootstrap/WebBeansBootstrap.java	2009-07-02 18:46:59 UTC (rev 2955)
+++ ri/trunk/impl/src/main/java/org/jboss/webbeans/bootstrap/WebBeansBootstrap.java	2009-07-02 19:14:00 UTC (rev 2956)
@@ -23,6 +23,7 @@
 import javax.enterprise.inject.spi.BeforeShutdown;
 import javax.enterprise.inject.spi.Extension;
 
+import org.jboss.webbeans.BeanIdStore;
 import org.jboss.webbeans.BeanManagerImpl;
 import org.jboss.webbeans.CurrentManager;
 import org.jboss.webbeans.DefinitionException;
@@ -214,6 +215,7 @@
       getServices().add(TypeStore.class, new TypeStore());
       getServices().add(ClassTransformer.class, new ClassTransformer(getServices().get(TypeStore.class)));
       getServices().add(MetaAnnotationStore.class, new MetaAnnotationStore(getServices().get(ClassTransformer.class)));
+      getServices().add(BeanIdStore.class, new BeanIdStore());
    }
    
    public BeanManagerImpl getManager()