Initial commit. Woot!

This commit is contained in:
JesseBrault0709 2024-05-03 12:01:45 +02:00
commit 03d1adf981
43 changed files with 1868 additions and 0 deletions

10
build.gradle Normal file
View File

@ -0,0 +1,10 @@
plugins {
id 'GroowtConventions'
id 'java-library'
}
dependencies {
api libs.jakarta.inject
compileOnlyApi libs.jetbrains.anotations
implementation libs.slf4j.api, libs.groovy
}

View File

@ -0,0 +1,197 @@
package groowt.util.di;
import jakarta.inject.Inject;
import org.jetbrains.annotations.Nullable;
import java.lang.reflect.*;
import java.util.*;
import static groowt.util.di.ObjectFactoryUtil.toTypes;
// TODO: maybe inject fields
public abstract class AbstractInjectingObjectFactory implements ObjectFactory {
protected record CachedInjectConstructor<T>(Class<T> clazz, Constructor<T> constructor) {}
protected record CachedNonInjectConstructor<T>(
Class<T> clazz,
Constructor<T> constructor,
Class<?>[] paramTypes
) {}
private final Map<Class<?>, Constructor<?>[]> cachedAllConstructors = new HashMap<>();
private final Collection<CachedInjectConstructor<?>> cachedInjectConstructors = new ArrayList<>();
private final Collection<CachedNonInjectConstructor<?>> cachedNonInjectConstructors = new ArrayList<>();
private final Map<Class<?>, Collection<Method>> cachedSetters = new HashMap<>();
private final Map<Method, Parameter> cachedSetterParameters = new HashMap<>();
@SuppressWarnings("unchecked")
private <T> @Nullable Constructor<T> findCachedInjectConstructor(Class<T> clazz) {
for (final CachedInjectConstructor<?> cachedConstructor : this.cachedInjectConstructors) {
if (clazz.equals(cachedConstructor.clazz())) {
return (Constructor<T>) cachedConstructor.constructor();
}
}
return null;
}
/**
* @implNote If overridden, please cache any found inject constructors using {@link #putCachedInjectConstructor}.
*
* @param clazz the {@link Class} in which to search for an <code>{@literal @}Inject</code> annotated constructor.
* @return the inject constructor, or {@code null} if none found.
* @param <T> the type of the class
*/
@SuppressWarnings("unchecked")
protected <T> @Nullable Constructor<T> findInjectConstructor(Class<T> clazz) {
final Constructor<T> cachedInjectConstructor = this.findCachedInjectConstructor(clazz);
if (cachedInjectConstructor != null) {
return cachedInjectConstructor;
}
final Constructor<?>[] constructors = this.cachedAllConstructors.computeIfAbsent(clazz, Class::getConstructors);
final List<Constructor<?>> injectConstructors = Arrays.stream(constructors)
.filter(constructor -> constructor.isAnnotationPresent(Inject.class))
.toList();
if (injectConstructors.size() > 1) {
// one day maybe support multiple inject constructors
throw new UnsupportedOperationException("Cannot have more than one @Inject constructor in class: " + clazz);
} else if (injectConstructors.size() == 1) {
final Constructor<T> injectConstructor = (Constructor<T>) injectConstructors.getFirst();
this.putCachedInjectConstructor(new CachedInjectConstructor<>(clazz, injectConstructor));
return injectConstructor;
} else {
return null;
}
}
protected final void putCachedInjectConstructor(CachedInjectConstructor<?> cached) {
this.cachedInjectConstructors.add(cached);
}
@SuppressWarnings("unchecked")
private <T> @Nullable Constructor<T> findCachedNonInjectConstructor(Class<T> clazz, Class<?>[] paramTypes) {
for (final CachedNonInjectConstructor<?> cachedConstructor : this.cachedNonInjectConstructors) {
if (clazz.equals(cachedConstructor.clazz()) && Arrays.equals(cachedConstructor.paramTypes(), paramTypes)) {
return (Constructor<T>) cachedConstructor.constructor();
}
}
return null;
}
/**
* @implNote If overridden, please cache any found non-inject constructors using {@link #putCachedNonInjectConstructor}.
*
* @param clazz the {@link Class} in which to search for a constructor which does not have an <code>{@literal @}Inject</code>
* annotation
* @param constructorArgs the given constructor args
* @return the found non-inject constructor appropriate for the given constructor args, or {@code null} if no
* such constructor exists
* @param <T> the type
*/
@SuppressWarnings("unchecked")
protected <T> @Nullable Constructor<T> findNonInjectConstructor(Class<T> clazz, Object[] constructorArgs) {
final Class<?>[] types = toTypes(constructorArgs);
final Constructor<T> cachedConstructor = this.findCachedNonInjectConstructor(clazz, types);
if (cachedConstructor != null) {
return cachedConstructor;
}
final Constructor<?>[] constructors = this.cachedAllConstructors.computeIfAbsent(clazz, Class::getConstructors);
for (Constructor<?> constructor : constructors) {
if (Arrays.equals(constructor.getParameterTypes(), types)) {
final Constructor<T> found = (Constructor<T>) constructor;
this.putCachedNonInjectConstructor(new CachedNonInjectConstructor<>(clazz, found, types));
return found;
}
}
return null;
}
protected final void putCachedNonInjectConstructor(CachedNonInjectConstructor<?> cached) {
this.cachedNonInjectConstructors.add(cached);
}
/**
* @implNote Please call {@code super.findConstructor()} first, and then implement custom
* constructor finding logic. If the custom logic finds a constructor, please cache it
* using either {@link #putCachedNonInjectConstructor} or {@link #putCachedInjectConstructor}.
*/
protected <T> Constructor<T> findConstructor(Class<T> clazz, Object[] args) {
final Constructor<T> injectConstructor = this.findInjectConstructor(clazz);
if (injectConstructor != null) {
return injectConstructor;
}
final Constructor<T> nonInjectConstructor = this.findNonInjectConstructor(clazz, args);
if (nonInjectConstructor != null) {
return nonInjectConstructor;
}
throw new RuntimeException("Could not find an appropriate constructor for " + clazz.getName()
+ " with args " + Arrays.toString(toTypes(args))
);
}
protected Collection<Method> getAllInjectSetters(Class<?> clazz) {
final Method[] allMethods = clazz.getMethods();
final Collection<Method> injectSetters = new ArrayList<>();
for (final var method : allMethods) {
if (
method.isAnnotationPresent(Inject.class)
&& method.getName().startsWith("set")
&& !Modifier.isStatic(method.getModifiers())
&& method.getParameterCount() == 1
) {
injectSetters.add(method);
}
}
return injectSetters;
}
protected Collection<Method> getCachedSettersFor(Object target) {
return this.cachedSetters.computeIfAbsent(target.getClass(), this::getAllInjectSetters);
}
protected Parameter getCachedInjectParameter(Method setter) {
return this.cachedSetterParameters.computeIfAbsent(setter, s -> {
if (s.getParameterCount() != 1) {
throw new IllegalArgumentException("Setter " + s.getName() + " has a parameter count other than one (1)!");
}
return s.getParameters()[0];
});
}
protected void injectSetter(Object target, Method setter) {
try {
setter.invoke(target, this.getSetterInjectArg(target.getClass(), setter, this.getCachedInjectParameter(setter)));
} catch (InvocationTargetException | IllegalAccessException e) {
throw new RuntimeException(e);
}
}
protected void injectSetters(Object target) {
this.getCachedSettersFor(target).forEach(setter -> this.injectSetter(target, setter));
}
/**
* {@inheritDoc}
*/
@Override
public <T> T createInstance(Class<T> clazz, Object... constructorArgs) {
final Constructor<T> constructor = this.findConstructor(clazz, constructorArgs);
final Object[] allArgs = this.createArgs(constructor, constructorArgs);
try {
final T instance = constructor.newInstance(allArgs);
this.injectSetters(instance);
return instance;
} catch (InvocationTargetException | IllegalAccessException | InstantiationException e) {
throw new RuntimeException(e); // In the future, we might have an option to ignore exceptions
}
}
protected abstract Object[] createArgs(Constructor<?> constructor, Object[] constructorArgs);
protected abstract Object getSetterInjectArg(Class<?> targetType, Method setter, Parameter toInject);
}

