Spring源码之手写IOC容器

Spring源码之手写IOC容器

一、IOC分析

1.Spring的核心

  在Spring中非常核心的内容是 IOC和 AOP.

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

2.逐步揭开IOC的面纱

2.1 何为IOC

  IOC : Inversion of Control 控制翻转，简单的说就是：依赖对象的获取被反转了。

2.2 IOC技术给开发人员带来的好处

IOC的好处：

1. 代码更加简洁，不需要去new 要使用的对象了
2. 面向接口编程，使用者与具体类，解耦，易扩展、替换实现者
3. 可以方便进行AOP编程

2.3 IOC容器的工作职责

  IOC容器工作职责：负责创建和管理类的实例，并能以很平和方式给使用者提供实例对象。

2.4 IOC容器采用的设计模式

 IOC容器在实现上采用了工厂模式，负责创建类实例对象，使用者需要时直接从IOC容器中获取既可。IOC容器我们也称之为Bean工厂。

  那么我们一直说的Bean是什么呢？bean：组件，也就是类的对象!!!

二、IOC实现

  通过上面的介绍我们也清楚了IOC的核心就是Bean工厂，那么这个Bean工厂我们应该要如何来设计实现它呢？我们来继续分析。

1.Bean工厂的作用

  首先Bean工厂的作用我们上面也分析了就是创建，管理Bean，并且需要对外提供Bean的实例。

2.Bean工厂的初步设计

  基于Bean工厂的基本作用，我们先来分析Bean工厂应该具备的相关行为。

1. Bean工厂应该能够对外提供获取bean实例的方法，因此需要定义一个getBean()方法。
2. Bean工厂需要知道生产的bean的类型，故而getBean()方法需要接受对应的参数。
3. Bean的类型种类繁多，导致getBean()方法的返回类型存在多种类型，所以需要用Object来表示。

这样一来Bean工厂的定义就跃然纸上了

  上面定义了Bean工厂对外提供bean实例的方法，但是Bean工厂如何知道要创建对象，以及怎么创建实例对象呢？

  从上图Bean实例对象的创建过程可知，我们需要把Bean的定义信息告诉BeanFactory工厂，BeanFactory工厂才能根据Bean的定义信息来生成对应的bean实例对象。

到这一步我们还需要做两件事

1. 需要定义一个模型来表示需要创建的Bean实例的信息，即Bean定义信息。
2. Bean工厂需要提供方法来处理这些Bean定义信息。

3.Bean的定义

  通过上面的介绍我们应该清楚了Bean定义的意义了。接下来我们思考一下定义Bean定义的模型要考虑的若干问题。

3.1 Bean定义的作用

  作用肯定是告诉Bean工厂应该如何来创建某类的Bean实例

3.2 需要给使用者提供哪些获取实例的方式

3.3 在BeanDefinition中需要给Bean工厂提供哪些信息

这样一来我们就清楚了BeanDefinition应该要具有的基本功能了。

3.4 功能增强

  我们可以在现有的基础上提供增强功能，例如：Bean工厂创建的是单例对象，具有特定的初始化方法和销毁逻辑的方法。

  同时创建BeanDefinition的一个通用实现类：GenericBeanDefinition。

具体代码为：

/**
 * bean定义接口
 */
public interface BeanDefinition {

    String SCOPE_SINGLETION = "singleton";

    String SCOPE_PROTOTYPE = "prototype";

    /**
     * 类
     */
    Class<?> getBeanClass();

    /**
     * Scope
     */
    String getScope();

    /**
     * 是否单例
     */
    boolean isSingleton();

    /**
     * 是否原型
     */
    boolean isPrototype();

    /**
     * 工厂bean名
     */
    String getFactoryBeanName();

    /**
     * 工厂方法名
     */
    String getFactoryMethodName();

    /**
     * 初始化方法
     */
    String getInitMethodName();

    /**
     * 销毁方法
     */
    String getDestroyMethodName();

    boolean isPrimary();

    /**
     * 校验bean定义的合法性
     */
    default boolean validate() {
        // 没定义class,工厂bean或工厂方法没指定，则不合法。
        if (this.getBeanClass() == null) {
            if (StringUtils.isBlank(getFactoryBeanName()) || StringUtils.isBlank(getFactoryMethodName())) {
                return false;
            }
        }

        // 定义了类，又定义工厂bean，不合法
        if (this.getBeanClass() != null && StringUtils.isNotBlank(getFactoryBeanName())) {
            return false;
        }

        return true;
    }

}

4.Bean的注册

  Bean的定义清楚后，我们需要考虑的是如何将BeanDefinition和BeanFactory进行关联。

  在这儿我们可以专门定义一个 BeanDefinitionRegistry来实现Bean定义的注册功能。

  那么我们需要考虑 BeanDefinitionRegistry 应该具备的功能

BeanDefinitionRegistry 应该具备的功能

1. 注册BeanDefinition
2. 获取BeanDefinition

  同时为了保证能够区分各个BeanDefinition的定义信息，我们需要给每一个Bean定义一个唯一的名称。

具体实现代码:

public interface BeanDefinitionRegistry {

	void registerBeanDefinition(String beanName, BeanDefinition beanDefinition) throws BeanDefinitionRegistException;

	BeanDefinition getBeanDefinition(String beanName);

	boolean containsBeanDefinition(String beanName);

}

5.BeanFactory实现

  到此我们来梳理一下已经实现的相关功能：

  通过上面的分析、梳理，接下来我们就要考虑BeanFactory的功能实现了。我们先来实现一个最基础的Bean工厂（DefaultBeanFactory）。DefaultBeanFactory需要实现如下的5个功能

DefaultBeanFactory待实现功能

1. 实现Bean定义信息的注册
2. 实现Bean工厂定义的getBean方法
3. 实现初始化方法的执行
4. 实现单例的要求
5. 实现容器关闭是执行单例的销毁操作

实现逻辑的代码部分太多，但是根据上面的UML图，也可以很快自己写出来，在这里就不贴这块儿的代码了（需要的小伙伴，可以私信我，可以单独返给大家）。

思考：对于单例Bean，我们可否提前实例化?

三、IOC增强

  上面第一版本的IOC容器我们已经实现了，我们可以在这个基础上来基础迭代增强IOC的功能

1.Bean别名的增强

  Bean除了标识唯一的名称外，理论上还可以有任意数量的别名，但要注意的是每个别名必须是唯一的。

别名的特点

1. 可以有多个别名
2. 也可以是别名的别名
3. 别名也是唯一的

这里说的别名增强功能实现时需要考虑如下问题：

1. 数据结构
2. 功能点

2. Type类型的增强

  上面实现的是根据 bean的 name来获取Bean实例，我们还希望能扩展通过 Type来获取实例对象。这时对应的接口为：

  也就是需要实现根据Type找到Bean对象的功能。正常的实例逻辑为：

  但是上面的实现方案性能比较糟糕，需要优化下，可以提前把Type和Bean的对应关系找出来，然后用Map缓存起来。

同时需要考虑几个问题：

1. Map中存储的数据用什么合适？
2. type和bean是一对一的关系吗？
3. 何时建立该关系呢？

private Map<Class<?>, Set<String>> typeMap = new ConcurrentHashMap<>(256);

  具体的实现我们可以在DefaultBeanFactory中添加一个buildTypeMap()方法来处理这个事情

  buildTypeMap()方法处理的逻辑如下：

  然后在BeanFactory中添加一个getType方法，封装获取Bean的Type的逻辑，方便buildTypeMap()方法的使用。最后就是getBean(Class) 方法的实现了。因为Class对应的类型可能有多个，这时需要通过Primary来处理了。

IOC容器-核心部分类图