徒手撸一个简单的IOC

Spring框架中最经典的两个就是IOC和AOP，其中IOC(Inversion of Control)是什么呢？控制反转，简单来说就是将控制实体Bean的动作交给了Spring容器进行管理。再简单点来说就是例如之前想用一个类，必需new一个，但是使用了Spring那么直接用@Autowired注解或者者用xml配置的方式就能直接取得此对象，而且你也不用管它的生命周期啊等等之类的。就不用自己new一个对象了。

image

假如是之前没有使用IOC的话，那么这些对象的创立以及赋值都是由我们自己创立了，下面简单的演示了假如有上面四个对象依赖的话，那么没有IOC我们必需要创立对象并且赋值。仅仅四个对象就这么多，那么一旦项目大了，对象成百上千，假如还这样写的话，那么绝对是一场灾难。

对象A a = new 对象A();对象B b = new 对象B();对象C c = new 对象C();对象D d = new 对象D();a.setB(b);a.setC(c);b.setD(d);c.setD(d);

因而在Spring中通过IOC将所有的对象统一放到Spring的容器中进行管理，所以就简单了很多。上面的实例化对象的代码也不需要我们写了。

image

上面说了那么多，其实就是一句话IOC非常重要，但是假如直接看Spring源码的话会非常懵逼，所以就简单的写一个IOC的小例子来了解这种思想。

分析并编写代码

还是编写代码前的分析阶段，Spring的IOC其实就是将所有的Bean放在统一容器中进行管理起来，而后在在获取的时候进行初始化，所以需要我们在程序启动的时候将被标记的类进行存储在自己设置的容器中管理。

• 初始化阶段：将被@MyIoc相似于Spring中@Service标记的类放入到自己设置的容器中。
• 使用：通过自己设置的获取Bean的类进行统一获取。

现在我们就以上面两个步骤进行详细点的分析

数据准备阶段

首先初始化阶段我们要先建立两个注解类用于类的发现(@MyIoc相似于@Service)。

@Target(ElementType.TYPE)@Retention(RetentionPolicy.RUNTIME)public @interface MyIoc {    }

而后要初始化信息进自己设置容器的话用什么类型的容器去存储这些信息呢？这里可以想到是用Map来存，用key为类名，value用什么呢？value就是要放在容器中进行管理的类的信息了，那么一个类有什么信息呢即类是由什么组成呢？有以下几个信息

• 类名
• 构造函数
• 属性值
• 父类

所以根据上面的分析我们可以建立一个实体类来存储这些信息，此时我们就不考虑复杂的构造函数了，就都是初始化的无参构造函数。而后父类的属性就不进行分析注入了。所以此时类实体类就简单了。

@Datapublic class BeanDefinition {    private String className;    private String alias;    private String superNames;}

初始化阶段

有了存储类信息的类了，那么我们在程序启动的时候就应该将这些信息给加载到Map中，此时建立一个启动类用于初始化被@MyIoc标记的类的信息。