View File

@ -0,0 +1,113 @@
package groowt.util.di;
import groowt.util.di.filters.FilterHandler;
import groowt.util.di.filters.IterableFilterHandler;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
import static groowt.util.di.RegistryObjectFactoryUtil.orElseSupply;
public abstract class AbstractRegistryObjectFactory extends AbstractInjectingObjectFactory implements RegistryObjectFactory {
public static abstract class AbstractBuilder<T extends DefaultRegistryObjectFactory> implements Builder<T> {
private final Collection<FilterHandler<?, ?>> filterHandlers = new ArrayList<>();
private final Collection<IterableFilterHandler<?, ?>> iterableFilterHandlers = new ArrayList<>();
private final Registry registry;
private @Nullable RegistryObjectFactory parent;
public AbstractBuilder(Registry registry) {
this.registry = registry;
}
public AbstractBuilder() {
this.registry = new DefaultRegistry();
}
protected Registry getRegistry() {
return this.registry;
}
protected Collection<FilterHandler<?, ?>> getFilterHandlers() {
return this.filterHandlers;
}
protected Collection<IterableFilterHandler<?, ?>> getIterableFilterHandlers() {
return this.iterableFilterHandlers;
}
protected @Nullable RegistryObjectFactory getParent() {
return this.parent;
}
@Override
public void configureRegistry(Consumer<? super Registry> configure) {
configure.accept(this.registry);
}
public void addFilterHandler(FilterHandler<?, ?> handler) {
this.filterHandlers.add(handler);
}
public void addIterableFilterHandler(IterableFilterHandler<?, ?> handler) {
this.iterableFilterHandlers.add(handler);
}
public void setParent(@Nullable RegistryObjectFactory parent) {
this.parent = parent;
}
}
protected final Registry registry;
@Nullable private final RegistryObjectFactory parent;
public AbstractRegistryObjectFactory(Registry registry, @Nullable RegistryObjectFactory parent) {
this.registry = registry;
this.parent = parent;
}
@Override
public void configureRegistry(Consumer<? super Registry> use) {
use.accept(this.registry);
}
@Override
public <A extends Annotation> @Nullable ScopeHandler<A> findScopeHandler(Class<A> scopeType) {
return this.registry.getScopeHandler(scopeType);
}
@Override
public <A extends Annotation> @Nullable QualifierHandler<A> findQualifierHandler(Class<A> qualifierType) {
return this.registry.getQualifierHandler(qualifierType);
}
protected final <T> Optional<T> findInParent(Function<? super RegistryObjectFactory, @Nullable T> finder) {
return this.parent != null ? Optional.ofNullable(finder.apply(this.parent)) : Optional.empty();
}
protected final <T> Optional<T> findInSelfOrParent(Function<? super RegistryObjectFactory, @Nullable T> finder) {
return orElseSupply(
finder.apply(this),
() -> this.parent != null ? finder.apply(this.parent) : null
);
}
protected final <T> T getInSelfOrParent(
Function<? super RegistryObjectFactory, @Nullable T> finder,
Supplier<? extends RuntimeException> exceptionSupplier
) {
return orElseSupply(
finder.apply(this),
() -> this.parent != null ? finder.apply(this.parent) : null
).orElseThrow(exceptionSupplier);
}
}

View File

@ -0,0 +1,3 @@
package groowt.util.di;
sealed public interface Binding<T> permits ClassBinding, ProviderBinding, SingletonBinding, LazySingletonBinding {}

View File

@ -0,0 +1,12 @@
package groowt.util.di;
import jakarta.inject.Provider;
import java.util.function.Supplier;
public interface BindingConfigurator<T> {
void to(Class<? extends T> target);
void toProvider(Provider<? extends T> provider);
void toSingleton(T target);
void toLazySingleton(Supplier<T> singletonSupplier);
}

View File

@ -0,0 +1,32 @@
package groowt.util.di;
import jakarta.inject.Provider;
import java.util.function.Consumer;
import java.util.function.Supplier;
public final class BindingUtil {
public static <T> Consumer<BindingConfigurator<T>> toClass(Class<? extends T> clazz) {
return bc -> bc.to(clazz);
}
public static <T> Consumer<BindingConfigurator<T>> toProvider(Provider<? extends T> provider) {
return bc -> bc.toProvider(provider);
}
public static <T> Consumer<BindingConfigurator<T>> toSingleton(T singleton) {
return bc -> bc.toSingleton(singleton);
}
public static <T> Consumer<BindingConfigurator<T>> toLazySingleton(Supplier<T> singletonSupplier) {
return bc -> bc.toLazySingleton(singletonSupplier);
}
public static <T> Consumer<BindingConfigurator<T>> toSelf() {
return bc -> {};
}
private BindingUtil() {}
}

View File

@ -0,0 +1,3 @@
package groowt.util.di;
public record ClassBinding<T>(Class<T> from, Class<? extends T> to) implements Binding<T> {}

View File

@ -0,0 +1,80 @@
package groowt.util.di;
import jakarta.inject.Named;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.Predicate;
public class DefaultNamedRegistryExtension implements NamedRegistryExtension {
protected static class NamedQualifierHandler implements QualifierHandler<Named> {
private final DefaultNamedRegistryExtension extension;
public NamedQualifierHandler(DefaultNamedRegistryExtension extension) {
this.extension = extension;
}
@Override
public <T> @Nullable Binding<T> handle(Named named, Class<T> dependencyClass) {
return this.extension.getBinding(
new SimpleKeyHolder<>(NamedRegistryExtension.class, dependencyClass, named.value())
);
}
}
protected final Map<String, Binding<?>> bindings = new HashMap<>();
protected final QualifierHandler<Named> qualifierHandler = this.getNamedQualifierHandler();
protected QualifierHandler<Named> getNamedQualifierHandler() {
return new NamedQualifierHandler(this);
}
@SuppressWarnings("unchecked")
@Override
public @Nullable <A extends Annotation> QualifierHandler<A> getQualifierHandler(Class<A> qualifierType) {
return Named.class.equals(qualifierType) ? (QualifierHandler<A>) this.qualifierHandler : null;
}
@Override
public Class<String> getKeyClass() {
return String.class;
}
@Override
public <B extends KeyBinder<String>, T> void bind(KeyHolder<B, ? extends String, T> keyHolder, Consumer<? super BindingConfigurator<T>> configure) {
final var configurator = new SimpleBindingConfigurator<>(keyHolder.type());
configure.accept(configurator);
this.bindings.put(keyHolder.key(), configurator.getBinding());
}
@SuppressWarnings("unchecked")
@Override
public @Nullable <B extends KeyBinder<String>, T> Binding<T> getBinding(KeyHolder<B, ? extends String, T> keyHolder) {
return (Binding<T>) this.bindings.getOrDefault(keyHolder.key(), null);
}
@Override
public <B extends KeyBinder<String>, T> void removeBinding(KeyHolder<B, ? extends String, T> keyHolder) {
this.bindings.remove(keyHolder.key());
}
@Override
public <B extends KeyBinder<String>, T> void removeBindingIf(KeyHolder<B, ? extends String, T> keyHolder, Predicate<? super Binding<T>> filter) {
final String key = keyHolder.key();
if (this.bindings.containsKey(key) && filter.test(this.getBinding(keyHolder))) {
this.bindings.remove(key);
}
}
@Override
public void clearAllBindings() {
this.bindings.clear();
}
}

