自己动手编写spring IOC源码

前言

对于spring IOC概念不是很了解的朋友可以阅读我上一篇博客__轻松理解spring IOC 通过这篇博客的理解之后，相信大家会对spring的IOC概念会有进一步的理解。接下来我先预览一下本例中java的类图关系。

解析：我们有一个Master接口，接口中定义了一个WalkDog()遛狗的方法，Hostess是对这个接口的具体实现。然后我们有一个Dog接口，接口中有一个bark()方法，Labuladuo和Taidi是对其的实现。最后我们的程序入口Client类调用Hostess对象的WalkDog方法。
需求：Hostess对象遛狗需要一个狗对象，目前我们的类中有两个符合需求的对象，我们只要在配置文件中进行相关配置便可以指定我们的Hostess对象调用的是哪一个具体的Dog对象。

public static void main(String[] args) {
        ApplicationContext context = new ClassPathXmlApplicationContext("applicationContext.xml");
        Master master = (Master)context.getBean("hostess");
        
        System.out.println();
        System.out.println();
        System.out.println();
        System.out.println("***********************************");
        master.WalkDog();
    }

解析：从main方法的前两句原spring的代码中我们可以猜想，spring框架中一定是定义了ApplicationContext这个接口，并且接口中定义了一个getBean()的方法，而ClassPathXmlApplicationContext类肯定是对其的实现。既然是我们自己动手写spring框架，我们把这个接口和类实现了也就可以了。

接口 ApplicationContext

public interface ApplicationContext {
    public Object getBean(String beanid);
}

实现类 ClassPathXmlApplicationContext

package com;

import java.io.File;
import java.lang.reflect.Method;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.io.SAXReader;

public class ClassPathXmlApplicationContext implements ApplicationContext {

    private String fileName;

    public ClassPathXmlApplicationContext(String fileName){
        this.fileName = fileName;
    }

    @Override
    public Object getBean(String beanid) {
        //获取本类的当前目录
        String currentPath = this.getClass().getResource("").getPath().toString();

        SAXReader reader = new SAXReader();//DOM4J解释器
        Document doc = null;//xml文档本身
        Object obj = null;//目标表创建出来的实例
        try {
            doc = reader.read(  new File(currentPath+fileName)  );
            String xpath = "/beans/bean[@id='"+beanid+"']";
            Element beanNode = (Element) doc.selectSingleNode(xpath);
            String className = beanNode.attributeValue("class");
            obj = Class.forName(className).newInstance();

            Element propertyNode = (Element) beanNode.selectSingleNode("property");

            if(propertyNode!=null){
                System.out.println("当前bean有属性需要注入");

                String propertyName = propertyNode.attributeValue("name");
                System.out.println("当前bean需要注入的属性为"+propertyName);

                //拼接出注入方法
                String setMethod = "set"+(propertyName.substring(0, 1)).toUpperCase()+propertyName.substring(1,propertyName.length());
                System.out.println("自动调用注入方法"+setMethod);

                String set_object_name = propertyNode.attributeValue("ref");
                System.out.println("需要注入的对象名"+set_object_name);

                Object di_object = getBean(set_object_name);
                System.out.println("注入的对象实例"+di_object);

                Method []methods = obj.getClass().getMethods();

                for (Method m : methods) {
                    if(setMethod.equals(m.getName())  ) {
                        m.invoke(obj, di_object);
                        break;
                    }
                }

            }else{
                System.out.println("当前bean没有属性，无需注入直接结束");
            }

        } catch (Exception e) {
            e.printStackTrace();
        }


        return obj;
    }

}

配置文件 applicationContext.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans>
    <bean id="hostess" class="com.Hostess">
        <property name="dog" ref="Taidi_dog"></property>
    </bean>

    <bean id="Taidi_dog" class="com.Taidi"></bean>

    <bean id="Labuladuo_dog" class="com.Labuladuo"></bean>
</beans>

解析：① 我们的applicationContext.xml文件主要是配置我们的java bean。这里我们自己写一份这样的文件通知我们自己的框架有哪些对象需要注入。
② 接口 ApplicationContext 这里我只是定义了一个方法就不多解释了。
③ 实现类 ClassPathXmlApplicationContext 主要是解析我们的xml文件然后构造实例的一个类。解析xml文件我们主要使用的是dom4j，获取各个节点和节点属性与属性值。创建对象则是通过反射的方式构造对象 [obj = Class.forName(className).newInstance();]。在判断一个对象是否有属性需要注入则是使用递归算法对其一一注入。

最后：我们来看一下运行结果