模拟实现解析@Bean注解

原创已于 2022-08-15 17:01:50 修改 · 304 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#java #spring

于 2022-08-15 17:01:35 首次发布

本文通过实例展示了如何模拟实现Spring框架中的@Bean注解解析过程，包括创建Bean1和Bean4，编写自定义BeanFactory后处理器，以及进行测试以确保Bean4和Bean5的初始化方法被正确调用。

准备一个Bean1

@ComponentScan({"com.ghc.springlearning.MyPostProcessors.components"})
@Configuration
public class Bean1 {
    @Bean(initMethod = "init")
    public Bean4 bean4(){
        return new Bean4();
    }
    @Bean
    public Bean5 bean5(){
        return new Bean5();
    }
}

Bean4

public class Bean4 {
    public void init(){
        System.out.println("bean4 初始化");
    }
}

编写自定义BeanFactory后处理器实现接口

public class AtBeanPostProcessor implements BeanFactoryPostProcessor {

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
        DefaultListableBeanFactory beanFactory=(DefaultListableBeanFactory) configurableListableBeanFactory;
        CachingMetadataReaderFactory readerFactory = new CachingMetadataReaderFactory();
        try {
            //1.拿到MetadataReader
            MetadataReader metadataReader = readerFactory.getMetadataReader(new ClassPathResource("com/ghc/springlearning/MyPostProcessors/components/Bean1.class"));
            //2.拿到注解元数据--》拿到注解的方法
            Set<MethodMetadata> methods = metadataReader.getAnnotationMetadata().getAnnotatedMethods(Bean.class.getName());
            for (MethodMetadata method : methods) {
                BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition();
                //工厂方法Bean，指定哪个方法,属于哪个对象
                //看@Bean有无设置其他属性，如初始化方法
                //获取方法加的注解属性,如initMethod
                String initMethod = method.getAnnotationAttributes(Bean.class.getName()).get("initMethod").toString();
                //如果initMethod不为空，说明有
                if(initMethod.length()>0){
                    //设置初始化方法
                    beanDefinitionBuilder.setInitMethodName(initMethod);
                }
                beanDefinitionBuilder.setFactoryMethodOnBean(method.getMethodName(),"bean1");
                //拿到beanDefinition
                AbstractBeanDefinition beanDefinition = beanDefinitionBuilder.getBeanDefinition();
                beanFactory.registerBeanDefinition(method.getMethodName(),beanDefinition);
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
}

将后处理器注册为Bean，测试代码：

public class Test {
    public static void main(String[] args) {
        GenericApplicationContext context = new GenericApplicationContext();
        context.registerBean("bean1",Bean1.class);
        context.registerBean(ComponentScanProcessor.class);
        context.registerBean(AtBeanPostProcessor.class);
        context.refresh();
        for (String name : context.getBeanDefinitionNames()) {
            System.out.println(name);
        }
    }
}