View File

@ -0,0 +1,200 @@
package groowt.util.di;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Predicate;
public class DefaultRegistry implements Registry {
protected record ClassKeyBinding<T>(Class<T> key, Binding<T> binding) {}
protected final Collection<ClassKeyBinding<?>> classBindings = new ArrayList<>();
protected final Collection<RegistryExtension> extensions = new ArrayList<>();
@Override
public void removeBinding(Class<?> key) {
this.classBindings.removeIf(classKeyBinding -> classKeyBinding.key().equals(key));
}
@SuppressWarnings("unchecked")
@Override
public <T> void removeBindingIf(Class<T> key, Predicate<Binding<T>> filter) {
this.classBindings.removeIf(classKeyBinding ->
classKeyBinding.key().equals(key) && filter.test((Binding<T>) classKeyBinding.binding())
);
}
private <E extends RegistryExtension> List<E> getAllRegistryExtensions(Class<E> extensionType) {
return this.extensions.stream()
.filter(extension -> extensionType.isAssignableFrom(extension.getClass()))
.map(extensionType::cast)
.toList();
}
private <E extends RegistryExtension> E getOneRegistryExtension(Class<E> extensionType) {
final List<E> extensions = this.getAllRegistryExtensions(extensionType);
if (extensions.size() == 1) {
return extensions.getFirst();
} else if (extensions.isEmpty()) {
throw new IllegalArgumentException("There is no " + extensionType + " registered for this " + this);
} else {
throw new IllegalArgumentException("There is more than one " + extensionType + " registered for this " + this);
}
}
@Override
public void addExtension(RegistryExtension extension) {
final List<? extends RegistryExtension> existing = this.getAllRegistryExtensions(extension.getClass());
if (existing.isEmpty()) {
this.extensions.add(extension);
} else {
throw new IllegalArgumentException("There is already at least one " + extension.getClass() + " registered in " + this);
}
}
@Override
public <E extends RegistryExtension> E getExtension(Class<E> extensionType) {
return this.getOneRegistryExtension(extensionType);
}
@Override
public <E extends RegistryExtension> Collection<E> getExtensions(Class<E> extensionType) {
return this.getAllRegistryExtensions(extensionType);
}
@Override
public void removeExtension(RegistryExtension extension) {
this.extensions.remove(extension);
}
@Override
public @Nullable <A extends Annotation> QualifierHandler<A> getQualifierHandler(Class<A> qualifierType) {
final List<QualifierHandler<A>> handlers = new ArrayList<>();
for (final var extension : this.extensions) {
if (extension instanceof QualifierHandlerContainer handlerContainer) {
final var handler = handlerContainer.getQualifierHandler(qualifierType);
if (handler != null) {
handlers.add(handler);
}
}
}
if (handlers.isEmpty()) {
return null;
} else if (handlers.size() > 1) {
throw new RuntimeException("There is more than one QualifierHandler for " + qualifierType.getName());
} else {
return handlers.getFirst();
}
}
@Override
public @Nullable <A extends Annotation> ScopeHandler<A> getScopeHandler(Class<A> scopeType) {
final List<ScopeHandler<A>> handlers = new ArrayList<>();
for (final var extension : this.extensions) {
if (extension instanceof ScopeHandlerContainer handlerContainer) {
final var handler = handlerContainer.getScopeHandler(scopeType);
if (handler != null) {
handlers.add(handler);
}
}
}
if (handlers.isEmpty()) {
return null;
} else if (handlers.size() > 1) {
throw new RuntimeException("There is more than one ScopeHandler for " + scopeType.getName());
} else {
return handlers.getFirst();
}
}
@Override
public <T> void bind(Class<T> key, Consumer<? super BindingConfigurator<T>> configure) {
final var configurator = new SimpleBindingConfigurator<>(key);
configure.accept(configurator);
this.classBindings.add(new ClassKeyBinding<>(key, configurator.getBinding()));
}
@SuppressWarnings("unchecked")
@Override
public @Nullable <T> Binding<T> getBinding(Class<T> key) {
for (final var classKeyBinding : this.classBindings) {
if (key.isAssignableFrom(classKeyBinding.key())) {
return (Binding<T>) classKeyBinding.binding();
}
}
return null;
}
private KeyBinder<?> findKeyBinder(Class<?> keyClass) {
final List<KeyBinder<?>> binders = new ArrayList<>();
for (final var extension : this.extensions) {
if (extension instanceof KeyBinder<?> keyBinder && keyBinder.getKeyClass().isAssignableFrom(keyClass)) {
binders.add(keyBinder);
}
}
if (binders.isEmpty()) {
throw new IllegalArgumentException("There are no configured RegistryExtensions that can handle keys with type " + keyClass.getName());
} else if (binders.size() > 1) {
throw new IllegalArgumentException("There is more than one configured RegistryExtension that can handle keys with type " + keyClass.getName());
} else {
return binders.getFirst();
}
}
@SuppressWarnings("rawtypes")
protected final void withKeyBinder(KeyHolder<?, ?, ?> keyHolder, Consumer<KeyBinder> action) {
action.accept(this.findKeyBinder(keyHolder.key().getClass()));
}
@SuppressWarnings("rawtypes")
protected final <R> @Nullable R tapKeyBinder(
KeyHolder<?, ?, ?> keyHolder,
Function<KeyBinder, @Nullable R> function
) {
return function.apply(this.findKeyBinder(keyHolder.key().getClass()));
}
@SuppressWarnings("unchecked")
@Override
public <T> void bind(KeyHolder<?, ?, T> keyHolder, Consumer<? super BindingConfigurator<T>> configure) {
this.withKeyBinder(keyHolder, b -> b.bind(keyHolder, configure));
}
@SuppressWarnings("unchecked")
@Override
public @Nullable <T> Binding<T> getBinding(KeyHolder<?, ?, T> keyHolder) {
return this.tapKeyBinder(keyHolder, b -> b.getBinding(keyHolder));
}
@SuppressWarnings("unchecked")
@Override
public <T> void removeBinding(KeyHolder<?, ?, T> keyHolder) {
this.withKeyBinder(keyHolder, b -> b.removeBinding(keyHolder));
}
@SuppressWarnings("unchecked")
@Override
public <T> void removeBindingIf(
KeyHolder<?, ?, T> keyHolder,
Predicate<? super Binding<T>> filter
) {
this.withKeyBinder(keyHolder, b -> b.removeBindingIf(keyHolder, filter));
}
@Override
public void clearAllBindings() {
this.classBindings.clear();
for (final var extension : this.extensions) {
if (extension instanceof KeyBinder<?> keyBinder) {
keyBinder.clearAllBindings();
}
}
}
}

View File

