自己写IOC-----第二篇

第一篇中介绍了读取xml节点，现在就可以按照配置要求生产对象了：个人没怎么重构代码就这样写了

反射操作类：

 

  package com.reflect;

import java.lang.reflect.Method;

public class IOCReflect {
	private static IOCReflect install = null;

	private IOCReflect() {

	}

	public synchronized static IOCReflect getInstall() {

		if (install == null) {
			install = new IOCReflect();
		}
		return install;
	}

	/**
	 *  根据class名生成对象
	 */
	public Object newInstallByClass(String classname) throws Exception {
		return Class.forName(classname).newInstance();
	}

	/**
	 * set注入对象
	 */
	public void setObject(Object bean, Object refOb, String refName)
			throws Exception {
		String refset = refName.substring(0, 1).toUpperCase()
				+ refName.substring(1);

		String pName = "set" + refset;
		System.out.println("refset---" + refset + "---bean---"
				+ bean.toString() + "  pname--" + pName);
		Method method = bean.getClass().getMethod(pName, refOb.getClass());
		System.out
				.println("method---" + method.getName() + "--refob--" + refOb);
		if (method != null) {

			method.invoke(bean, refOb);
			System.out.println("注入成功");
		} else {
			System.out.println("注入失败");
		}
	}
}

 

上面基本实现了set注入功能

下面看 IOC操作类

 

package com.ioc;

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

import org.jdom2.Element;

import com.jdom.xml.IOCdomXml;
import com.reflect.IOCReflect;

/**
 * set 注入功能
 * @author lijunqing
 *
 */
public class IOC1 {
	private Map<String, Object> beanOb = null;

	public IOC1(String xmlPath) throws Exception {
		beanOb = objectBean(xmlPath);
		setObjectByRef(xmlPath);
	}

	/**
	 * 读取xml文件，保存生成相应bean的对象
	 * @throws Exception 
	 */
	public Map<String, Object> objectBean(String xmlPath) throws Exception {

		String beanID = null;
		Map<String, Object> obMap = new HashMap<String, Object>();
		Map<String, Element> beanMap = new HashMap<String, Element>();
		beanMap = IOCdomXml.getInstall().getBeanElement(xmlPath);
		for (Map.Entry<String, Element> beanEntry : beanMap.entrySet()) {
			beanID = beanEntry.getKey();
			String beanName = beanEntry.getValue().getAttributeValue("class");
			//根据 beanName生成对象
			System.out.println("beanName----" + beanName + "---beanID--"
					+ beanID);
			Object beanOb = IOCReflect.getInstall().newInstallByClass(beanName);
			obMap.put(beanID, beanOb);
		}
		return obMap;
	}

	/*
	 * set注入对象
	 */
	public void setObjectByRef(String xmlPath) throws Exception {
		Map<String, Object> obMap = objectBean(xmlPath);
		Map<String, Element> refBeanMap = null;
		Object bean = null;
		Map<String, Map<String, Element>> refMap = IOCdomXml.getInstall()
				.getRefBeanElements(xmlPath);
		for (Map.Entry<String, Map<String, Element>> entry : refMap.entrySet()) {
			String beanID = entry.getKey();
			bean = obMap.get(beanID);
			refBeanMap = entry.getValue();
			for (Map.Entry<String, Element> refBean : refBeanMap.entrySet()) {
				String refName = refBean.getKey();
				String refBeanID = refBean.getValue().getAttributeValue("ref");
				//判断bean对象中是否用set加上这个refName的set方法
				System.out.println("refName--" + refName + "     refBeanID--"
						+ refBeanID);
				//拿出refBeanID对象
				Object refOb = obMap.get(refBeanID);
				if (refOb != null) {
					//		    		    	    //判断有没有set方法，调用bean对象的Method方法插入对象
					//		    		    	     Method beanMethod = bean.getClass().getMethod("set"+refName);
					//		    		    	     if(beanMethod != null){
					//		    		    	    	    beanMethod.invoke(bean, refOb);
					//		    		    	     }
					IOCReflect.getInstall().setObject(bean, refOb, refName);
				}

			}
		}
	}

	public Object getBean(String beanID) {
		return beanOb.get(beanID);
	}
}

 

下面是测试类

 

package com.test;

import org.junit.Test;

import com.domain.Person;
import com.ioc.IOC1;
import com.servce.LoginServce;

public class TestIOC {
      @Test
      public void TestgetBean() throws Exception{
    	    IOC1 ioc = new IOC1("beans.xml");
    	    LoginServce login =(LoginServce)ioc.getBean("loginServce");   
    	    login.getPersonInfo();
    	    Person p = (Person) ioc.getBean("person");
    	    if(login.getPerson() != null){
    	    	   System.out.println("pname----"+login+"person--"+p);
    	    }
    	    
    	    
      }
}

 

测试结果如下：

beanName----com.domain.Person---beanID--person
beanName----com.servce.LoginServce---beanID--loginServce
beanName----com.domain.Person---beanID--person
beanName----com.servce.LoginServce---beanID--loginServce
propety------- prooety
refName--person     refBeanID--person
refset---Person---bean---com.servce.LoginServce@f4f44a  pname--setPerson
method---setPerson--refob--<name>ljq</name><age>13</age>
注入成功
pname----com.servce.LoginServce@1d256faperson--<name>ljq</name><age>13</age>