Spring自动注入源码分析

39 阅读 0 评论 26 点赞

我是靠谱客的博主老迟到煎饼，最近开发中收集的这篇文章主要介绍Spring自动注入源码分析，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

autowireByName

@Configuration
@ComponentScan("com.su.spring.common")
public class CommonConfig {
}

@Service
public class UserService {
    
    UserDao userDao;
    
    public void query(){
        System.out.println("UserService query() -----------");
        userDao.query();
    }

    public void setUserDao(UserDao userDao){
        this.userDao = userDao;
    }
}

@Repository()
public class UserDao {
    
    public void query(){
        System.out.println("UserDao query() -----------");
    }
}

通过自定义BeanDefinitionRegistryPostProcessor设置UserService的属性自动装配模式为byName

@Component
public class MyBeanDefinitionRegistryPostProcessor implements BeanDefinitionRegistryPostProcessor {
    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {

        GenericBeanDefinition beanDefinition = (GenericBeanDefinition) registry.getBeanDefinition("userService");
        beanDefinition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_NAME);
        registry.registerBeanDefinition("userService",beanDefinition);
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {

    }
}

接下来，我们直接从populateBean方法进行分析吧
在这里插入图片描述

在这里插入图片描述
接下来我们进入applyPropertyValues看看，spring是怎么通过setUserDao(UserDao userDao)这个writeMethod来完成属性注入的

经过了一些调用后，最终调用下面这个方法完成属性的注入

值的注意的是，autowwiredByName需要和set方法一起使用，才能够完成属性的注入。

autowireByType

@Component
public class MyBeanDefinitionRegistryPostProcessor implements BeanDefinitionRegistryPostProcessor {
    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {

        GenericBeanDefinition beanDefinition = (GenericBeanDefinition) registry.getBeanDefinition("userService");
        beanDefinition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_TYPE);
        registry.registerBeanDefinition("userService",beanDefinition);
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {

    }
}

这时候进入的是populateBean的autowiredByType了

在这里插入图片描述
之后的流程和autowiredByName一样。

AUTOWIRE_NO

如果没有通过自定义BeanDefinitionRegistryPostProcessor设置UserService的属性自动装配模式，spring默认是AUTOWIRE_NO，那么UserService属性需要加@Autowired注解才能完成自注入

@Service
public class UserService {

    @Autowired
    UserDao userDao;

    public void query(){
        System.out.println("UserService query() -----------");
        userDao.query();
    }

    public void setUserDao(UserDao userDao){
        this.userDao = userDao;
    }

}

在这里插入图片描述
那么接下来，我们进入AutowiredAnnotationBeanPostProcessor的postProcessPropertyValues方法看看吧

那么，@Autowired是根据ByName还是ByType来完成属性注入的呢？
这里，我们对测试类稍微调整一下，定义一个接口Dao,有两个实现类。一个是UserDao，另一个是UserDao2。

@Service
public class UserService {

    @Autowired
    Dao dao;

    public void query(){
        System.out.println("UserService query() -----------");
        dao.query();
    }
}

接下来，我们就要分析一下上面的beanFactory.resolveDependency的调用了。
在这里插入图片描述

在这里插入图片描述
最终会抛出异常出来

加@Qualifier限定后就能完成正常注入了。

在调整下测试类

@Service
public class UserService {

    @Autowired
    Dao userDao2;

    public void query(){
        System.out.println("UserService query() -----------");
        userDao2.query();
    }
}