Temporarily revert gh-29246

This new hint contribution triggers a GraalVM bug and breaks native
compilation for several samples, we need to temporarily revert this
change until a fix is available in GraalVM.

See gh-29246
This commit is contained in:
Brian Clozel 2022-10-03 14:14:30 +02:00
parent ff9792bae0
commit 054579a892
5 changed files with 13 additions and 69 deletions

View File

@ -1,27 +0,0 @@
/*
* Copyright 2002-2022 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.beans.factory.aot;
/**
* Record class holding key information for beans registered in a bean factory.
* @param beanName the name of the registered bean
* @param beanClass the type of the registered bean
* @author Brian Clozel
* @since 6.0
*/
record BeanRegistrationKey(String beanName, Class<?> beanClass) {
}

View File

@ -26,8 +26,6 @@ import org.springframework.aot.generate.GeneratedMethods;
import org.springframework.aot.generate.GenerationContext; import org.springframework.aot.generate.GenerationContext;
import org.springframework.aot.generate.MethodReference; import org.springframework.aot.generate.MethodReference;
import org.springframework.aot.generate.MethodReference.ArgumentCodeGenerator; import org.springframework.aot.generate.MethodReference.ArgumentCodeGenerator;
import org.springframework.aot.hint.MemberCategory;
import org.springframework.aot.hint.RuntimeHints;
import org.springframework.beans.factory.support.DefaultListableBeanFactory; import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.javapoet.ClassName; import org.springframework.javapoet.ClassName;
import org.springframework.javapoet.CodeBlock; import org.springframework.javapoet.CodeBlock;
@ -38,7 +36,6 @@ import org.springframework.javapoet.MethodSpec;
* register bean definitions. * register bean definitions.
* *
* @author Phillip Webb * @author Phillip Webb
* @author Brian Clozel
* @since 6.0 * @since 6.0
* @see BeanRegistrationsAotProcessor * @see BeanRegistrationsAotProcessor
*/ */
@ -47,11 +44,11 @@ class BeanRegistrationsAotContribution
private static final String BEAN_FACTORY_PARAMETER_NAME = "beanFactory"; private static final String BEAN_FACTORY_PARAMETER_NAME = "beanFactory";
private final Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations; private final Map<String, BeanDefinitionMethodGenerator> registrations;
BeanRegistrationsAotContribution( BeanRegistrationsAotContribution(
Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations) { Map<String, BeanDefinitionMethodGenerator> registrations) {
this.registrations = registrations; this.registrations = registrations;
} }
@ -70,15 +67,8 @@ class BeanRegistrationsAotContribution
GeneratedMethod generatedMethod = codeGenerator.getMethods().add("registerBeanDefinitions", method -> GeneratedMethod generatedMethod = codeGenerator.getMethods().add("registerBeanDefinitions", method ->
generateRegisterMethod(method, generationContext, codeGenerator)); generateRegisterMethod(method, generationContext, codeGenerator));
beanFactoryInitializationCode.addInitializer(generatedMethod.toMethodReference()); beanFactoryInitializationCode.addInitializer(generatedMethod.toMethodReference());
generateRegisterHints(generationContext.getRuntimeHints(), this.registrations);
} }
private void generateRegisterHints(RuntimeHints runtimeHints, Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations) {
registrations.keySet().forEach(beanRegistrationKey -> runtimeHints.reflection()
.registerType(beanRegistrationKey.beanClass(), MemberCategory.INTROSPECT_DECLARED_METHODS));
}
private void generateRegisterMethod(MethodSpec.Builder method, private void generateRegisterMethod(MethodSpec.Builder method,
GenerationContext generationContext, GenerationContext generationContext,
BeanRegistrationsCode beanRegistrationsCode) { BeanRegistrationsCode beanRegistrationsCode) {
@ -88,14 +78,14 @@ class BeanRegistrationsAotContribution
method.addParameter(DefaultListableBeanFactory.class, method.addParameter(DefaultListableBeanFactory.class,
BEAN_FACTORY_PARAMETER_NAME); BEAN_FACTORY_PARAMETER_NAME);
CodeBlock.Builder code = CodeBlock.builder(); CodeBlock.Builder code = CodeBlock.builder();
this.registrations.forEach((beanRegistrationKey, beanDefinitionMethodGenerator) -> { this.registrations.forEach((beanName, beanDefinitionMethodGenerator) -> {
MethodReference beanDefinitionMethod = beanDefinitionMethodGenerator MethodReference beanDefinitionMethod = beanDefinitionMethodGenerator
.generateBeanDefinitionMethod(generationContext, .generateBeanDefinitionMethod(generationContext,
beanRegistrationsCode); beanRegistrationsCode);
CodeBlock methodInvocation = beanDefinitionMethod.toInvokeCodeBlock( CodeBlock methodInvocation = beanDefinitionMethod.toInvokeCodeBlock(
ArgumentCodeGenerator.none(), beanRegistrationsCode.getClassName()); ArgumentCodeGenerator.none(), beanRegistrationsCode.getClassName());
code.addStatement("$L.registerBeanDefinition($S, $L)", code.addStatement("$L.registerBeanDefinition($S, $L)",
BEAN_FACTORY_PARAMETER_NAME, beanRegistrationKey.beanName(), BEAN_FACTORY_PARAMETER_NAME, beanName,
methodInvocation); methodInvocation);
}); });
method.addCode(code.build()); method.addCode(code.build());

View File

@ -27,7 +27,6 @@ import org.springframework.beans.factory.support.RegisteredBean;
* register beans. * register beans.
* *
* @author Phillip Webb * @author Phillip Webb
* @author Brian Clozel
* @since 6.0 * @since 6.0
*/ */
class BeanRegistrationsAotProcessor implements BeanFactoryInitializationAotProcessor { class BeanRegistrationsAotProcessor implements BeanFactoryInitializationAotProcessor {
@ -36,13 +35,13 @@ class BeanRegistrationsAotProcessor implements BeanFactoryInitializationAotProce
public BeanRegistrationsAotContribution processAheadOfTime(ConfigurableListableBeanFactory beanFactory) { public BeanRegistrationsAotContribution processAheadOfTime(ConfigurableListableBeanFactory beanFactory) {
BeanDefinitionMethodGeneratorFactory beanDefinitionMethodGeneratorFactory = BeanDefinitionMethodGeneratorFactory beanDefinitionMethodGeneratorFactory =
new BeanDefinitionMethodGeneratorFactory(beanFactory); new BeanDefinitionMethodGeneratorFactory(beanFactory);
Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>(); Map<String, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>();
for (String beanName : beanFactory.getBeanDefinitionNames()) { for (String beanName : beanFactory.getBeanDefinitionNames()) {
RegisteredBean registeredBean = RegisteredBean.of(beanFactory, beanName); RegisteredBean registeredBean = RegisteredBean.of(beanFactory, beanName);
BeanDefinitionMethodGenerator beanDefinitionMethodGenerator = beanDefinitionMethodGeneratorFactory BeanDefinitionMethodGenerator beanDefinitionMethodGenerator = beanDefinitionMethodGeneratorFactory
.getBeanDefinitionMethodGenerator(registeredBean, null); .getBeanDefinitionMethodGenerator(registeredBean, null);
if (beanDefinitionMethodGenerator != null) { if (beanDefinitionMethodGenerator != null) {
registrations.put(new BeanRegistrationKey(beanName, registeredBean.getBeanClass()), beanDefinitionMethodGenerator); registrations.put(beanName, beanDefinitionMethodGenerator);
} }
} }
if (registrations.isEmpty()) { if (registrations.isEmpty()) {

View File

@ -32,8 +32,6 @@ import org.springframework.aot.generate.ClassNameGenerator;
import org.springframework.aot.generate.GenerationContext; import org.springframework.aot.generate.GenerationContext;
import org.springframework.aot.generate.MethodReference; import org.springframework.aot.generate.MethodReference;
import org.springframework.aot.generate.MethodReference.ArgumentCodeGenerator; import org.springframework.aot.generate.MethodReference.ArgumentCodeGenerator;
import org.springframework.aot.hint.MemberCategory;
import org.springframework.aot.hint.predicate.RuntimeHintsPredicates;
import org.springframework.aot.test.generate.TestGenerationContext; import org.springframework.aot.test.generate.TestGenerationContext;
import org.springframework.beans.factory.support.DefaultListableBeanFactory; import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.beans.factory.support.RegisteredBean; import org.springframework.beans.factory.support.RegisteredBean;
@ -78,13 +76,13 @@ class BeanRegistrationsAotContributionTests {
@Test @Test
void applyToAppliesContribution() { void applyToAppliesContribution() {
Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>(); Map<String, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>();
RegisteredBean registeredBean = registerBean( RegisteredBean registeredBean = registerBean(
new RootBeanDefinition(TestBean.class)); new RootBeanDefinition(TestBean.class));
BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator( BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator(
this.methodGeneratorFactory, registeredBean, null, this.methodGeneratorFactory, registeredBean, null,
Collections.emptyList()); Collections.emptyList());
registrations.put(new BeanRegistrationKey(registeredBean.getBeanName(), registeredBean.getBeanClass()), generator); registrations.put("testBean", generator);
BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution( BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution(
registrations); registrations);
contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode); contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode);
@ -100,13 +98,13 @@ class BeanRegistrationsAotContributionTests {
this.generationContext = new TestGenerationContext( this.generationContext = new TestGenerationContext(
new ClassNameGenerator(TestGenerationContext.TEST_TARGET, "Management")); new ClassNameGenerator(TestGenerationContext.TEST_TARGET, "Management"));
this.beanFactoryInitializationCode = new MockBeanFactoryInitializationCode(this.generationContext); this.beanFactoryInitializationCode = new MockBeanFactoryInitializationCode(this.generationContext);
Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>(); Map<String, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>();
RegisteredBean registeredBean = registerBean( RegisteredBean registeredBean = registerBean(
new RootBeanDefinition(TestBean.class)); new RootBeanDefinition(TestBean.class));
BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator( BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator(
this.methodGeneratorFactory, registeredBean, null, this.methodGeneratorFactory, registeredBean, null,
Collections.emptyList()); Collections.emptyList());
registrations.put(new BeanRegistrationKey(registeredBean.getBeanName(), registeredBean.getBeanClass()), generator); registrations.put("testBean", generator);
BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution( BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution(
registrations); registrations);
contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode); contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode);
@ -119,7 +117,7 @@ class BeanRegistrationsAotContributionTests {
@Test @Test
void applyToCallsRegistrationsWithBeanRegistrationsCode() { void applyToCallsRegistrationsWithBeanRegistrationsCode() {
List<BeanRegistrationsCode> beanRegistrationsCodes = new ArrayList<>(); List<BeanRegistrationsCode> beanRegistrationsCodes = new ArrayList<>();
Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>(); Map<String, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>();
RegisteredBean registeredBean = registerBean( RegisteredBean registeredBean = registerBean(
new RootBeanDefinition(TestBean.class)); new RootBeanDefinition(TestBean.class));
BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator( BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator(
@ -136,7 +134,7 @@ class BeanRegistrationsAotContributionTests {
} }
}; };
registrations.put(new BeanRegistrationKey(registeredBean.getBeanName(), registeredBean.getBeanClass()), generator); registrations.put("testBean", generator);
BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution( BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution(
registrations); registrations);
contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode); contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode);
@ -145,22 +143,6 @@ class BeanRegistrationsAotContributionTests {
assertThat(actual.getMethods()).isNotNull(); assertThat(actual.getMethods()).isNotNull();
} }
@Test
void applyToRegisterReflectionHints() {
Map<BeanRegistrationKey, BeanDefinitionMethodGenerator> registrations = new LinkedHashMap<>();
RegisteredBean registeredBean = registerBean(
new RootBeanDefinition(TestBean.class));
BeanDefinitionMethodGenerator generator = new BeanDefinitionMethodGenerator(
this.methodGeneratorFactory, registeredBean, null,
Collections.emptyList());
registrations.put(new BeanRegistrationKey(registeredBean.getBeanName(), registeredBean.getBeanClass()), generator);
BeanRegistrationsAotContribution contribution = new BeanRegistrationsAotContribution(
registrations);
contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode);
assertThat(RuntimeHintsPredicates.reflection().onType(TestBean.class).withMemberCategory(MemberCategory.INTROSPECT_DECLARED_METHODS))
.accepts(this.generationContext.getRuntimeHints());
}
private RegisteredBean registerBean(RootBeanDefinition rootBeanDefinition) { private RegisteredBean registerBean(RootBeanDefinition rootBeanDefinition) {
String beanName = "testBean"; String beanName = "testBean";
this.beanFactory.registerBeanDefinition(beanName, rootBeanDefinition); this.beanFactory.registerBeanDefinition(beanName, rootBeanDefinition);

View File

@ -50,7 +50,7 @@ class BeanRegistrationsAotProcessorTests {
BeanRegistrationsAotContribution contribution = processor BeanRegistrationsAotContribution contribution = processor
.processAheadOfTime(beanFactory); .processAheadOfTime(beanFactory);
assertThat(contribution).extracting("registrations") assertThat(contribution).extracting("registrations")
.asInstanceOf(InstanceOfAssertFactories.MAP).hasSize(2); .asInstanceOf(InstanceOfAssertFactories.MAP).containsKeys("b1", "b2");
} }
} }