Java简单模拟实现Spring

前言

为什么要用Spring？可能大多数回答是：公司的人都在用，我就用了。当然，这是一个很重要的原因。众所周知，Spring的两大特性IOC(DI)和AOP。我一直把Spring当成一个大的工厂模式，Spring容器在初始化过程中，会装载各个在XML文件中已经声明的Java bean。并且会通过XML文件中bean之间的依赖关系，实现自动装配和注入。除此之外，还需提供一个Map容器，用KV的方式装所有的Java bean。因此简单的实现一个Spring需要以下三点：

1、XML解析器；

2、Java bean容器即Map，并提供一个context环境；

3、利用Java反射实现自动装配和依赖注入。

实现

代码结构如下：

上图中红线框出的部分为模拟Spring以及Spring的测试用例。

首先写一个接口BeanFactory，里面有一个getBean的方法。

BeanFactory.java

public interface BeanFactory {
    public Object getBean(String name);
}

ClassPathXmlApplicationContext文件完成了Spring上下文的加载，可以读取XML文件，将声明的bean装载进容器中，并完成依赖注入。

ClassPathXmlApplicationContext.java

import org.jdom.Document;
import org.jdom.Element;
import org.jdom.input.SAXBuilder;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created by jichenxiao on 2016/12/20.
 */
public class ClassPathXmlApplicationContext implements BeanFactory{
    private Map<String, Object> beans = new HashMap<String, Object>();

    public ClassPathXmlApplicationContext() throws Exception {
        SAXBuilder sb = new SAXBuilder();
        Document doc = sb.build(this.getClass().getClassLoader().getResourceAsStream("beans.xml"));
        Element root = doc.getRootElement();
        List list = root.getChildren("bean");
        for(int i = 0; i < list.size(); i++){
            Element element = (Element) list.get(i);
            String id = element.getAttributeValue("id");
            String clazz = element.getAttributeValue("class");
            Object obj = Class.forName(clazz).newInstance();
            beans.put(id, obj);
            for(Element propertyElement : (List<Element>)element.getChildren("property")){
                // userDao
                String name = propertyElement.getAttributeValue("name");
                // u
                String bean = propertyElement.getAttributeValue("bean");
                // UserDaoImpl instance
                Object beanObject = beans.get(bean);
                String methodName = "set" + name.substring(0,1).toUpperCase() + name.substring(1);
                System.out.println("method name = " + methodName);
                // setUserDao(UserDao.class)
                Method m = obj.getClass().getMethod(methodName, beanObject.getClass().getInterfaces()[0]);
                m.invoke(obj, beanObject);
            }
        }
    }
    public Object getBean(String name) {
        return beans.get(name);
    }
}

代码中首先获取XML文件中为bean的元素，然后解析其属性id和class，通过反射Class.forName获取到Class对象后再初始化一个实例，并将其添加到Map容器中，K为属性id，V为属性class。随后再去遍历其名为property的子节点，通过Java反射获取set方法，并通过set方法将当前实例依赖的实例注入进来。

以下是其他代码：

实体类User.java

public class User {
    private String username;
    private String password;

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }
}

Dao（Data Access Object）层

UserDao.java

public interface UserDao {
    public void save(User u);
}

UserDaoImpl.java

public class UserDaoImpl implements UserDao{
    public void save(User u) {
        System.out.println("User saved...");
    }
}

Service层

UserService.java

public class UserService {
    private UserDao userDao;

    public UserDao getUserDao() {
        return userDao;
    }

    public void setUserDao(UserDao userDao) {
        this.userDao = userDao;
    }

    public void addUser(User u){
        this.userDao.save(u);
    }
}

XML文件

<beans>
    <bean id="u" class="com.jd.jr.risk.springsim.dao.impl.UserDaoImpl"></bean>
    <bean id="userService" class="com.jd.jr.risk.springsim.service.UserService">
        <property name="userDao" bean="u"/>
    </bean>
</beans>

分层的原因是为了实现面向接口编程，如果有不同的dao或者service实现方式，只需要去实现它的接口即可，然后再通过其名字获取到期实例。

最后测试用例：

UserServiceTest.java

public class UserServiceTest {
    @Test
    public void testAdd() throws Exception {
        BeanFactory factory = new ClassPathXmlApplicationContext();
        UserService service = (UserService) factory.getBean("userService");
        User u = new User();
        service.addUser(u);
    }
}

运行结果为：

method name = setUserDao
User saved...