spring ioc模块手写demo

package com.kkb.spring.test; import java.io.InputStream; import java.lang.reflect.Constructor; import java.lang.reflect.Field; import java.lang.reflect.Method; import java.util.HashMap; import java.util.List; import java.util.Map; import org.dom4j.Document; import org.dom4j.DocumentException; import org.dom4j.Element; import org.dom4j.io.SAXReader; import org.junit.Test; import com.kkb.spring.ioc.BeanDefinition; import com.kkb.spring.ioc.PropertyValue; import com.kkb.spring.ioc.RuntimeBeanReference; import com.kkb.spring.ioc.TypedStringValue; import com.kkb.spring.po.User; import com.kkb.spring.service.UserService; public class TestSpringV2222 { private Map<String, BeanDefinition> beanDefinitions = new HashMap<>(); /** * 专门存储单例bean实例的集合 */ private Map<String, Object> singletonObjects = new HashMap<String, Object>(); @Test public void test() throws Exception { // 读取XML中的配置信息，封装成BeanDefinition，并且注册到Map集合中 loadAndregisterBeanDefinitions("beans.xml"); // 获取UserService的bean实例 UserService userService = (UserService) getBean("userService"); // 调用UserService的方法 // 入参对象 Map<String, Object> param = new HashMap<>(); param.put("username", "王五"); // 根据用户名称查询用户信息 List<User> users = userService.queryUsers(param); System.out.println("结果：" + users); } private Object getBean(String name) { // 从缓存中获取要找的对象 Object singleton = this.singletonObjects.get(name); if (singleton != null) { return singleton; } // 找不到，则获取指定名称的BeanDefinition对象 // 此处使用到的就是抽象模板方法，我此处只定流程，不去实现，我也不懂如何实现，这不是我干的事情 BeanDefinition bd = this.beanDefinitions.get(name); // 根据BeanDefinition中的信息，判断是单例还是多例（原型） if (bd.isSingleton()) { // 单例 // 根据BeanDefinition对象，完成bean的创建 singleton = createBean(bd); // 缓存已经创建的单例bean实例 this.singletonObjects.put(name, singleton); } else if (bd.isPrototype()) {// 原型 // 根据BeanDefinition对象，完成bean的创建 singleton = createBean(bd); } return singleton; } private Object createBean(BeanDefinition bd) { Class<?> clazz = bd.getClazzType(); if (clazz == null) { return null; } // 实例化bean // 注意：此时只是new了一个空对象 Object sinleton = createBeanInstance(clazz); // bean的属性填充 populateBean(sinleton, bd); // 初始化bean initBean(sinleton, bd); return sinleton; } private void initBean(Object sinleton, BeanDefinition bd) { // TODO 完成Aware接口（标记接口）相关的处理,spring mvc代码会用到 // 1. aware接口的处理 // TODO BeanPostProcessor的前置方法执行 initMethod(sinleton, bd); // TODO BeanPostProcessor的后置方法执行（AOP代理对象产生的入口） } private void initMethod(Object sinleton, BeanDefinition bd) { // TODO 完成InitializingBean接口（标记接口）的处理，调用的是afterPropertySet方法 // 完成init-metho标签属性对应的方法调用 invokeMethod(sinleton, bd.getInitMethod()); } private void setProperty(Object beanInstance, String name, Object valueToUse) { try { Class<?> clazz = beanInstance.getClass(); Field field = clazz.getDeclaredField(name); field.setAccessible(true); field.set(beanInstance, valueToUse); } catch (Exception e) { e.printStackTrace(); } } private void invokeMethod(Object beanInstance, String initMethod) { try { if (initMethod == null || "".equals(initMethod)) { return; } Class<?> clazz = beanInstance.getClass(); Method method = clazz.getDeclaredMethod(initMethod); // 设置允许访问私有方法和变量，此处也称之为暴力破解 method.setAccessible(true); method.invoke(beanInstance); } catch (Exception e) { e.printStackTrace(); } } private void populateBean(Object sinleton, BeanDefinition bd) { List<PropertyValue> propertyValues = bd.getPropertyValues(); for (PropertyValue propertyValue : propertyValues) { // 获取属性名称 String name = propertyValue.getName(); // 未处理的value对象 Object value = propertyValue.getValue(); // 解决之后的value值 Object valueToUse = null; if (value instanceof TypedStringValue) { TypedStringValue typedStringValue = (TypedStringValue) value; String stringValue = typedStringValue.getValue(); // 获取参数的类型 Class<?> targetType = typedStringValue.getTargetType(); // TODO 建议使用策略模式进行优化 if (targetType == Integer.class) { valueToUse = Integer.parseInt(stringValue); } else if (targetType == String.class) { valueToUse = stringValue; } else { // .... } } else if (value instanceof RuntimeBeanReference) { RuntimeBeanReference reference = (RuntimeBeanReference) value; // 递归获取指定名称的bean实例 // TODO 此处可能会发送循环依赖问题 valueToUse = getBean(reference.getRef()); } else { // .... } // 利用反射去设置bean的属性 setProperty(sinleton, name, valueToUse); } } private Object createBeanInstance(Class<?> clazz) { // TODO 可以根据bean标签的配置选择使用实例工厂去创建Bean // TODO 可以根据bean标签的配置选择使用静态工厂去创建Bean // 还可以选择使用我们的构造方法去创建Bean return createObject(clazz); } private Object createObject(Class<?> clazz, Object... args) { try { // TODO 可以根据输入参数获取指定构造参数的构造方法 Constructor<?> constructor = clazz.getConstructor(); // 默认调用无参构造进行对象的创建 return constructor.newInstance(args); } catch (Exception e) { e.printStackTrace(); } return null; } private void loadAndregisterBeanDefinitions(String location) { // 获取流对象 InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream(location); // 创建文档对象 Document document = createDocument(inputStream); // 按照spring定义的标签语义去解析Document文档 registerBeanDefinitions(document.getRootElement()); } private Document createDocument(InputStream inputStream) { Document document = null; try { SAXReader reader = new SAXReader(); document = reader.read(inputStream); return document; } catch (DocumentException e) { e.printStackTrace(); } return null; } @SuppressWarnings("unchecked") public void registerBeanDefinitions(Element rootElement) { // 获取<bean>和自定义标签（比如mvc:interceptors） List<Element> elements = rootElement.elements(); for (Element element : elements) { // 获取标签名称 String name = element.getName(); if (name.equals("bean")) { // 解析默认标签，其实也就是bean标签 parseDefaultElement(element); } else { // 解析自定义标签，比如mvc:interceptors标签回去 parseCustomElement(element); } } } @SuppressWarnings("unchecked") private void parseDefaultElement(Element beanElement) { try { if (beanElement == null) return; // 获取id属性 String id = beanElement.attributeValue("id"); // 获取name属性 String name = beanElement.attributeValue("name"); // 获取class属性 String clazzName = beanElement.attributeValue("class"); if (clazzName == null || "".equals(clazzName)) { return; } Class<?> clazzType = Class.forName(clazzName); // 获取init-method属性 String initMethod = beanElement.attribu