IOC手动实现

IOC(Inversion of Control)

Inversion of Control即控制反转，其意思是将我们之前由客户端代码来创建的对象交由IOC容器来进行控制，对象的创建，初始化以及后面的管理都由IOC完成。

解耦：IOC的出现解决了类于类之间的耦合，我们在Web开发的Servlet时代，如果一个Servlet需要依赖另一个类的某些实现，那么我们需要在当前类对依赖的类进行创建和初始化，如果其他类也依赖了这个类，那也需要进行创建和初始化，而交给了IOC来管理的话，那么在需要的时候只需向IOC进行申请，而不需要重复的创建和初始化。当然，IOC也允许每次都重新创建一个新的对象。
方便与AOP进行配合：AOP也是一个使用十分频繁的功能，通过IOC可以十分方便的与AOP进行配合。

IOC中设计的设计模式

工厂模式。IOC容器来负责创建管理类实例对象，在需要时向IOC进行申请，从IOC中获取。所以IOC容器也称为bean工厂。

简单实现

Bean定义

IOC的主要的功能便是对Bean进行管理，包括创建、初始化、管理以及销魂的工作。首先我们面对的问题就是我们怎么让IOC能够创建一个Bean？为了创建Bean我们需要提供一些什么？

如何创建Bean

除了new创建对象：

1.反射：通过反射的方法可以创建类的实例：clazz.getClass().newInstance();。 2.工厂模式：工厂模式可以让我们在不接触实例类的情况下创建出实例。

为了创建bean需要提供什么

如果需要反射模式创建对象需要提供实例的class对象

对于工厂方法需要提供该类的工厂名(factoryName)和方法名(methodName)

IOC容器是对bean的整个生命周期进行管理，除了创建之外还需要对bean进行初始化，以及不需要时对bean进行销毁的工作(如释放资源等)。所以我们还需要提供初始化和销毁等操作。

创建对象时需要判断是否是单例

类图

bean工厂

Bean的定义解决了，但是这个bean定义以及创建好的Bean实例放在哪里呢，我们需要一个统一的地方来存放这些东西以方便我们要用的时候方便取。我们定义一个Bean工厂来存放bean，在需要的时候懂bean工厂中取即可，bean工厂对外提供的也仅仅是一个获取bean的方法即可，由于bean的类型不定，所以返回值定位Object。

注册Bean定义

现在需要考虑的事怎么来联系这两个类，我们还需要另外一个接口，接口的功能是让我们能注册和获取bean定义，这里我们通过beanName来区分不同的bean。

代码实现

BeanDefinition

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 11:23 AM
 */
public interface BeanDefinition {

    String SINGLETION = "singleton";

    String PROTOTYPE = "prototype";

    Class<?> getBeanClass();

    String getBeanFactory();

    String getCreateBeanMethod();

    String getStaticCreateBeanMethod();

    String getBeanInitMethodName();

    String getBeanDestoryMethodName();

    String getScope();

    boolean isSingleton();

    boolean isPrototype();

    boolean validate();
}

BeanFactory接口

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 11:33 AM
 */
public interface BeanFactory {

    Object getBean(String beanName);

    void close() throws IOException;
}

注册类 BeanDefinitionRegistry

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 11:34 AM
 */
public interface BeanDefinitionRegistry {
    void register(BeanDefinition bd,String beanName);

    BeanDefinition getBeanDefinition(String beanName);

    Boolean containsBeanDefinition(String beanName);
}

BeanDefinition 实现类

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 11:27 AM
 */
public class DefaultBeanDefinition implements BeanDefinition{
    private Class<?> clazz;

    private String beanFactoryName;

    private String createBeanMethodName;

    private String staticCreateBeanMethodName;

    private String beanInitMethodName;

    private String beanDestoryMethodName;

    private boolean isSingleton;

    // setter

    @Override
    public boolean validate() {
        return false;
    }

    public void setSingleton(boolean singleton) {
        isSingleton = singleton;
    }

    @Override
    public Class<?> getBeanClass() {
        return this.clazz;
    }

    @Override
    public String getBeanFactory() {
        return this.beanFactoryName;
    }

    @Override
    public String getCreateBeanMethod() {
        return this.createBeanMethodName;
    }

    @Override
    public String getStaticCreateBeanMethod() {
        return this.staticCreateBeanMethodName;
    }

    @Override
    public String getBeanInitMethodName() {
        return this.beanInitMethodName;
    }

    @Override
    public String getBeanDestoryMethodName() {
        return this.beanDestoryMethodName;
    }

    @Override
    public String getScope() {
        return this.isSingleton ? BeanDefinition.SINGLETION : BeanDefinition.PROTOTYPE;
    }

    @Override
    public boolean isSingleton() {
        return this.isSingleton;
    }

    @Override
    public boolean isPrototype() {
        return !this.isSingleton;
    }


}

package cn.jessej.study.IOC;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Assert;

import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 11:35 AM
 */
public class DefaultBeanFactory implements BeanFactory,BeanDefinitionRegistry,Cloneable{
    private Log log = LogFactory.getLog(this.getClass());

    //ConcurrentHashMap应对并发环境
    private Map<String, BeanDefinition> bdMap = new ConcurrentHashMap<>();

    private Map<String, Object> beanMap = new ConcurrentHashMap<>();