@Component@Order(value = 1)public class IoCInitConifg implements CommandLineRunner{    @Override    public void run(String... args){        ConcurrentHashMap<String,BeanDefinition> concurrentHashMap = new ConcurrentHashMap<>();        Reflections reflections = new Reflections();        //取得项目中所有被MyIoc标记得类        Set<Class<?>> typesAnnotatedWith = reflections.getTypesAnnotatedWith(MyIoc.class);        //将其信息初始进自己设置容器MyBeanFactory中        for (Class clazz : typesAnnotatedWith){            BeanDefinition beanDefinition = new BeanDefinition();            String className = clazz.getName();            String superclassName = clazz.getSuperclass().getName();            beanDefinition.setClassName(className);            beanDefinition.setSuperNames(superclassName);            beanDefinition.setAlias(getClassName(className));            concurrentHashMap.put(className,beanDefinition);        }        MyBeanFactoryImpl.setBeanDineMap(concurrentHashMap);    }    private String getClassName(String beanClassName) {        String className = beanClassName.substring(beanClassName.lastIndexOf(".") + 1);        className = className.substring(0, 1).toLowerCase() + className.substring(1);        return className;    }}

此时得说一下自己设置的统一容器管理的类MyBeanFactory此类用作统一取得类的途径

public interface MyBeanFactory {    Object getBeanByName(String name) throws Exception;}

此时还有其实现类

@Log4jpublic class MyBeanFactoryImpl implements MyBeanFactory{    //存储对象名称和已经实例化的对象映射    private static ConcurrentHashMap<String,Object> beanMap = new ConcurrentHashMap<>();    //存储对象名称和对应对象信息的映射    private static ConcurrentHashMap<String,BeanDefinition> beanDefineMap= new ConcurrentHashMap<>();    //存储存储在容器中对象的名称    private static Set<String> beanNameSet = Collections.synchronizedSet(new HashSet<>());    @Override    public Object getBeanByName(String name) throws Exception {        //看有没有已经实例化的对象,有的话就直接返回        Object object = beanMap.get(name);        if (object != null){            return object;        }        //没有的话就实例化一个对象        object = getObject(beanDefineMap.get(name));        if (object != null){            //对实例化中对象的注入需要的参数            setFild(object);            //将实例化的对象放入Map中,便于下次使用            beanMap.put(name,object);        }        return object;    }    public void setFild(Object bean) throws Exception {        Field[] declaredFields = bean.getClass().getDeclaredFields();        for (Field field: declaredFields){            String filedAllName = field.getType().getName();            if (beanNameSet.contains(filedAllName)){                Object findBean = getBeanByName(filedAllName);                //为对象中的属性赋值                field.setAccessible(true);                field.set(bean,findBean);            }        }    }    public Object getObject(BeanDefinition beanDefinition) throws Exception {        String className = beanDefinition.getClassName();        Class<?> clazz = null;        try {            clazz = Class.forName(className);        } catch (ClassNotFoundException e) {            log.info("can not find bean by beanName: "+className);            throw new Exception("can not find bean by beanName: "+className);        }        return clazz;    }    public static void setBeanDineMap(ConcurrentHashMap<String,BeanDefinition> beanDefineMap){        MyBeanFactoryImpl.beanDefineMap = beanDefineMap;    }    public static void setBeanNameSet(Set<String> beanNameSet){        MyBeanFactoryImpl.beanNameSet = beanNameSet;    }}

此时初始化的阶段已经完成了，即已经将所有被@MyIoc标记的类已经被一律存放在了自己设置的容器中了。其实在这里我们已经能使用自己的自己设置的容器进行取得Bean了。

@MyIoc@Datapublic class User {    private Student student;}

@MyIocpublic class Student {    public String play(){        return "student"+ this.toString();    }}

此时我们在启动类中写如下

        User user1 = (User)beanFactory.getBeanByName("com.example.ioc.domain.User");        User user2 = (User)beanFactory.getBeanByName("com.example.ioc.domain.User");        Student student1 = user1.getStudent();        Student student2 = user1.getStudent();        Student student3 = (Student)beanFactory.getBeanByName("com.example.ioc.domain.Student");        System.out.println(user1);        System.out.println(user2);        System.out.println(student1);        System.out.println(student2);        System.out.println(student3);

发现控制台中输出的对象都是同一个对象，并且在User中也自动注入了Student对象。此时一个简单的IOC就完成了。

User(student=com.example.ioc.domain.Student@705e7b93)User(student=com.example.ioc.domain.Student@705e7b93)com.example.ioc.domain.Student@705e7b93com.example.ioc.domain.Student@705e7b93com.example.ioc.domain.Student@705e7b93

总结

原本一开始的想法的是想要写一个相似于@Autowired注解的自己设置注解，但是在编码过程中遇到了一个困难，就是例如下面的代码，实例化B容易，但是如何将B注入到每一个实例化的A中，这个问题困扰了我好几天，也查找了许多的资料，至今还是没有处理，预计是只有研究Spring源码才能够理解是如何做到的。

@MyIocpublic class A{@MyIocUseprivate B b;}

完整项目地址

参考文章

• https://juejin.im/post/5a5875a4518825733a30a463
• https://juejin.im/entry/599f8ba6518825241f788ad1