Change BeanDefinitionDsl to implement ApplicationContextInitializer

This commit is contained in:
Sebastien Deleuze 2017-09-04 10:46:20 +02:00
parent 2969af82d2
commit 8b8a6766de
2 changed files with 15 additions and 12 deletions

View File

@ -17,6 +17,7 @@
package org.springframework.context.support
import org.springframework.beans.factory.config.BeanDefinitionCustomizer
import org.springframework.context.ApplicationContextInitializer
import org.springframework.core.env.ConfigurableEnvironment
import java.util.function.Supplier
@ -75,7 +76,7 @@ fun beans(init: BeanDefinitionDsl.() -> Unit): BeanDefinitionDsl {
* @author Sebastien Deleuze
* @since 5.0
*/
class BeanDefinitionDsl(private val condition: (ConfigurableEnvironment) -> Boolean = { true }) : (GenericApplicationContext) -> Unit {
class BeanDefinitionDsl(private val condition: (ConfigurableEnvironment) -> Boolean = { true }) : ApplicationContextInitializer<GenericApplicationContext> {
@PublishedApi
internal val registrations = arrayListOf<(GenericApplicationContext) -> Unit>()
@ -210,17 +211,17 @@ class BeanDefinitionDsl(private val condition: (ConfigurableEnvironment) -> Bool
}
/**
* Register the bean defined via the DSL on thAh pe provided application context.
* Register the bean defined via the DSL on the provided application context.
* @param context The `ApplicationContext` to use for registering the beans
*/
override fun invoke(context: GenericApplicationContext) {
override fun initialize(context: GenericApplicationContext) {
for (registration in registrations) {
if (condition.invoke(context.environment)) {
registration.invoke(context)
}
}
for (child in children) {
child.invoke(context)
child.initialize(context)
}
}
}

View File

@ -36,9 +36,10 @@ class BeanDefinitionDslTests {
bean { Baz(ref("bar")) }
}
val context = GenericApplicationContext()
beans.invoke(context)
context.refresh()
val context = GenericApplicationContext().apply {
beans.initialize(this)
refresh()
}
assertNotNull(context.getBean<Foo>())
assertNotNull(context.getBean<Bar>("bar"))
@ -60,9 +61,10 @@ class BeanDefinitionDslTests {
}
}
val context = GenericApplicationContext()
beans.invoke(context)
context.refresh()
val context = GenericApplicationContext().apply {
beans.initialize(this)
refresh()
}
assertNotNull(context.getBean<Foo>())
assertNotNull(context.getBean<Bar>("bar"))
@ -87,9 +89,9 @@ class BeanDefinitionDslTests {
val context = GenericApplicationContext().apply {
environment.propertySources.addFirst(SimpleCommandLinePropertySource("--name=foofoo"))
beans.initialize(this)
refresh()
}
beans.invoke(context)
context.refresh()
assertNotNull(context.getBean<Foo>())
assertNotNull(context.getBean<Bar>("bar"))