Question

假设我有一个接口Yoyo以及此接口的不同实现：

interface Yoyo { 
    fun haha() {
        println("hello world")
    }
}

@Component class Yoyo1 : Yoyo
@Component class Yoyo2 : Yoyo
@Component class Yoyo3 : Yoyo
@Component class YoyoN : Yoyo

现在我想在初始化上下文后实例化所有bean并做一些逻辑：

@SpringBootApplication
class YoyoApp

fun main(args: Array<String>) {
    SpringApplicationBuilder()
            .sources(YoyoApp::class.java)
            .initializers(beans {
               bean {
                   CommandLineRunner {
                       val y1 = ref<Yoyo1>()
                       val y2 = ref<Yoyo2>()
                       val y3 = ref<Yoyo3>()
                       val yN = ref<YoyoN>()
                       arrayOf(y1, y2, y3, yN).forEach { it.haha() }
                   }
               }
            })
            .run(*args)
}

我不想手动获取所有bean的引用（这相当繁琐），我想这样做：

val list = ref<List<Yoyo>>()
list.forEach { it.haha() }

但是我得到一个例外：

Caused by: org.springframework.beans.factory.NoSuchBeanDefinitionException: No qualifying bean of type 'java.util.List<?>' available

我知道我可以这样做，但我想改用新的Kotlin DSL：

@Component
class Hoho : CommandLineRunner {
    @Autowired
    lateinit var list: List<Yoyo>

    override fun run(vararg args: String?) {
        list.forEach { it.haha() }
    }
}

有可能吗？有任何想法吗？

P.S。这是gist。

Answer 1

可以在框架的源代码中找到DSL中使用的ref函数here。获取类型的所有bean没有等价物，但您可以将自己的扩展添加到BeanDefinitionDsl类来执行此操作：

inline fun <reified T : Any> BeanDefinitionDsl.refAll() : Map<String, T> {
    return context.getBeansOfType(T::class.java)
}

唯一的问题是，在当前发布的框架版本中，context所需的internal是BeanDefinitionDsl。从8天前开始This commit使其“可用于高级用例”，但是hasn't been框架的新版本已经发布，因此它尚未提供。

（同样的提交也使类直接扩展BeanDefinitionDsl.BeanDefinitionContext类，而不是@Component public class Http401UnauthorizedEntryPoint implements AuthenticationEntryPoint { private final Logger log = LoggerFactory.getLogger(Http401UnauthorizedEntryPoint.class); /** * Always returns a 401 error code to the client. */ @Override public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException arg2) throws IOException, ServletException { log.debug("Pre-authenticated entry point called. Rejecting access"); response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "Access Denied"); } }。）

结论：您可能不得不等待包含上述提交的下一个版本，然后您就可以自己创建此扩展。我还提交了pull request，希望这可以包含在框架本身中。

Answer 2

@ zsmb13在上一个答案中提到的context是left internal，而赞成provider<Any>()函数（从Spring 5.1.1开始）。所以最后我得到了以下内容：

interface Yoyo {
    fun haha() {
        println("hello world from: ${this.javaClass.canonicalName}")
    }
}

@Component class Yoyo1 : Yoyo
@Component class Yoyo2 : Yoyo
@Component class Yoyo3 : Yoyo
@Component class YoyoN : Yoyo


@SpringBootApplication
class YoyoApp

fun main(args: Array<String>) {
    SpringApplicationBuilder()
        .sources(YoyoApp::class.java)
        .initializers(beans {
            bean {
                CommandLineRunner {
                    val list = provider<Yoyo>().toList()
                    list.forEach { it.haha() }
                }
            }
        })
        .run(*args)
}

Spring Kotlin DSL：获取某种类型的所有豆类

2 个答案: