Add an env accessor to BeanDefinitionContext
Issue: SPR-15755
This commit is contained in:
parent
c292a89b24
commit
f72e0daa54
|
@ -17,7 +17,6 @@
|
|||
package org.springframework.context.support
|
||||
|
||||
import org.springframework.beans.factory.config.BeanDefinitionCustomizer
|
||||
import org.springframework.context.ApplicationContext
|
||||
import org.springframework.core.env.ConfigurableEnvironment
|
||||
import java.util.function.Supplier
|
||||
|
||||
|
@ -38,12 +37,20 @@ open class BeanDefinitionDsl(val condition: (ConfigurableEnvironment) -> Boolean
|
|||
PROTOTYPE
|
||||
}
|
||||
|
||||
class BeanDefinitionContext(val context: ApplicationContext) {
|
||||
class BeanDefinitionContext(val context: GenericApplicationContext) {
|
||||
|
||||
|
||||
inline fun <reified T : Any> ref(name: String? = null) : T = when (name) {
|
||||
null -> context.getBean(T::class.java)
|
||||
else -> context.getBean(name, T::class.java)
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the [ConfigurableEnvironment] associated to the underlying [GenericApplicationContext].
|
||||
*/
|
||||
val env : ConfigurableEnvironment
|
||||
get() = context.environment
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
@ -21,6 +21,7 @@ import org.junit.Test
|
|||
import org.springframework.beans.factory.NoSuchBeanDefinitionException
|
||||
import org.springframework.beans.factory.getBean
|
||||
import org.springframework.context.support.BeanDefinitionDsl.*
|
||||
import org.springframework.core.env.SimpleCommandLinePropertySource
|
||||
|
||||
class BeanDefinitionDslTests {
|
||||
|
||||
|
@ -75,13 +76,16 @@ class BeanDefinitionDslTests {
|
|||
val beans = beans {
|
||||
bean<Foo>()
|
||||
bean<Bar>("bar")
|
||||
bean { FooFoo(it.env.getProperty("name")) }
|
||||
environment({it.activeProfiles.contains("baz")}) {
|
||||
bean { Baz(it.ref()) }
|
||||
bean { Baz(it.ref("bar")) }
|
||||
}
|
||||
}
|
||||
|
||||
val context = GenericApplicationContext()
|
||||
val context = GenericApplicationContext().apply {
|
||||
environment.propertySources.addFirst(SimpleCommandLinePropertySource("--name=foofoo"))
|
||||
}
|
||||
beans.invoke(context)
|
||||
context.refresh()
|
||||
|
||||
|
@ -92,6 +96,8 @@ class BeanDefinitionDslTests {
|
|||
fail("Expect NoSuchBeanDefinitionException to be thrown")
|
||||
}
|
||||
catch(ex: NoSuchBeanDefinitionException) { null }
|
||||
val foofoo = context.getBean<FooFoo>()
|
||||
assertEquals("foofoo", foofoo.name)
|
||||
}
|
||||
|
||||
}
|
||||
|
@ -99,3 +105,4 @@ class BeanDefinitionDslTests {
|
|||
class Foo
|
||||
class Bar
|
||||
class Baz(val bar: Bar)
|
||||
class FooFoo(val name: String)
|
||||
|
|
Loading…
Reference in New Issue