@ -0,0 +1,313 @@
package groowt.util.di;
import groowt.util.di.filters.FilterHandler;
import groowt.util.di.filters.IterableFilterHandler;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.lang.annotation.Annotation;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.function.Supplier;
import static groowt.util.di.RegistryObjectFactoryUtil.*;
public class DefaultRegistryObjectFactory extends AbstractRegistryObjectFactory {
public static final class Builder extends AbstractRegistryObjectFactory.AbstractBuilder<DefaultRegistryObjectFactory> {
/**
* Creates a {@code Builder} initialized with a {@link DefaultRegistry}, which is in-turn configured with a
* {@link NamedRegistryExtension} and a {@link SingletonScopeHandler}.
*
* @return the builder
*/
public static Builder withDefaults() {
final var b = new Builder();
b.configureRegistry(r -> {
r.addExtension(new DefaultNamedRegistryExtension());
r.addExtension(new SingletonRegistryExtension(r));
});
return b;
}
/**
* @return a blank builder with a {@link Registry} from the given {@link Supplier}.
*/
public static Builder withRegistry(Supplier<? extends Registry> registrySupplier) {
return new Builder(registrySupplier.get());
}
/**
* @return a blank builder which will use {@link DefaultRegistry}.
*/
public static Builder blank() {
return new Builder();
}
private Builder(Registry registry) {
super(registry);
}
private Builder() {
super();
}
@Override
public DefaultRegistryObjectFactory build() {
return new DefaultRegistryObjectFactory(
this.getRegistry(),
this.getParent(),
this.getFilterHandlers(),
this.getIterableFilterHandlers()
);
}
}
private static final Object[] EMPTY_OBJECT_ARRAY = new Object[0];
private static final Logger logger = LoggerFactory.getLogger(DefaultRegistryObjectFactory.class); // leave it for the future!
private final Collection<FilterHandler<?, ?>> filterHandlers;
private final Collection<IterableFilterHandler<?, ?>> iterableFilterHandlers;
protected DefaultRegistryObjectFactory(
Registry registry,
@Nullable RegistryObjectFactory parent,
Collection<? extends FilterHandler<?, ?>> filterHandlers,
Collection<? extends IterableFilterHandler<?, ?>> iterableFilterHandlers
) {
super(registry, parent);
this.filterHandlers = new ArrayList<>(filterHandlers);
this.filterHandlers.forEach(handler -> checkIsValidFilter(handler.getAnnotationClass()));
this.iterableFilterHandlers = new ArrayList<>(iterableFilterHandlers);
this.iterableFilterHandlers.forEach(handler -> checkIsValidIterableFilter(handler.getAnnotationClass()));
}
/**
* Checks if the given parameter has any qualifier annotations; if it does,
* it delegates finding the desired object to the registered {@link QualifierHandler}.
*
* @param parameter the parameter
* @return the object returned from the {@code QualifierHandler}, or {@code null} if no qualifier
* is present or the {@code QualifierHandler} itself returns {@code null}.
*
* @throws RuntimeException if no {@code QualifierHandler} is registered for a qualifier annotation present on the
* given parameter, or if the handler itself throws an exception.
*/
@SuppressWarnings("unchecked")
protected final @Nullable Object tryQualifiers(Parameter parameter) {
final Class<?> paramType = parameter.getType();
final List<Annotation> qualifiers = RegistryObjectFactoryUtil.getQualifierAnnotations(parameter.getAnnotations());
if (qualifiers.size() > 1) {
throw new RuntimeException("Parameter " + parameter + " cannot have more than one Qualifier annotation.");
} else if (qualifiers.size() == 1) {
final Annotation qualifier = qualifiers.getFirst();
@SuppressWarnings("rawtypes")
final QualifierHandler handler = this.getInSelfOrParent(
f -> f.findQualifierHandler(qualifier.annotationType()),
() -> new RuntimeException("There is no configured QualifierHandler for " + qualifier.annotationType().getName())
);
final Binding<?> binding = handler.handle(qualifier, paramType);
if (binding != null) {
return this.handleBinding(binding, EMPTY_OBJECT_ARRAY);
}
}
// no Qualifier or the QualifierHandler didn't return a Binding
return null;
}
/**
* Checks the {@code resolvedArg} against all filters present on the given parameter.
*
* @param parameter the parameter
* @param resolvedArg the resolved argument
*
* @throws RuntimeException if the {@link FilterHandler} itself throws an exception.
*/
@SuppressWarnings({"unchecked", "rawtypes"})
protected final void checkFilters(Parameter parameter, Object resolvedArg) {
final Annotation[] allAnnotations = parameter.getAnnotations();
final Collection<Annotation> filterAnnotations = getFilterAnnotations(allAnnotations);
if (!filterAnnotations.isEmpty()) {
final Collection<FilterHandler<?, ?>> filtersForParamType = this.filterHandlers.stream()
.filter(filterHandler ->
filterHandler.getArgumentClass().isAssignableFrom(parameter.getType())
)
.toList();
for (final Annotation filterAnnotation : filterAnnotations) {
for (final FilterHandler<?, ?> filterHandler : filtersForParamType) {
if (filterAnnotation.annotationType().equals(filterHandler.getAnnotationClass())) {
// hopefully we've checked everything
((FilterHandler<Annotation, Object>) filterHandler).check(filterAnnotation, resolvedArg);
}
}
}
}
final Collection<Annotation> iterableFilterAnnotations = getIterableFilterAnnotations(allAnnotations);
if (!iterableFilterAnnotations.isEmpty() && resolvedArg instanceof Iterable iterable) {
for (final var annotation : iterableFilterAnnotations) {
this.iterableFilterHandlers.stream()
.filter(handler -> handler.getAnnotationClass().equals(annotation.annotationType()))
.forEach(handler -> {
((IterableFilterHandler<Annotation, Object>) handler).check(annotation, iterable);
});
}
}
}
protected final Object resolveInjectedArg(Parameter parameter) {
final Object qualifierProvidedArg = this.tryQualifiers(parameter);
if (qualifierProvidedArg != null) {
this.checkFilters(parameter, qualifierProvidedArg);
return qualifierProvidedArg;
} else {
final Object created = this.get(parameter.getType());
this.checkFilters(parameter, created);
return created;
}
}
protected final void resolveInjectedArgs(Object[] dest, Parameter[] params) {
for (int i = 0; i < params.length; i++) {
dest[i] = this.resolveInjectedArg(params[i]);
}
}
protected final void resolveGivenArgs(Object[] dest, Parameter[] params, Object[] givenArgs, int startIndex) {
for (int i = startIndex; i < dest.length; i++) {
final int resolveIndex = i - startIndex;
final Object arg = givenArgs[resolveIndex];
this.checkFilters(params[resolveIndex], arg);
dest[i] = arg;
}
}
// TODO: when there is a null arg, we lose the type. Therefore this algorithm breaks. Fix this.
@Override
protected Object[] createArgs(Constructor<?> constructor, Object[] givenArgs) {
final Class<?>[] paramTypes = constructor.getParameterTypes();
// check no arg
if (paramTypes.length == 0 && givenArgs.length == 0) {
// no args given, none needed, so return empty array
return EMPTY_OBJECT_ARRAY;
} else if (paramTypes.length == 0) { // implicit that givenArgs.length != 0
// zero expected, but got given args
throw new RuntimeException(
"Expected zero args for constructor " + constructor + " but received " + Arrays.toString(givenArgs)
);
} else if (givenArgs.length > paramTypes.length) {
// expected is more than zero, but received too many given
throw new RuntimeException(
"Too many args given for constructor " + constructor + "; received " + Arrays.toString(givenArgs)
);
}
final Parameter[] allParams = constructor.getParameters();
final Object[] resolvedArgs = new Object[allParams.length];
if (givenArgs.length == 0) {
// if no given args, then they are all injected
this.resolveInjectedArgs(resolvedArgs, allParams);
} else if (givenArgs.length == paramTypes.length) {
// all are given
this.resolveGivenArgs(resolvedArgs, allParams, givenArgs, 0);
} else {
// some are injected, some are given
// everything before (non-inclusive) is injected
// everything after (inclusive) is given
// ex: 1 inject, 1 given -> 2 (allParams) - 1 = 1
// ex: 0 inject, 1 given -> 1 - 1 = 0
final int firstGivenIndex = allParams.length - givenArgs.length;
final Parameter[] injectedParams = new Parameter[firstGivenIndex];
final Parameter[] givenParams = new Parameter[allParams.length - firstGivenIndex];
System.arraycopy(allParams, 0, injectedParams, 0, injectedParams.length);
System.arraycopy(allParams, firstGivenIndex, givenParams, 0, allParams.length - firstGivenIndex);
this.resolveInjectedArgs(resolvedArgs, injectedParams);
this.resolveGivenArgs(resolvedArgs, givenParams, givenArgs, firstGivenIndex);
}
return resolvedArgs;
}
@SuppressWarnings("unchecked")
private <T> T handleBinding(Binding<T> binding, Object[] constructorArgs) {
return switch (binding) {
case ClassBinding<T>(Class<T> ignored, Class<? extends T> to) -> {
final Annotation scopeAnnotation = getScopeAnnotation(to);
if (scopeAnnotation != null) {
final Class<? extends Annotation> scopeClass = scopeAnnotation.annotationType();
@SuppressWarnings("rawtypes")
final ScopeHandler scopeHandler = this.getInSelfOrParent(
f -> f.findScopeHandler(scopeClass),
() -> new RuntimeException("There is no configured ScopeHandler for " + scopeClass.getName())
);
final Binding<T> scopedBinding = scopeHandler.onScopedDependencyRequest(scopeAnnotation, to, this);
yield this.handleBinding(scopedBinding, constructorArgs);
} else {
yield this.createInstance(to, constructorArgs);
}
}
case ProviderBinding<T> providerBinding -> providerBinding.provider().get();
case SingletonBinding<T> singletonBinding -> singletonBinding.to();
case LazySingletonBinding<T> lazySingletonBinding -> lazySingletonBinding.singletonSupplier().get();
};
}
protected final <T> @Nullable Binding<T> searchRegistry(Class<T> from) {
return this.registry.getBinding(from);
}
protected @Nullable <T> T tryParent(Class<T> clazz, Object[] constructorArgs) {
return this.findInParent(f -> f.getOrNull(clazz, constructorArgs)).orElse(null);
}
@Override
protected Object getSetterInjectArg(Class<?> targetType, Method setter, Parameter toInject) {
return this.resolveInjectedArg(toInject);
}
/**
* {@inheritDoc}
*/
@Override
public <T> T get(Class<T> clazz, Object... constructorArgs) {
final Binding<T> binding = this.searchRegistry(clazz);
if (binding != null) {
return this.handleBinding(binding, constructorArgs);
}
final T parentResult = this.tryParent(clazz, constructorArgs);
if (parentResult != null) {
return parentResult;
} else {
throw new RuntimeException("No bindings for " + clazz + " with args " + Arrays.toString(constructorArgs) + ".");
}
}
/**
* {@inheritDoc}
*/
@Override
public <T> T getOrDefault(Class<T> clazz, T defaultValue, Object... constructorArgs) {
final Binding<T> binding = this.searchRegistry(clazz);
if (binding != null) {
return this.handleBinding(binding, constructorArgs);
}
final T parentResult = this.tryParent(clazz, constructorArgs);
return parentResult != null ? parentResult : defaultValue;
}
}

