Add support of dependsOn

Closes gh-1458
spring-attic · Jan 19, 2022 · 9dc62d5 · 9dc62d5
1 parent a6e807d
commit 9dc62d5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/...g/springframework/aot/context/bootstrap/generator/bean/DefaultBeanRegistrationWriter.java b/...g/springframework/aot/context/bootstrap/generator/bean/DefaultBeanRegistrationWriter.java
@@ -51,6 +51,7 @@
 import org.springframework.core.AttributeAccessor;
 import org.springframework.core.ResolvableType;
 import org.springframework.util.ClassUtils;
+import org.springframework.util.ObjectUtils;
 import org.springframework.util.StringUtils;
 
 /**
@@ -205,6 +206,11 @@ private void handleBeanDefinitionMetadata(Builder code) {
 		if (StringUtils.hasText(scope) && !ConfigurableBeanFactory.SCOPE_SINGLETON.equals(scope)) {
 			statements.add("$L.setScope($S)", bdVariable, scope);
 		}
+		String[] dependsOn = this.beanDefinition.getDependsOn();
+		if (!ObjectUtils.isEmpty(dependsOn)) {
+			statements.add("$L.setDependsOn($L)", bdVariable,
+					this.parameterWriter.writeParameterValue(dependsOn));
+		}
 		if (this.beanDefinition.isLazyInit()) {
 			statements.add("$L.setLazyInit(true)", bdVariable);
 		}

diff --git a/...ingframework/aot/context/bootstrap/generator/bean/DefaultBeanRegistrationWriterTests.java b/...ingframework/aot/context/bootstrap/generator/bean/DefaultBeanRegistrationWriterTests.java
@@ -229,6 +229,15 @@ void writeWithSyntheticFlag() {
 						"    .instanceSupplier(() -> SampleFactory::new).customize((bd) -> bd.setSynthetic(true)).register(beanFactory);");
 	}
 
+	@Test
+	void writeWithDependsOn() {
+		BeanDefinition beanDefinition = BeanDefinitionBuilder.rootBeanDefinition(SampleFactory.class, "create")
+				.addDependsOn("test").getBeanDefinition();
+		assertThat(beanRegistration(beanDefinition, (code) -> code.add("() -> SampleFactory::new"))).lines()
+				.containsOnly("BeanDefinitionRegistrar.of(\"test\", SampleFactory.class)",
+						"    .instanceSupplier(() -> SampleFactory::new).customize((bd) -> bd.setDependsOn(new String[] { \"test\" })).register(beanFactory);");
+	}
+
 	@Test
 	void writeWithMultipleFlags() {
 		BeanDefinition beanDefinition = BeanDefinitionBuilder.rootBeanDefinition(SampleFactory.class, "create")