如何区分由@Bean注解创建的Spring Bean实例？

刚好我之前做Spring扩展库的时候碰到过类似的需求，给你分享几个靠谱的解决方案，完美匹配你不想用类型区分、靠工厂方法注解识别Bean的需求：

这是最贴合你需求的方案，核心思路是用自定义注解标记@Bean工厂方法，然后通过Spring的BeanPostProcessor拦截初始化后的Bean，判断它是否来自标记了注解的工厂方法，再执行自动操作。

步骤1：定义自定义注解 #

先写一个用来标记@Bean方法的注解，还可以根据需求加一些属性：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyManagedBean {
    // 可选：比如添加操作类型、优先级之类的属性
    String operationType() default "default";
}

步骤2：实现BeanPostProcessor处理逻辑 #

通过ConfigurableListableBeanFactory获取Bean的定义信息，解析出对应的工厂方法，判断是否带有我们的自定义注解：

@Component
public class MyBeanAutoProcessor implements BeanPostProcessor {

    private final ConfigurableListableBeanFactory beanFactory;

    // 构造注入Bean工厂，用来获取Bean定义
    public MyBeanAutoProcessor(ConfigurableListableBeanFactory beanFactory) {
        this.beanFactory = beanFactory;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        BeanDefinition beanDef = beanFactory.getBeanDefinition(beanName);
        // 只有@Bean方法创建的Bean才会是RootBeanDefinition类型
        if (beanDef instanceof RootBeanDefinition) {
            RootBeanDefinition rootDef = (RootBeanDefinition) beanDef;
            Method factoryMethod = rootDef.getResolvedFactoryMethod();
            // 检查工厂方法是否标记了我们的注解
            if (factoryMethod != null && factoryMethod.isAnnotationPresent(MyManagedBean.class)) {
                // 这里执行你的自动操作逻辑，比如配置Bean、注册到管理器等
                performAutoOperation(bean, factoryMethod.getAnnotation(MyManagedBean.class));
            }
        }
        return bean;
    }

    private void performAutoOperation(Object bean, MyManagedBean annotation) {
        // 示例逻辑：根据注解属性执行不同操作
        System.out.printf("对Bean[%s]执行%s类型的自动操作%n", 
                          bean.getClass().getSimpleName(), 
                          annotation.operationType());
    }
}

用法示例 #

在你的@Bean方法上标记注解即可：

@Configuration
public class MyConfig {

    @MyManagedBean(operationType = "database")
    @Bean
    public DataSource dataSource() {
        return new HikariDataSource();
    }

    @MyManagedBean
    @Bean
    public RedisTemplate<String, Object> redisTemplate() {
        return new RedisTemplate<>();
    }
}

如果你的库需要在Spring容器启动阶段就提前收集所有需要处理的Bean（而不是等Bean初始化时才处理），可以先通过BeanFactoryPostProcessor遍历所有Bean定义，给符合条件的Bean打标记，后续再用BeanPostProcessor处理。

步骤1：实现BeanFactoryPostProcessor标记Bean #

@Component
public class MyBeanMarkerProcessor implements BeanFactoryPostProcessor {

    // 自定义标记键
    private static final String MANAGED_BEAN_KEY = "isMyManagedBean";

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        for (String beanName : beanFactory.getBeanDefinitionNames()) {
            BeanDefinition beanDef = beanFactory.getBeanDefinition(beanName);
            if (beanDef instanceof RootBeanDefinition) {
                RootBeanDefinition rootDef = (RootBeanDefinition) beanDef;
                Method factoryMethod = rootDef.getResolvedFactoryMethod();
                if (factoryMethod != null && factoryMethod.isAnnotationPresent(MyManagedBean.class)) {
                    // 给Bean定义添加标记属性
                    rootDef.setAttribute(MANAGED_BEAN_KEY, true);
                    // 还可以提前存储注解信息，后续直接使用
                    rootDef.setAttribute("myManagedAnnotation", factoryMethod.getAnnotation(MyManagedBean.class));
                }
            }
        }
    }
}

步骤2：修改BeanPostProcessor使用标记 #

这样后续处理时就不用重复解析工厂方法，直接判断标记即可：

@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
    BeanDefinition beanDef = beanFactory.getBeanDefinition(beanName);
    if (Boolean.TRUE.equals(beanDef.getAttribute("isMyManagedBean"))) {
        MyManagedBean annotation = (MyManagedBean) beanDef.getAttribute("myManagedAnnotation");
        performAutoOperation(bean, annotation);
    }
    return bean;
}

• 针对懒加载Bean：BeanPostProcessor只会在Bean第一次被初始化时触发操作，如果需要懒加载Bean也被处理，这个逻辑是没问题的，只是触发时机延后。
• 工厂方法重载：Spring的RootBeanDefinition.getResolvedFactoryMethod()会正确解析到实际使用的重载方法，不用担心注解标记错的问题。
• 非@Configuration类的工厂方法：如果是通过普通工厂Bean（实现FactoryBean接口）的方法创建的Bean，需要额外处理，但你需求里是@Bean方法，所以这个场景可以忽略。

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