View File

@ -0,0 +1,10 @@
package groowt.util.di;
import java.util.Collection;
public interface ExtensionContainer {
void addExtension(RegistryExtension extension);
<E extends RegistryExtension> E getExtension(Class<E> extensionType);
<E extends RegistryExtension> Collection<E> getExtensions(Class<E> extensionType);
void removeExtension(RegistryExtension extension);
}

View File

@ -0,0 +1,16 @@
package groowt.util.di;
import org.jetbrains.annotations.Nullable;
import java.util.function.BiPredicate;
import java.util.function.Consumer;
import java.util.function.Predicate;
public interface KeyBinder<K> {
Class<K> getKeyClass();
<B extends KeyBinder<K>, T> void bind(KeyHolder<B, ? extends K, T> keyHolder, Consumer<? super BindingConfigurator<T>> configure);
<B extends KeyBinder<K>, T> @Nullable Binding<T> getBinding(KeyHolder<B, ? extends K, T> keyHolder);
<B extends KeyBinder<K>, T> void removeBinding(KeyHolder<B, ? extends K, T> keyHolder);
<B extends KeyBinder<K>, T> void removeBindingIf(KeyHolder<B, ? extends K, T> keyHolder, Predicate<? super Binding<T>> filter);
void clearAllBindings();
}

View File

@ -0,0 +1,7 @@
package groowt.util.di;
public interface KeyHolder<B extends KeyBinder<K>, K, T> {
Class<B> binderType();
Class<T> type();
K key();
}

View File

@ -0,0 +1,5 @@
package groowt.util.di;
import java.util.function.Supplier;
public record LazySingletonBinding<T>(Supplier<T> singletonSupplier) implements Binding<T> {}

View File

@ -0,0 +1,9 @@
package groowt.util.di;
public interface NamedRegistryExtension extends RegistryExtension, KeyBinder<String>, QualifierHandlerContainer {
static <T> KeyHolder<NamedRegistryExtension, String, T> named(String name, Class<T> type) {
return new SimpleKeyHolder<>(NamedRegistryExtension.class, type, name);
}
}

View File

@ -0,0 +1,61 @@
package groowt.util.di;
import org.jetbrains.annotations.Contract;
import java.util.function.Function;
/**
* An {@link ObjectFactory} is an object that can construct objects of given types.
*/
@FunctionalInterface
public interface ObjectFactory {
/**
* Create a new instance of the given {@code instanceType} with the given constructor args.
*
* @apiNote An implementation may provide a subclass of the given instance type,
* or it may directly instantiate the given type, if it is a class
* and it can determine the correct constructor from the given arguments.
* See individual implementation documentation for exact behavior.
*
* @implSpec It is up to individual implementations of {@link ObjectFactory} to determine how to
* select the appropriate constructor for the given type. The returned
* instance must be new and in a valid state.
*
* @param instanceType the {@link Class} of the desired type
* @param constructorArgs any arguments to pass to the constructor(s) of the class.
* @return the new instance
* @param <T> the desired type
*/
@Contract("_, _-> new")
<T> T createInstance(Class<T> instanceType, Object... constructorArgs);
/**
* Very similar to {@link #createInstance(Class, Object...)}, but catches any {@link RuntimeException}
* thrown by {@link #createInstance} and subsequently passes it to the given {@link Function}, returning
* instead the return value of the {@link Function}.
*
* @param instanceType the desired type of the created instance
* @param onException a {@link Function} to handle when an exception occurs and return a value nonetheless
* @param constructorArgs arguments to pass to the constructor
* @return the created instance
* @param <T> the desired type
*
* @throws RuntimeException if the given {@link Function} itself throws a RuntimeException
*
* @see #createInstance(Class, Object...)
*/
@Contract("_, _, _ -> new")
default <T> T createInstanceCatching(
Class<T> instanceType,
Function<? super RuntimeException, ? extends T> onException,
Object... constructorArgs
) {
try {
return this.createInstance(instanceType, constructorArgs);
} catch (RuntimeException runtimeException) {
return onException.apply(runtimeException);
}
}
}

View File

@ -0,0 +1,23 @@
package groowt.util.di;
import org.jetbrains.annotations.ApiStatus;
@ApiStatus.Internal
public final class ObjectFactoryUtil {
public static Class<?>[] toTypes(Object... objects) {
final Class<?>[] types = new Class<?>[objects.length];
for (int i = 0; i < objects.length; i++) {
final Object o = objects[i];
if (o != null) {
types[i] = o.getClass();
} else {
types[i] = Object.class;
}
}
return types;
}
private ObjectFactoryUtil() {}
}

View File

@ -0,0 +1,5 @@
package groowt.util.di;
import jakarta.inject.Provider;
public record ProviderBinding<T>(Class<T> to, Provider<? extends T> provider) implements Binding<T> {}

View File

@ -0,0 +1,10 @@
package groowt.util.di;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
@FunctionalInterface
public interface QualifierHandler<A extends Annotation> {
<T> @Nullable Binding<T> handle(A annotation, Class<T> dependencyClass);
}

View File

@ -0,0 +1,20 @@
package groowt.util.di;
import jakarta.inject.Qualifier;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
public interface QualifierHandlerContainer {
static void checkIsValidQualifier(Class<? extends Annotation> annotationClass) {
if (!annotationClass.isAnnotationPresent(Qualifier.class)) {
throw new IllegalArgumentException(
"The given qualifier annotation + " + annotationClass + " is itself not annotated with @Qualifier"
);
}
}
<A extends Annotation> @Nullable QualifierHandler<A> getQualifierHandler(Class<A> qualifierType);
}

View File

@ -0,0 +1,20 @@
package groowt.util.di;
import org.jetbrains.annotations.Nullable;
import java.util.function.Consumer;
import java.util.function.Predicate;
public interface Registry extends ExtensionContainer, QualifierHandlerContainer, ScopeHandlerContainer {
<T> void bind(Class<T> key, Consumer<? super BindingConfigurator<T>> configure);
@Nullable <T> Binding<T> getBinding(Class<T> key);
void removeBinding(Class<?> key);
<T> void removeBindingIf(Class<T> key, Predicate<Binding<T>> filter);
<T> void bind(KeyHolder<?, ?, T> keyHolder, Consumer<? super BindingConfigurator<T>> configure);
<T> @Nullable Binding<T> getBinding(KeyHolder<?, ?, T> keyHolder);
<T> void removeBinding(KeyHolder<?, ?, T> keyHolder);
<T> void removeBindingIf(KeyHolder<?, ?, T> keyHolder, Predicate<? super Binding<T>> filter);
void clearAllBindings();
}

View File

@ -0,0 +1,3 @@
package groowt.util.di;
public interface RegistryExtension {}

View File

@ -0,0 +1,99 @@
package groowt.util.di;
import groowt.util.di.filters.FilterHandler;
import groowt.util.di.filters.IterableFilterHandler;
import jakarta.inject.Provider;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
import java.util.function.Consumer;
/**
* A {@link RegistryObjectFactory} is an {@link ObjectFactory} that offers the ability
* to provide desired objects based on an instance of
* {@link Registry} to determine how to provide those objects.
*/
public interface RegistryObjectFactory extends ObjectFactory {
interface Builder<T extends RegistryObjectFactory> {
void configureRegistry(Consumer<? super Registry> configure);
void addFilterHandler(FilterHandler<?, ?> handler);
void addIterableFilterHandler(IterableFilterHandler<?, ?> handler);
T build();
}
void configureRegistry(Consumer<? super Registry> use);
<A extends Annotation> @Nullable ScopeHandler<A> findScopeHandler(Class<A> scopeType);
<A extends Annotation> @Nullable QualifierHandler<A> findQualifierHandler(Class<A> qualifierType);
/**
* Get an object with the desired type. How it is retrieved/created
* depends upon the {@link Binding} present in this {@link RegistryObjectFactory}'s held
* instances of {@link Registry}. The type of the {@link Binding} determines
* how the object is fetched:
*
* <ul>
* <li>{@link ClassBinding}: A new instance of the object is created using the given {@code constructorArgs}.</li>
* <li>{@link ProviderBinding}: An instance of the object is fetched from the bound {@link Provider}.
* Whether the instance is new or not depends on the {@link Provider}.</li>
* <li>{@link SingletonBinding}: The bound singleton object is returned.</li>
* </ul>
*
* @implNote If {@code constructorArgs} are provided
* and the {@link Binding} for the desired type is not a
* {@link ClassBinding}, the implementation should
* either throw an exception or log a warning at the least.
*
* @param clazz the {@link Class} of the desired type
* @param constructorArgs As in {@link #createInstance(Class, Object...)},
* the arguments which will be used to create the desired object
* if the {@link Binding} is a {@link ClassBinding}.
* @return an object of the desired type
* @param <T> the desired type
*
* @throws RuntimeException if there is no registered {@link Binding} or there is a problem
* fetching or constructing the object.
*/
<T> T get(Class<T> clazz, Object... constructorArgs);
/**
* Similarly to {@link #get(Class, Object...)}, fetches an object
* of the desired type, but does not throw if there is no registered {@link Binding}
* in any of the held instances of {@link Registry},
* and instead returns the given {@code defaultValue}.
*
* @param clazz the {@link Class} of the desired type
* @param defaultValue the defaultValue to return
* @param constructorArgs see {@link #get(Class, Object...)}
* @return an object of the desired type
* @param <T> the desired type
*
* @throws RuntimeException if there <em>is</em> a registered {@link Binding} and there is a problem
* fetching or constructing the object.
*
* @see #get(Class, Object...)
*/
<T> T getOrDefault(Class<T> clazz, T defaultValue, Object... constructorArgs);
/**
* Similar to {@link #getOrDefault(Class, Object, Object...)}, except that
* it returns null by default if there is no registered {@link Binding}.
*
* @param clazz the {@link Class} of the desired type
* @param constructorArgs see {@link RegistryObjectFactory#get(Class, Object...)}
* @return an object of the desired type
* @param <T> the desired type
*
* @see RegistryObjectFactory#get(Class, Object...)
* @see RegistryObjectFactory#getOrDefault(Class, Object, Object...)
*
* @throws RuntimeException if there <em>is</em> a registered {@code Binding} and there
* is a problem fetching or constructing the object.
*/
default <T> @Nullable T getOrNull(Class<T> clazz, Object... constructorArgs) {
return this.getOrDefault(clazz, null, constructorArgs);
}
}

View File

@ -0,0 +1,69 @@
package groowt.util.di;
import groowt.util.di.filters.Filter;
import groowt.util.di.filters.IterableFilter;
import jakarta.inject.Qualifier;
import jakarta.inject.Scope;
import org.jetbrains.annotations.ApiStatus;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import java.util.function.Supplier;
@ApiStatus.Internal
public final class RegistryObjectFactoryUtil {
private RegistryObjectFactoryUtil() {}
public static List<Annotation> getQualifierAnnotations(Annotation[] annotations) {
return Arrays.stream(annotations)
.filter(a -> a.annotationType().isAnnotationPresent(Qualifier.class))
.toList();
}
public static List<Annotation> getFilterAnnotations(Annotation[] annotations) {
return Arrays.stream(annotations)
.filter(a -> a.annotationType().isAnnotationPresent(Filter.class))
.toList();
}
public static List<Annotation> getIterableFilterAnnotations(Annotation[] annotations) {
return Arrays.stream(annotations)
.filter(a -> a.annotationType().isAnnotationPresent(IterableFilter.class))
.toList();
}
public static <T> Optional<T> orElseSupply(T first, Supplier<T> onNullFirst) {
return first != null ? Optional.of(first) : Optional.ofNullable(onNullFirst.get());
}
public static void checkIsValidFilter(Class<? extends Annotation> annotationClass) {
if (!annotationClass.isAnnotationPresent(Filter.class)) {
throw new IllegalArgumentException(
"The given filter annotation " + annotationClass.getName() + " is itself not annotated with @Filter"
);
}
}
public static void checkIsValidIterableFilter(Class<? extends Annotation> annotationClass) {
if (!annotationClass.isAnnotationPresent(IterableFilter.class)) {
throw new IllegalArgumentException(
"The given iterable filter annotation " + annotationClass.getName() + " is itself not annotated with @IterableFilter"
);
}
}
public static @Nullable Annotation getScopeAnnotation(Class<?> clazz) {
final List<Annotation> scopeAnnotations = Arrays.stream(clazz.getAnnotations())
.filter(annotation -> annotation.annotationType().isAnnotationPresent(Scope.class))
.toList();
if (scopeAnnotations.size() > 1) {
throw new RuntimeException(clazz.getName() + " has too many annotations that are themselves annotated with @Scope");
}
return scopeAnnotations.size() == 1 ? scopeAnnotations.getFirst() : null;
}
}

