CdiClientInjectionManager.java
/*
* Copyright (c) 2021, 2024 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the
* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
* version 2 with the GNU Classpath Exception, which is available at
* https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
*/
package org.glassfish.jersey.inject.weld.internal.managed;
import jakarta.enterprise.context.spi.CreationalContext;
import jakarta.enterprise.inject.spi.Bean;
import jakarta.enterprise.inject.spi.BeanManager;
import jakarta.enterprise.inject.spi.InjectionTarget;
import jakarta.ws.rs.RuntimeType;
import org.glassfish.jersey.inject.weld.internal.bean.ClassBean;
import org.glassfish.jersey.inject.weld.internal.bean.JerseyBean;
import org.glassfish.jersey.inject.weld.internal.bean.SupplierClassBean;
import org.glassfish.jersey.inject.weld.internal.inject.InitializableBinding;
import org.glassfish.jersey.inject.weld.internal.inject.InitializableInstanceBinding;
import org.glassfish.jersey.inject.weld.internal.inject.InitializableSupplierInstanceBinding;
import org.glassfish.jersey.inject.weld.internal.inject.MatchableBinding;
import org.glassfish.jersey.inject.weld.internal.injector.JerseyClientCreationalContext;
import org.glassfish.jersey.inject.weld.internal.injector.JerseyInjectionTarget;
import org.glassfish.jersey.innate.inject.InternalBinding;
import org.glassfish.jersey.innate.inject.Bindings;
import org.glassfish.jersey.innate.inject.ClassBinding;
import org.glassfish.jersey.innate.inject.InstanceBinding;
import org.glassfish.jersey.innate.inject.SupplierClassBinding;
import org.glassfish.jersey.innate.inject.SupplierInstanceBinding;
import org.glassfish.jersey.internal.inject.Binding;
import org.glassfish.jersey.internal.inject.InjectionManager;
import org.jboss.weld.contexts.CreationalContextImpl;
import java.util.IdentityHashMap;
import java.util.Map;
/**
* Each Client Runtime has a unique CdiClientInjectionManager, which passes proper {@link InternalBinding} to the Weld.
*/
public final class CdiClientInjectionManager extends CdiInjectionManager {
private Map<InitializableInstanceBinding, InitializableInstanceBinding> clientInstanceBindings = new IdentityHashMap<>();
private Map<InitializableSupplierInstanceBinding, InitializableSupplierInstanceBinding> clientSupplierInstanceBindings
= new IdentityHashMap<>();
private Map<SupplierClassBinding, SupplierClassBinding> clientSupplierClassBinding = new IdentityHashMap<>();
private Map<ClassBinding, ClassBinding> clientClassBinding = new IdentityHashMap<>();
CdiClientInjectionManager(BeanManager beanManager, BinderRegisterExtension.MergedBindings bindings) {
super(beanManager, bindings, RuntimeType.CLIENT);
}
@Override
public void register(Binding binding) {
if (InstanceBinding.class.isInstance(binding)) {
// final Collection<Binding> preBindings = getBindings().getBindings();
// MatchableBinding.Matching<InitializableInstanceBinding> matching = MatchableBinding.Matching.noneMatching();
// for (Binding preBinding : preBindings) {
// if (InitializableInstanceBinding.class.isInstance(preBinding)) {
// matching = matching.better(((InitializableInstanceBinding) preBinding).matches((InstanceBinding) binding));
// if (matching.isBest()) {
// break;
// }
// }
// }
MatchableBinding.Matching<InitializableInstanceBinding> matching;
// MatchableBinding.Matching<InitializableInstanceBinding> matching =
// findPrebinding(InitializableInstanceBinding.class, binding /*, RuntimeType.CLIENT*/);
// if (matching.matches()) {
// matching.getBinding().init(((InstanceBinding) binding).getService());
// return;
// }
matching = findPrebinding(InitializableInstanceBinding.class, binding/*, RuntimeType.SERVER*/);
if (matching.matches()) {
final InitializableInstanceBinding clone = matching.getBinding().clone();
clone.init(((InstanceBinding) binding).getService());
clientInstanceBindings.put(matching.getBinding(), clone);
} else {
if (!userBindings.init((InstanceBinding<?>) binding)) {
throw new IllegalStateException("Not initialized " + ((InstanceBinding<?>) binding).getService());
}
}
} else if (SupplierInstanceBinding.class.isInstance(binding)) {
// final Collection<Binding> preBindings = getBindings().getBindings();
// MatchableBinding.Matching<InitializableSupplierInstanceBinding> matching = MatchableBinding.Matching.noneMatching();
// for (Binding preBinding : preBindings) {
// if (InitializableSupplierInstanceBinding.class.isInstance(preBinding)) {
// matching = matching.better(((InitializableSupplierInstanceBinding) preBinding).matchesContracts(binding));
// if (matching.isBest()) {
// break;
// }
// }
// }
MatchableBinding.Matching<InitializableSupplierInstanceBinding> matching =
findPrebinding(InitializableSupplierInstanceBinding.class, binding);
if (matching.matches()) {
final InitializableSupplierInstanceBinding clone = matching.getBinding().clone();
clone.init(((SupplierInstanceBinding) binding).getSupplier());
clientSupplierInstanceBindings.put(matching.getBinding(), clone);
} else {
throw new IllegalStateException("Not initialized " + ((SupplierInstanceBinding<?>) binding).getSupplier());
}
// } else if (SupplierClassBinding.class.isInstance(binding)) {
// final Collection<Binding> preBindings = getBindings().getBindings();
// BindingMatching.Matching<SupplierClassBinding> matching = BindingMatching.Matching.noneMatching();
// for (Binding preBinding : preBindings) {
// if (SupplierClassBinding.class.isInstance(preBinding)) {
// matching = matching.better(BindingMatching.matches(preBinding, binding));
// if (matching.isBest()) {
// break;
// }
// }
// }
// if (matching.matches()) {
// final SupplierClassBinding clone = BindingCloner.clone((SupplierClassBinding) binding);
// clientSupplierClassBinding.put(matching.getBinding(), clone);
// } else {
// throw new IllegalStateException("Not initialized " + ((SupplierInstanceBinding<?>) binding).getSupplier());
// }
} else if (ClassBinding.class.isInstance(binding)) {
ClassBinding prebinding = findClassBinding(binding.getImplementationType());
if (prebinding == null) {
//User registered provider
if (!userBindings.init((ClassBinding<?>) binding)) {
throw new IllegalStateException("Unexpected");
}
clientClassBinding.put(prebinding, (ClassBinding) binding);
}
}
}
public InitializableInstanceBinding getInjectionManagerBinding(InitializableInstanceBinding binding) {
InitializableInstanceBinding clientBinding = clientInstanceBindings.get(binding);
return clientBinding != null ? clientBinding : binding;
}
public InitializableSupplierInstanceBinding getInjectionManagerBinding(InitializableSupplierInstanceBinding binding) {
InitializableSupplierInstanceBinding clientBinding = clientSupplierInstanceBindings.get(binding);
return clientBinding != null ? clientBinding : binding;
}
public SupplierClassBinding getInjectionManagerBinding(SupplierClassBinding binding) {
SupplierClassBinding clientBinding = clientSupplierClassBinding.get(binding);
return clientBinding != null ? clientBinding : binding;
}
public ClassBinding getInjectionManagerBinding(ClassBinding binding) {
ClassBinding clientBinding = clientClassBinding.get(binding);
return clientBinding != null ? clientBinding : binding;
}
@Override
protected void lockContext() {
ContextSafe.lockContext(this);
}
@Override
protected void unlockContext() {
ContextSafe.unlock(this);
}
@Override
public void shutdown() {
clientInstanceBindings.clear();
}
@Override
protected <T> CreationalContext<T> createCreationalContext(Bean<T> bean) {
final CreationalContext<T> ctx = new JerseyClientCreationalContext<T>(
(CreationalContextImpl<T>) super.createCreationalContext(bean)).setInjectionManager(this);
return ctx;
}
@Override
public void completeRegistration() throws IllegalStateException {
register(Bindings.service(this).to(InjectionManager.class));
}
@Override
protected boolean isRuntimeTypeBean(Bean<?> bean) {
return !JerseyBean.class.isInstance(bean)
|| ((JerseyBean) bean).getRutimeType() == RuntimeType.CLIENT
|| clientInstanceBindings.containsKey(((JerseyBean<?>) bean).getBinding())
|| clientSupplierInstanceBindings.containsKey(((JerseyBean<?>) bean).getBinding())
|| isTwoFoldInstantiator((JerseyBean) bean);
}
private static boolean isTwoFoldInstantiator(JerseyBean bean) {
InjectionTarget injectionTarget = null;
if (ClassBean.class.isInstance(bean)) {
injectionTarget = ((ClassBean) bean).getInjectionTarget();
} else if (SupplierClassBean.class.isInstance(bean)) {
injectionTarget = ((SupplierClassBean) bean).getInjectionTarget();
}
if (injectionTarget != null) {
if (JerseyInjectionTarget.class.isInstance(injectionTarget)) {
JerseyInjectionTarget jerseyInjectionTarget = (JerseyInjectionTarget) injectionTarget;
return jerseyInjectionTarget.getTwofoldInstantiator().hasOptionalConstructorInjectionPoint();
}
}
return false;
}
private boolean isInit(JerseyBean bean) {
if (InitializableBinding.class.isInstance(bean.getBinding())) {
return ((InitializableBinding) bean.getBinding()).isInit();
} else {
return true;
}
}
@Override
public RuntimeType getRuntimeType() {
return RuntimeType.CLIENT;
}
}