如何通过自定义注解@MyBean实现Spring Bean注入？

当然可以！通过BeanPostProcessor（或其扩展接口）配合你自定义的@MyBean注解，完全能实现对标注字段/构造参数的Bean赋值。下面是具体的实现思路和代码：

1. 基本逻辑 #

Spring的BeanPostProcessor允许我们在Bean实例化完成后、初始化前后拦截并修改Bean。我们可以利用这个特性，扫描Bean中带有@MyBean注解的字段，从Spring容器中获取对应类型的Bean，再通过反射完成字段赋值。

如果要支持构造参数上的@MyBean（比如你示例里Foo的构造参数），就得用InstantiationAwareBeanPostProcessor——因为构造参数是在Bean实例化前处理的，普通BeanPostProcessor覆盖不到这块逻辑。

2. 代码实现 #

处理字段注入的BeanPostProcessor #

import org.springframework.beans.BeansException
import org.springframework.beans.factory.config.BeanPostProcessor
import org.springframework.context.ApplicationContext
import org.springframework.context.ApplicationContextAware
import org.springframework.stereotype.Component
import java.lang.reflect.Field

@Component
class MyBeanPostProcessor : BeanPostProcessor, ApplicationContextAware {
    private lateinit var applicationContext: ApplicationContext

    override fun setApplicationContext(context: ApplicationContext) {
        this.applicationContext = context
    }

    override fun postProcessBeforeInitialization(bean: Any, beanName: String): Any? {
        // 遍历当前Bean的所有字段
        bean.javaClass.declaredFields.forEach { field ->
            // 检查字段是否标注了@MyBean
            if (field.isAnnotationPresent(MyBean::class.java)) {
                field.isAccessible = true // 放开私有字段的访问权限
                // 从容器中获取对应类型的Bean
                val targetBean = applicationContext.getBean(field.type)
                // 为字段赋值
                field.set(bean, targetBean)
            }
        }
        return bean
    }
}

补充构造参数注入的处理类 #

如果要支持构造参数上的@MyBean，需要额外实现这个类：

import org.springframework.beans.factory.config.InstantiationAwareBeanPostProcessorAdapter
import org.springframework.context.ApplicationContext
import org.springframework.context.ApplicationContextAware
import org.springframework.stereotype.Component

@Component
class MyBeanConstructorPostProcessor : InstantiationAwareBeanPostProcessorAdapter(), ApplicationContextAware {
    private lateinit var applicationContext: ApplicationContext

    override fun setApplicationContext(context: ApplicationContext) {
        this.applicationContext = context
    }

    override fun postProcessConstructorArgs(
        args: Array<out Any>?,
        parameterTypes: Array<out Class<*>?>?,
        beanName: String?
    ): Array<out Any>? {
        parameterTypes?.forEachIndexed { index, paramType ->
            val beanClass = applicationContext.getType(beanName) ?: return@forEachIndexed
            val constructor = beanClass.declaredConstructors.firstOrNull() ?: return@forEachIndexed
            // 检查当前构造参数是否标注了@MyBean
            val paramAnnotation = constructor.parameters.getOrNull(index)?.getAnnotation(MyBean::class.java)
            
            if (paramAnnotation != null && paramType != null) {
                // 将参数替换为容器中的对应Bean
                args?.set(index, applicationContext.getBean(paramType))
            }
        }
        return args
    }
}

3. 注意事项 #

• 这两个处理器类都要添加@Component注解，确保Spring能扫描到并加入容器
• 给私有字段赋值时必须设置field.isAccessible = true，否则反射会抛出访问权限异常
• 如果容器中存在多个同类型的Bean，需要额外指定Bean名称或配合@Qualifier注解，否则会抛出NoUniqueBeanDefinitionException

内容的提问来源于stack exchange，提问作者Razork