View File

@ -0,0 +1,8 @@
package groowt.util.di;
import java.lang.annotation.Annotation;
public interface ScopeHandler<A extends Annotation> {
<T> Binding<T> onScopedDependencyRequest(A annotation, Class<T> dependencyClass, RegistryObjectFactory objectFactory);
void reset();
}

View File

@ -0,0 +1,20 @@
package groowt.util.di;
import jakarta.inject.Scope;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
public interface ScopeHandlerContainer {
static void checkIsValidScope(Class<? extends Annotation> scope) {
if (!scope.isAnnotationPresent(Scope.class)) {
throw new IllegalArgumentException(
"The given scope annotation " + scope + " is itself not annotated with @Scope"
);
}
}
<A extends Annotation> @Nullable ScopeHandler<A> getScopeHandler(Class<A> scopeType);
}

View File

@ -0,0 +1,43 @@
package groowt.util.di;
import jakarta.inject.Provider;
import org.jetbrains.annotations.Nullable;
import java.util.function.Supplier;
public class SimpleBindingConfigurator<T> implements BindingConfigurator<T> {
private final Class<T> from;
private @Nullable Binding<T> binding;
public SimpleBindingConfigurator(Class<T> from) {
this.from = from;
}
public final Binding<T> getBinding() {
return this.binding != null
? this.binding
: new ClassBinding<>(this.from, this.from); // return SelfBinding in case we never called anything
}
@Override
public void to(Class<? extends T> target) {
this.binding = new ClassBinding<>(this.from, target);
}
@Override
public void toProvider(Provider<? extends T> provider) {
this.binding = new ProviderBinding<>(this.from, provider);
}
@Override
public void toSingleton(T target) {
this.binding = new SingletonBinding<>(target);
}
@Override
public void toLazySingleton(Supplier<T> singletonSupplier) {
this.binding = new LazySingletonBinding<>(singletonSupplier);
}
}

View File

@ -0,0 +1,4 @@
package groowt.util.di;
public record SimpleKeyHolder<B extends KeyBinder<K>, K, T>(Class<B> binderType, Class<T> type, K key)
implements KeyHolder<B, K, T> {}

View File

@ -0,0 +1,3 @@
package groowt.util.di;
public record SingletonBinding<T>(T to) implements Binding<T> {}

View File

@ -0,0 +1,22 @@
package groowt.util.di;
import jakarta.inject.Singleton;
import org.jetbrains.annotations.Nullable;
import java.lang.annotation.Annotation;
public class SingletonRegistryExtension implements RegistryExtension, ScopeHandlerContainer {
private final SingletonScopeHandler handler;
public SingletonRegistryExtension(Registry owner) {
this.handler = new SingletonScopeHandler(owner);
}
@SuppressWarnings("unchecked")
@Override
public @Nullable <A extends Annotation> ScopeHandler<A> getScopeHandler(Class<A> scopeType) {
return Singleton.class.isAssignableFrom(scopeType) ? (ScopeHandler<A>) this.handler : null;
}
}

View File

@ -0,0 +1,35 @@
package groowt.util.di;
import jakarta.inject.Singleton;
import static groowt.util.di.BindingUtil.toSingleton;
public final class SingletonScopeHandler implements ScopeHandler<Singleton> {
private final Registry owner;
public SingletonScopeHandler(Registry owner) {
this.owner = owner;
}
@Override
public <T> Binding<T> onScopedDependencyRequest(
Singleton annotation,
Class<T> dependencyClass,
RegistryObjectFactory objectFactory
) {
final Binding<T> potentialBinding = this.owner.getBinding(dependencyClass);
if (potentialBinding != null) {
return potentialBinding;
} else {
this.owner.bind(dependencyClass, toSingleton(objectFactory.createInstance(dependencyClass)));
return this.owner.getBinding(dependencyClass);
}
}
@Override
public void reset() {
throw new UnsupportedOperationException("Cannot reset the Singleton scope!");
}
}

View File

@ -0,0 +1,10 @@
package groowt.util.di.annotation;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.SOURCE)
@Target(ElementType.PARAMETER)
public @interface Given {}

View File

@ -0,0 +1,10 @@
package groowt.util.di.filters;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target({ ElementType.ANNOTATION_TYPE })
public @interface Filter {}

View File

@ -0,0 +1,39 @@
package groowt.util.di.filters;
import java.lang.annotation.Annotation;
import java.util.Objects;
import java.util.function.BiPredicate;
public interface FilterHandler<A extends Annotation, T> {
boolean check(A annotation, T arg);
Class<A> getAnnotationClass();
Class<T> getArgumentClass();
default FilterHandler<A, T> and(BiPredicate<A, T> and) {
Objects.requireNonNull(and);
return new SimpleFilterHandler<>(
(a, t) -> this.check(a, t) && and.test(a, t),
this.getAnnotationClass(),
this.getArgumentClass()
);
}
default FilterHandler<A, T> or(BiPredicate<A, T> or) {
Objects.requireNonNull(or);
return new SimpleFilterHandler<>(
(a, t) -> this.check(a, t) || or.test(a, t),
this.getAnnotationClass(),
this.getArgumentClass()
);
}
default FilterHandler<A, T> negate() {
return new SimpleFilterHandler<>(
(a, t) -> !this.check(a, t),
this.getAnnotationClass(),
this.getArgumentClass()
);
}
}

View File

@ -0,0 +1,35 @@
package groowt.util.di.filters;
import java.lang.annotation.*;
import java.util.function.BiPredicate;
import static groowt.util.di.filters.FilterUtil.isAssignableToAnyOf;
public final class FilterHandlers {
private FilterHandlers() {}
public static <A extends Annotation, T> FilterHandler<A, T> of(
Class<A> annotationClass,
Class<T> argClass,
BiPredicate<A, T> predicate
) {
return new SimpleFilterHandler<>(predicate, annotationClass, argClass);
}
@Filter
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.PARAMETER)
public @interface AllowTypes {
Class<?>[] value();
}
public static <T> FilterHandler<AllowTypes, T> getAllowsTypesFilterHandler(Class<T> targetType) {
return of(
AllowTypes.class,
targetType,
(annotation, target) -> isAssignableToAnyOf(target.getClass(), annotation.value())
);
}
}

View File

@ -0,0 +1,16 @@
package groowt.util.di.filters;
public final class FilterUtil {
private FilterUtil() {}
public static boolean isAssignableToAnyOf(Class<?> subject, Class<?>[] tests) {
for (final var test : tests) {
if (test.isAssignableFrom(subject)) {
return true;
}
}
return false;
}
}

View File