    @Override
    public void register(BeanDefinition bd, String beanName) {

        Assert.assertNotNull("beanName不能为空 beanName", beanName);
        Assert.assertNotNull("BeanDefinition不能为空", bd);

        if(bdMap.containsKey(beanName)){
            log.info("[" + beanName + "]已经存在");
        }

        if(!bd.validate()){
            log.info("BeanDefinition不合法");
        }

        if(!bdMap.containsKey(beanName)){
            bdMap.put(beanName, bd);
        }
    }

    @Override
    public Boolean containsBeanDefinition(String beanName) {
        return bdMap.containsKey(beanName);
    }

    @Override
    public BeanDefinition getBeanDefinition(String beanName) {
        if(!bdMap.containsKey(beanName)){
            //log.info("[" + beanName + "]不存在");
        }
        return bdMap.get(beanName);
    }

    public Object doGetBean(String beanName) throws InstantiationException, IllegalAccessException {
        if(!beanMap.containsKey(beanName)){
            //log.info("[" + beanName + "]不存在");
        }

        Object instance = beanMap.get(beanName);

        if(instance != null){
            return instance;
        }

        //不存在则进行创建
        if(!this.bdMap.containsKey(beanName)){
            log.info("不存在名为：[" + beanName + "]的bean定义");
        }

        BeanDefinition bd = this.bdMap.get(beanName);

        Class<?> beanClass = bd.getBeanClass();

        if(beanClass != null){
            instance = createBeanByConstruct(beanClass);
            if(instance == null){
                instance = createBeanByStaticFactoryMethod(bd);
            }
        }else if(instance == null && StringUtils.isNotBlank(bd.getStaticCreateBeanMethod())){
            instance = createBeanByFactoryMethod(bd);
        }

        this.doInit(bd, instance);

        if(instance != null && bd.isSingleton()){
            beanMap.put(beanName, instance);
        }

        return instance;
    }

    private void doInit(BeanDefinition bd, Object instance) {
        Class<?> beanClass = instance.getClass();
        if(StringUtils.isNotBlank(bd.getBeanInitMethodName())){
            try {
                Method method = beanClass.getMethod(bd.getBeanInitMethodName(), null);
                method.invoke(instance, null);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 构造方法创建实例
     * @param beanClass
     * @return
     */
    private Object createBeanByConstruct(Class<?> beanClass) {
        Object instance = null;
        try {
            instance = beanClass.newInstance();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return instance;
    }

    /**
     * 普通工厂方法创建实例
     * @param bd
     * @return
     */
    private Object createBeanByFactoryMethod(BeanDefinition bd) {
        Object instance = null;
        try {
            //获取工厂类
            Object factory = doGetBean(bd.getBeanFactory());
            //获取创建实例的方法
            Method method = factory.getClass().getMethod(bd.getCreateBeanMethod());
            //执行方法
            instance = method.invoke(factory, null);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return instance;
    }

    /**
     * 静态方法创建实例
     * @param bd
     * @return
     */
    private Object createBeanByStaticFactoryMethod(BeanDefinition bd) {
        Object instance = null;
        try {
            Class<?> beanClass = bd.getBeanClass();
            //获取创建实例的方法
            Method method = beanClass.getMethod(bd.getStaticCreateBeanMethod());
            instance = method.invoke(beanClass, null);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return instance;
    }

    @Override
    public Object getBean(String beanName) {
        if(!beanMap.containsKey(beanName)){
            //log.info("[" + beanName + "]不存在");
        }
        return beanMap.get(beanName);
    }

    @Override
    public void close() throws IOException {
        Set<Map.Entry<String, BeanDefinition>> entries = bdMap.entrySet();
        for(Map.Entry<String, BeanDefinition>  entry: entries){
            BeanDefinition value = entry.getValue();
            String destoryMethodName = value.getBeanDestoryMethodName();
            try {
                Method method = value.getBeanClass().getMethod(destoryMethodName, null);
                method.invoke(value.getBeanClass(), null);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }


}

package cn.jessej.study.IOC;

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 1:34 PM
 */
public class User {
    private String name;
    private int age;

    //getter setter

    public void init(){
        System.out.println("init...");
    }

    public void destory(){
        System.out.println("destory...");
    }

}

package cn.jessej.study.IOC;

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 1:37 PM
 */
public class TestFactory {
    public Object createMethod(){
        return new User();
    }

    public static Object staticCreateMethod(){
        return new User();
    }

}

package cn.jessej.study.IOC;

import org.junit.Test;

/**
 * @Author: jesse
 * @Date: 2018/12/20
 * @Time: 1:39 PM
 */
public class MySpringTest {
    static DefaultBeanFactory factory = new DefaultBeanFactory();

    @Test
    public void test() throws IllegalAccessException, InstantiationException {
        DefaultBeanDefinition bd = new DefaultBeanDefinition();
        bd.setClazz(User.class);
        bd.setSingleton(true);
        bd.setBeanFactoryName("TestFactory");
        bd.setCreateBeanMethodName("createMethod");
        bd.setStaticCreateBeanMethodName("staticCreateMethod");

        bd.setBeanInitMethodName("init");

        factory.register(bd, "user");

        System.out.println(factory.doGetBean("user"));
    }

}

spring源码学习