自己实现Spring IoC容器（四）IoC容器的Bug

Bug的发现

之前我自己写了一个类似Spring中的IoC容器 自己实现Spring IoC容器（三）完成IoC容器，然后最近想在这个项目基础上把Spring的AOP也实现一下，然后就悲剧的发现了一句错误代码……

这个错误代码就在edu.jyu.core.ClassPathXmlApplicationContext类的Object createBeanByConfig(Bean bean)方法中，下面是这个方法

/** * 根据bean的配置信息创建bean对象 *  * @param bean * @return */private Object createBeanByConfig(Bean bean) {    // 根据bean信息创建对象    Class clazz = null;    Object beanObj = null;    try {        clazz = Class.forName(bean.getClassName());        // 创建bean对象        beanObj = clazz.newInstance();        // 获取bean对象中的property配置        List<Property> properties = bean.getProperties();        // 遍历bean对象中的property配置,并将对应的value或者ref注入到bean对象中        for (Property prop : properties) {            Map<String, Object> params = new HashMap<>();            if (prop.getValue() != null) {                params.put(prop.getName(), prop.getValue());                // 将value值注入到bean对象中                BeanUtils.populate(beanObj, params);            } else if (prop.getRef() != null) {                Object ref = context.get(prop.getRef());                // 如果依赖对象还未被加载则递归创建依赖的对象                if (ref == null) {                    ref = createBeanByConfig(bean);                }                params.put(prop.getName(), ref);                // 将ref对象注入bean对象中                BeanUtils.populate(beanObj, params);            }        }    } catch (Exception e1) {        e1.printStackTrace();        throw new RuntimeException("创建" + bean.getClassName() + "对象失败");    }    return beanObj;}

错误就在如果依赖对象还未被加载条件成立后，ref = createBeanByConfig(bean); 这句代码的问题是什么了，很明显我一不小心又把当前要创建的bean对象的配置信息传入createBeanByConfig方法中了，所以就会无限递归下去，最后发生StackOverflowError错误。

至于为什么我的测试代码能通过也是比较凑巧，我的测试bean是一个A类，一个B类，其中B类依赖A类对象，所以我们要把A类对象注入到B类中，但是就是这么巧，读取配置文件的时候先读到了A类，所以在要创建B类对象时，A类对象已经创建好了，ref == null就为false，也就是说没执行到那句错误代码，所以就没发现……

要是我改为A类依赖B类，那就可以发现问题了，因为要创建A类对象时，B类对象还没创建。

A类

package edu.jyu.bean;public class A {    private String name;    private B b;    public B getB() {        return b;    }    public void setB(B b) {        this.b = b;    }    public String getName() {        return name;    }    public void setName(String name) {        this.name = name;    }}

B类

package edu.jyu.bean;public class B {    private int age;    public int getAge() {        return age;    }    public void setAge(int age) {        this.age = age;    }}

此时配置文件applicationContext.xml也需要修改一下

<?xml version="1.0" encoding="UTF-8"?><beans>    <bean name="A" class="edu.jyu.bean.A">        <property name="name" value="Jason"></property>        <property name="b" ref="B"></property>    </bean>    <bean name="B" class="edu.jyu.bean.B" scope="prototype">        <property name="age" value="13"></property>    </bean></beans>

测试类TestApplicationContext也改一下

package edu.jyu.core;import org.junit.Test;import edu.jyu.bean.A;import edu.jyu.bean.B;public class TestApplicationContext {    @Test    public void test() {        BeanFactory ac = new ClassPathXmlApplicationContext("/applicationContext.xml");        A a = (A) ac.getBean("A");        A a1 = (A) ac.getBean("A");        B b = (B) ac.getBean("B");        B b1 = (B) ac.getBean("B");        System.out.println(a.getB());        System.out.println("a==a1 : "+(a==a1));        System.out.println("b==b1 : "+(b==b1));    }}

运行这个测试，你就会惊喜地发现爆栈了

Bug的解决

解决上面的那个Bug并不难，只需要把那句错误代码ref = createBeanByConfig(bean);换成ref = createBeanByConfig(config.get(prop.getRef()));

完整方法

/** * 根据bean的配置信息创建bean对象 *  * @param bean * @return */private Object createBeanByConfig(Bean bean) {    // 根据bean信息创建对象    Class clazz = null;    Object beanObj = null;    try {        clazz = Class.forName(bean.getClassName());        // 创建bean对象        beanObj = clazz.newInstance();        // 获取bean对象中的property配置        List<Property> properties = bean.getProperties();        // 遍历bean对象中的property配置,并将对应的value或者ref注入到bean对象中        for (Property prop : properties) {            Map<String, Object> params = new HashMap<>();            if (prop.getValue() != null) {                params.put(prop.getName(), prop.getValue());                // 将value值注入到bean对象中                BeanUtils.populate(beanObj, params);            } else if (prop.getRef() != null) {                Object ref = context.get(prop.getRef());                // 如果依赖对象还未被加载则递归创建依赖的对象                if (ref == null) {                    //下面这句的错误在于传入了当前bean配置信息，这会导致不断递归最终发生StackOverflowError                    //解决办法是传入依赖对象的bean配置信息                    //ref = createBeanByConfig(bean);                    ref = createBeanByConfig(config.get(prop.getRef()));                }                params.put(prop.getName(), ref);                // 将ref对象注入bean对象中                BeanUtils.populate(beanObj, params);            }        }    } catch (Exception e1) {        e1.printStackTrace();        throw new RuntimeException("创建" + bean.getClassName() + "对象失败");    }    return beanObj;}

现在运行测试类TestApplicationContext的测试方法就没问题了