@ -0,0 +1,10 @@
package groowt.util.di.filters;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.ANNOTATION_TYPE)
public @interface IterableFilter {}

View File

@ -0,0 +1,8 @@
package groowt.util.di.filters;
import java.lang.annotation.Annotation;
public interface IterableFilterHandler<A extends Annotation, E> {
boolean check(A annotation, Iterable<E> iterable);
Class<A> getAnnotationClass();
}

View File

@ -0,0 +1,33 @@
package groowt.util.di.filters;
import java.lang.annotation.*;
import java.util.function.BiPredicate;
import static groowt.util.di.filters.FilterUtil.isAssignableToAnyOf;
public final class IterableFilterHandlers {
private IterableFilterHandlers() {}
public static <A extends Annotation, E> IterableFilterHandler<A, E> of(
Class<A> filterType,
BiPredicate<A, E> elementPredicate
) {
return new SimpleIterableFilterHandler<>(filterType, elementPredicate);
}
@IterableFilter
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.PARAMETER)
public @interface IterableElementTypes {
Class<?>[] value();
}
public static IterableFilterHandler<IterableElementTypes, Object> getIterableElementTypesFilterHandler() {
return of(
IterableElementTypes.class,
(annotation, element) -> isAssignableToAnyOf(element.getClass(), annotation.value())
);
}
}

View File

@ -0,0 +1,33 @@
package groowt.util.di.filters;
import java.lang.annotation.Annotation;
import java.util.function.BiPredicate;
final class SimpleFilterHandler<A extends Annotation, T> implements FilterHandler<A, T> {
private final BiPredicate<A, T> predicate;
private final Class<A> annotationClass;
private final Class<T> argClass;
public SimpleFilterHandler(BiPredicate<A, T> predicate, Class<A> annotationClass, Class<T> argClass) {
this.predicate = predicate;
this.annotationClass = annotationClass;
this.argClass = argClass;
}
@Override
public boolean check(A annotation, T arg) {
return this.predicate.test(annotation, arg);
}
@Override
public Class<A> getAnnotationClass() {
return this.annotationClass;
}
@Override
public Class<T> getArgumentClass() {
return this.argClass;
}
}

View File

@ -0,0 +1,32 @@
package groowt.util.di.filters;
import java.lang.annotation.Annotation;
import java.util.Objects;
import java.util.function.BiPredicate;
final class SimpleIterableFilterHandler<A extends Annotation, E> implements IterableFilterHandler<A, E> {
private final Class<A> annotationClass;
private final BiPredicate<A, E> elementPredicate;
public SimpleIterableFilterHandler(Class<A> annotationClass, BiPredicate<A, E> elementPredicate) {
this.annotationClass = annotationClass;
this.elementPredicate = elementPredicate;
}
@Override
public boolean check(A annotation, Iterable<E> iterable) {
for (final var e : Objects.requireNonNull(iterable)) {
if (!this.elementPredicate.test(annotation, e)) {
return false;
}
}
return true;
}
@Override
public Class<A> getAnnotationClass() {
return this.annotationClass;
}
}

View File

@ -0,0 +1,170 @@
package groowt.util.di;
import jakarta.inject.Inject;
import jakarta.inject.Named;
import org.junit.jupiter.api.Test;
import static groowt.util.di.BindingUtil.*;
import static groowt.util.di.NamedRegistryExtension.named;
import static org.junit.jupiter.api.Assertions.*;
public class DefaultRegistryObjectFactoryTests {
public interface Greeter {
String greet();
}
public static final class DefaultGreeter implements Greeter {
@Override
public String greet() {
return "Hello, World!";
}
}
public static final class GivenArgGreeter implements Greeter {
private final String greeting;
@Inject
public GivenArgGreeter(String greeting) {
this.greeting = greeting;
}
@Override
public String greet() {
return this.greeting;
}
}
public static final class InjectedArgGreeter implements Greeter {
private final String greeting;
@Inject
public InjectedArgGreeter(String greeting) {
this.greeting = greeting;
}
@Override
public String greet() {
return this.greeting;
}
}
public static final class InjectedNamedArgGreeter implements Greeter {
private final String greeting;
@Inject
public InjectedNamedArgGreeter(@Named("greeting") String greeting) {
this.greeting = greeting;
}
@Override
public String greet() {
return this.greeting;
}
}
public static final class InjectedNamedSetterGreeter implements Greeter {
private String greeting;
@Inject
public void setGreeting(@Named("greeting") String greeting) {
this.greeting = greeting;
}
@Override
public String greet() {
return this.greeting;
}
}
@Test
public void classSmokeScreen() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(registry -> {
registry.bind(Greeter.class, bc -> bc.to(DefaultGreeter.class));
});
final RegistryObjectFactory container = b.build();
final Greeter greeter = container.get(Greeter.class);
assertEquals("Hello, World!", greeter.greet());
}
@Test
public void singletonSmokeScreen() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(registry -> {
registry.bind(Greeter.class, toSingleton(new DefaultGreeter()));
});
final RegistryObjectFactory container = b.build();
final Greeter greeter = container.get(Greeter.class);
assertEquals("Hello, World!", greeter.greet());
}
@Test
public void providerSmokeScreen() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(registry -> {
registry.bind(Greeter.class, toProvider(DefaultGreeter::new));
});
final RegistryObjectFactory container = b.build();
final Greeter greeter = container.get(Greeter.class);
assertEquals("Hello, World!", greeter.greet());
}
@Test
public void givenArgSmokeScreen() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(registry -> {
registry.bind(Greeter.class, bc -> bc.to(GivenArgGreeter.class));
});
final RegistryObjectFactory container = b.build();
final Greeter greeter = container.get(Greeter.class, "Hello, World!");
assertEquals("Hello, World!", greeter.greet());
}
@Test
public void injectedArg() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(registry -> {
registry.bind(Greeter.class, bc -> bc.to(InjectedArgGreeter.class));
registry.bind(String.class, toSingleton("Hello, World!"));
});
final RegistryObjectFactory container = b.build();
final Greeter greeter = container.get(Greeter.class);
assertEquals("Hello, World!", greeter.greet());
}
@Test
public void injectedNamedArg() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(registry -> {
registry.bind(Greeter.class, bc -> bc.to(InjectedNamedArgGreeter.class));
registry.bind(named("greeting", String.class), toSingleton("Hello, World!"));
});
final RegistryObjectFactory container = b.build();
final Greeter greeter = container.get(Greeter.class);
assertEquals("Hello, World!", greeter.greet());
}
@Test
public void injectedSetter() {
final var b = DefaultRegistryObjectFactory.Builder.withDefaults();
b.configureRegistry(r -> {
r.bind(Greeter.class, toClass(InjectedNamedSetterGreeter.class));
r.bind(named("greeting", String.class), toSingleton("Hello, World!"));
});
final RegistryObjectFactory f = b.build();
final Greeter greeter = f.get(Greeter.class);
assertEquals("Hello, World!", greeter.greet());
}
}

View File

@ -0,0 +1,17 @@
<?xml version="1.0" encoding="UTF-8" ?>
<Configuration xmlns="http://logging.apache.org/log4j/2.0/config">
<Appenders>
<Console name="root">
<PatternLayout>
<LevelPatternSelector defaultPattern="[%t] %-5level %logger{1.} %msg%n">
<PatternMatch key="DEBUG" pattern="[%t] %-5level %logger{1.}.%M() %msg%n" />
</LevelPatternSelector>
</PatternLayout>
</Console>
</Appenders>
<Loggers>
<Root level="DEBUG">
<AppenderRef ref="root" />
</Root>
</Loggers>
</Configuration>