OWB-1157 ensure to use OwbBean instances whenever possible
git-svn-id: https://svn.apache.org/repos/asf/openwebbeans/trunk@1765934 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/webbeans-impl/src/main/java/org/apache/webbeans/container/BeanManagerImpl.java b/webbeans-impl/src/main/java/org/apache/webbeans/container/BeanManagerImpl.java
index 56064e4..1e65bf1 100644
--- a/webbeans-impl/src/main/java/org/apache/webbeans/container/BeanManagerImpl.java
+++ b/webbeans-impl/src/main/java/org/apache/webbeans/container/BeanManagerImpl.java
@@ -214,6 +214,10 @@
private static Map<Class<? extends Annotation>, Boolean> isScopeTypeNormalCache =
new ConcurrentHashMap<Class<? extends Annotation>, Boolean>();
+ /**
+ * Map to be able to lookup always 3rd party beans when user does lookups with custom beans.
+ */
+ private Map<Bean<?>, Bean<?>> thirdPartyMapping = new HashMap<>();
/**
* Creates a new {@link BeanManager} instance.
@@ -374,6 +378,7 @@
}
addPassivationInfo(bean);
deploymentBeans.add(bean);
+ thirdPartyMapping.put(newBean, bean);
}
return this;
@@ -703,13 +708,18 @@
* {@inheritDoc}
*/
@Override
- public Object getReference(Bean<?> bean, Type beanType, CreationalContext<?> creationalContext)
+ public Object getReference(Bean<?> providedBean, Type beanType, CreationalContext<?> creationalContext)
{
- Asserts.assertNotNull(bean, "bean parameter");
+ Asserts.assertNotNull(providedBean, "bean parameter");
Context context = null;
Object instance = null;
+ Bean<?> bean = !OwbBean.class.isInstance(providedBean) ? thirdPartyMapping.get(providedBean) : providedBean;
+ if (bean == null) // more than unlikely but still possible and not invalid (user could create new instance of bean each time, not forbidden)
+ {
+ bean = providedBean;
+ }
if (bean instanceof SerializableBean)
{
bean = ((SerializableBean)bean).getBean();
diff --git a/webbeans-impl/src/test/java/org/apache/webbeans/test/bean/ThirdPartyBeanLookupTest.java b/webbeans-impl/src/test/java/org/apache/webbeans/test/bean/ThirdPartyBeanLookupTest.java
new file mode 100644
index 0000000..23bbb24
--- /dev/null
+++ b/webbeans-impl/src/test/java/org/apache/webbeans/test/bean/ThirdPartyBeanLookupTest.java
@@ -0,0 +1,161 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you 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.apache.webbeans.test.bean;
+
+import org.apache.webbeans.annotation.DefaultLiteral;
+import org.apache.webbeans.test.AbstractUnitTest;
+import org.junit.Test;
+
+import javax.enterprise.context.ApplicationScoped;
+import javax.enterprise.context.spi.CreationalContext;
+import javax.enterprise.event.Observes;
+import javax.enterprise.inject.spi.AfterBeanDiscovery;
+import javax.enterprise.inject.spi.AfterDeploymentValidation;
+import javax.enterprise.inject.spi.Bean;
+import javax.enterprise.inject.spi.BeanManager;
+import javax.enterprise.inject.spi.CDI;
+import javax.enterprise.inject.spi.Extension;
+import javax.enterprise.inject.spi.InjectionPoint;
+import javax.inject.Inject;
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Type;
+import java.util.Collections;
+import java.util.Set;
+
+import static org.junit.Assert.assertEquals;
+
+public class ThirdPartyBeanLookupTest extends AbstractUnitTest
+{
+ @Inject
+ private Ext ext;
+ @Inject
+ private Some some;
+
+ @Test
+ public void areEquals()
+ {
+ addExtension(new Ext());
+ startContainer(Collections.<Class<?>>emptyList(), Collections.<String>emptyList(), true);
+ assertEquals(ext.getInstance().getVal(), some.getVal());
+
+ final BeanManager bm = CDI.current().getBeanManager();
+ final Bean<?> bean = bm.resolve(bm.getBeans(Some.class));
+ final Some instance = Some.class.cast(bm.getReference(bean, bean.getBeanClass(), null));
+ assertEquals(instance.getVal(), some.getVal());
+ }
+
+ public static class Ext implements Extension
+ {
+ private SomeBean bean;
+ private Some instance;
+
+ void add(@Observes final AfterBeanDiscovery abf)
+ {
+ bean = new SomeBean();
+ abf.addBean(bean);
+ }
+
+ void lookup(@Observes final AfterDeploymentValidation adv, final BeanManager bm)
+ {
+ instance = Some.class.cast(bm.getReference(bean, bean.getBeanClass(), null));
+ }
+
+ public Some getInstance()
+ {
+ return instance;
+ }
+ }
+
+ private static class SomeBean implements Bean<Some>
+ {
+ @Override
+ public Set<InjectionPoint> getInjectionPoints()
+ {
+ return Collections.emptySet();
+ }
+
+ @Override
+ public Class<?> getBeanClass()
+ {
+ return Some.class;
+ }
+
+ @Override
+ public boolean isNullable()
+ {
+ return false;
+ }
+
+ @Override
+ public Some create(final CreationalContext<Some> context)
+ {
+ return new Some();
+ }
+
+ @Override
+ public void destroy(final Some instance, final CreationalContext<Some> context)
+ {
+ // no-op
+ }
+
+ @Override
+ public Set<Type> getTypes()
+ {
+ return Collections.<Type>singleton(Some.class);
+ }
+
+ @Override
+ public Set<Annotation> getQualifiers()
+ {
+ return Collections.<Annotation>singleton(DefaultLiteral.INSTANCE);
+ }
+
+ @Override
+ public Class<? extends Annotation> getScope()
+ {
+ return ApplicationScoped.class;
+ }
+
+ @Override
+ public String getName()
+ {
+ return null;
+ }
+
+ @Override
+ public Set<Class<? extends Annotation>> getStereotypes()
+ {
+ return Collections.emptySet();
+ }
+
+ @Override
+ public boolean isAlternative()
+ {
+ return false;
+ }
+ }
+
+ public static class Some {
+ private final int val = System.identityHashCode(this);
+
+ public int getVal() {
+ return val;
+ }
+ }
+}