Add hints for Configurable objects supertypes and their generics. (#2890)
This commit is contained in:
committed by
GitHub
parent
b42ba9128f
commit
1d6690cdbb
@@ -235,6 +235,11 @@
|
||||
<version>1.20</version>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.springframework</groupId>
|
||||
<artifactId>spring-core-test</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
<build>
|
||||
<plugins>
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2013-2022 the original author or authors.
|
||||
* Copyright 2013-2023 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
@@ -92,18 +92,35 @@ class ConfigurableHintsRegistrationProcessor implements BeanFactoryInitializatio
|
||||
private static Set<Class<?>> getConfigurableTypes() {
|
||||
Set<Class<?>> classesToAdd = getClassesToAdd();
|
||||
Set<Class<?>> genericsToAdd = new HashSet<>();
|
||||
Set<Class<?>> superTypes = new HashSet<>();
|
||||
for (Class<?> clazz : classesToAdd) {
|
||||
ResolvableType resolvableType = ResolvableType.forType(clazz);
|
||||
if (resolvableType.getSuperType().hasGenerics()) {
|
||||
genericsToAdd.addAll(Arrays.stream(resolvableType.getSuperType().getGenerics())
|
||||
.map(ResolvableType::toClass).collect(Collectors.toSet()));
|
||||
}
|
||||
addGenericsForClass(genericsToAdd, resolvableType);
|
||||
addSuperTypesForClass(resolvableType, superTypes, genericsToAdd);
|
||||
}
|
||||
classesToAdd.addAll(genericsToAdd);
|
||||
classesToAdd.addAll(superTypes);
|
||||
return classesToAdd.stream().filter(Objects::nonNull).collect(Collectors.toSet());
|
||||
|
||||
}
|
||||
|
||||
private static void addGenericsForClass(Set<Class<?>> genericsToAdd, ResolvableType resolvableType) {
|
||||
if (resolvableType.getSuperType().hasGenerics()) {
|
||||
genericsToAdd.addAll(Arrays.stream(resolvableType.getSuperType().getGenerics()).map(ResolvableType::toClass)
|
||||
.collect(Collectors.toSet()));
|
||||
}
|
||||
}
|
||||
|
||||
private static void addSuperTypesForClass(ResolvableType resolvableType, Set<Class<?>> supertypesToAdd,
|
||||
Set<Class<?>> genericsToAdd) {
|
||||
ResolvableType superType = resolvableType.getSuperType();
|
||||
if (!ResolvableType.NONE.equals(superType)) {
|
||||
addGenericsForClass(genericsToAdd, superType);
|
||||
supertypesToAdd.add(superType.toClass());
|
||||
addSuperTypesForClass(superType, supertypesToAdd, genericsToAdd);
|
||||
}
|
||||
}
|
||||
|
||||
@SuppressWarnings({ "rawtypes" })
|
||||
private static Set<Class<?>> getClassesToAdd() {
|
||||
Set<Class<?>> classesToAdd = new HashSet<>();
|
||||
|
||||
@@ -0,0 +1,97 @@
|
||||
/*
|
||||
* Copyright 2013-2023 the original author or authors.
|
||||
*
|
||||
* 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
|
||||
*
|
||||
* https://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.springframework.cloud.gateway.config;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import org.springframework.aot.generate.GeneratedClass;
|
||||
import org.springframework.aot.generate.GeneratedMethods;
|
||||
import org.springframework.aot.generate.GenerationContext;
|
||||
import org.springframework.aot.generate.MethodReference;
|
||||
import org.springframework.aot.hint.ReflectionHints;
|
||||
import org.springframework.aot.test.generate.TestGenerationContext;
|
||||
import org.springframework.beans.factory.aot.BeanFactoryInitializationAotContribution;
|
||||
import org.springframework.beans.factory.aot.BeanFactoryInitializationCode;
|
||||
import org.springframework.beans.factory.config.BeanDefinition;
|
||||
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
|
||||
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
|
||||
import org.springframework.cloud.gateway.filter.factory.SpringCloudCircuitBreakerFilterFactory;
|
||||
import org.springframework.cloud.gateway.filter.factory.SpringCloudCircuitBreakerResilience4JFilterFactory;
|
||||
import org.springframework.javapoet.TypeSpec;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
/**
|
||||
* Tests for {@link ConfigurableHintsRegistrationProcessor}.
|
||||
*
|
||||
* @author Olga Maciaszek-Sharma
|
||||
*/
|
||||
class ConfigurableHintsRegistrationProcessorTests {
|
||||
|
||||
private final ConfigurableHintsRegistrationProcessor processor = new ConfigurableHintsRegistrationProcessor();
|
||||
|
||||
private final TestGenerationContext generationContext = new TestGenerationContext();
|
||||
|
||||
private final BeanFactoryInitializationCode beanFactoryInitializationCode = new MockBeanFactoryInitializationCode(
|
||||
generationContext);
|
||||
|
||||
private final DefaultListableBeanFactory beanFactory = new DefaultListableBeanFactory();
|
||||
|
||||
@Test
|
||||
void shouldRegisterReflectionHintsForTypeAndSuperTypesAndGenerics() {
|
||||
BeanDefinition beanDefinition = BeanDefinitionBuilder
|
||||
.rootBeanDefinition(SpringCloudCircuitBreakerResilience4JFilterFactory.class).getBeanDefinition();
|
||||
beanFactory.registerBeanDefinition("test", beanDefinition);
|
||||
|
||||
BeanFactoryInitializationAotContribution contribution = processor.processAheadOfTime(beanFactory);
|
||||
assertThat(contribution).isNotNull();
|
||||
contribution.applyTo(generationContext, beanFactoryInitializationCode);
|
||||
|
||||
ReflectionHints reflectionHints = generationContext.getRuntimeHints().reflection();
|
||||
assertThat(reflectionHints.getTypeHint(SpringCloudCircuitBreakerResilience4JFilterFactory.class)).isNotNull();
|
||||
assertThat(reflectionHints.getTypeHint(SpringCloudCircuitBreakerFilterFactory.class)).isNotNull();
|
||||
assertThat(reflectionHints.getTypeHint(SpringCloudCircuitBreakerFilterFactory.Config.class)).isNotNull();
|
||||
}
|
||||
|
||||
@SuppressWarnings("NullableProblems")
|
||||
static class MockBeanFactoryInitializationCode implements BeanFactoryInitializationCode {
|
||||
|
||||
private static final Consumer<TypeSpec.Builder> emptyTypeCustomizer = type -> {
|
||||
};
|
||||
|
||||
private final GeneratedClass generatedClass;
|
||||
|
||||
MockBeanFactoryInitializationCode(GenerationContext generationContext) {
|
||||
generatedClass = generationContext.getGeneratedClasses().addForFeature("Test", emptyTypeCustomizer);
|
||||
}
|
||||
|
||||
@Override
|
||||
public GeneratedMethods getMethods() {
|
||||
return generatedClass.getMethods();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addInitializer(MethodReference methodReference) {
|
||||
new ArrayList<>();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user