模仿Spring实现注解注入

写这个极其蛋疼，我一直在想我们用SSM写项目时，写Service和Controller的时候，会给Service和Controller私有属性，比如Service需要dao，Controller需要Service，但是我们没有写过setter方法，而且也没有写带参构造器。那么它是怎么注入的呢？

我绞尽脑汁，用了Field类的 set（Object,Object）办法，发现不能对private修饰的私有属性进行注入，其实我已经很接近答案了。但是！我辗转了一个晚上，才知道setAccessible（boolean）这个方法可以强行干进去。。。

不说了，一会儿还要睡觉，直接上代码。我用的是注解注入的，而且稍微修改了昨晚mapper的代码，最后目录被我改成这样了。。。

把昨晚的DaoProxy改成了DaoFactory。我其实是想写工厂模式的，但是我没有这个经验，现在先把基础功能写起来，后面再看看设计模式改造一下吧。。。

下面这些接受困难的可以先移步

模仿Mybatis用mapper.xml实现Dao层接口的功能

最后DaoProxy被我改成了这样。。。暴露出来的getBean方法里的参数是在xml文件中读取的。。。

package com.yck.yebatis;

import java.io.File;
import java.io.FileFilter;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
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 com.yck.bean.Function;
import com.yck.bean.MapperBean;
import com.yck.exception.NoConfigFileException;
import com.yck.util.DataUtil;

public class DaoFactory implements BeanFactory
{

    private static Map<String,Object> beans;

    static
    {
        configDao();
    }

    @Override
    public Object getBean(String beanName) //暴露获取Bean的方法
    {
        return beans.get(beanName);
    }

    private static void configDao() //初始化
    {
        Map<String,Object> map = new HashMap<String,Object>();
        try
        {
            File[] files = getAllFiles();
            for(File f:files)
            {
                MapperBean mapper = readMapper(f.getPath());
                Object obj = implDao(mapper);
                map.put(mapper.getProxyName(), obj);
            }
        } catch (NoConfigFileException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        beans = map;
    }

    /**
     * 得到所有的mapper.xml文件
     * @return
     * @throws NoConfigFileException
     */
    private static File[] getAllFiles() throws NoConfigFileException
    {
        File configPath = new File("src/mapper"); //为了简单起见，规定写在该目录下
        FileFilter fileFilter = new FileFilter()  //写过滤器，筛选后缀为.xml的文件
        {

            @Override
            public boolean accept(File pathname)
            {
                String str = pathname.getName().toLowerCase();
                if(str.endsWith(".xml"))
                    return true;
                return false;
            }
        };

        File[] files = configPath.listFiles(fileFilter);
        if(files == null || files.length == 0)  //如果没有这样的文件，抛出异常
        {
            files = null;
            throw new NoConfigFileException();
        }
        return files;
    }

    /**
     * 通过读取配置文件实现dao接口
     * @param path
     * @return
     */
    private  static Object implDao(MapperBean mapper)
    {
        ClassLoader classLoader = DaoFactory.class.getClassLoader();
        Class<?> interfaze = null;
        try
        {
            interfaze = classLoader.loadClass(mapper.getInterfaceName()); //加载一个接口类
        } catch (ClassNotFoundException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        /**
         * 下面这几句相当重要了，是用xml文件实现dao接口的核心代码，因为数据库相关的大量工作我之前都写过了，所以这个看起来代码量很少
         * 我也不太懂下面这个东西，我查API查了相当久，一开始写总是错，因为我理解错了InvocationHandler接口下那个方法的Object数组参数
         * 它应该理解为一个可变长数组，而不是必须为数组
         */
        Object instanze = Proxy.newProxyInstance(classLoader, new Class[]{interfaze}, new InvocationHandler(){

            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable
            {
                List<Function> list = mapper.getList();
                Object obj = null;
                for(Function f:list)
                {
                    if(f.getFuncName().equals(method.getName()))
                    {
                        /**
                         * 判断是不是select语句，是则调用DateUtil的select方法
                         * 否则调用update的方法
                         */
                        if(f.getSqltype().equals("select"))
                        {
                            if("java.util.ArrayList".equals(f.getResultType()))
                            {
                                if(f.isParameter())
                                    obj = DataUtil.selectForBeanList(Class.forName(f.getResultOf()), f.getSql(), args);
                                else obj = DataUtil.selectForBeanList(Class.forName(f.getResultOf()), f.getSql());
                            }
                            else
                            {
                                if(f.isParameter())
                                    obj = DataUtil.selectForBean(Class.forName(f.getResultType()), f.getSql(), args);
                                else obj = DataUtil.selectForBean(Class.forName(f.getResultType()), f.getSql());
                            }

                        }

                        else
                        {
                            if(f.isParameter())
                                obj = DataUtil.updata(f.getSql(), args);
                            else obj = DataUtil.updata(f.getSql());
                        }
                    }
                }
                return obj;
            }
        });
        return instanze;
        //返回这个接口，即mapper.getInterfaceName()这个接口

    }

    /**
     * 读取xml文件的信息
     * @param path
     * @return
     */
    private static MapperBean readMapper(String path)
    {
        File file = new File(path);
        SAXReader reader = new SAXReader();
        MapperBean mapper = new MapperBean();
        try
        {
            Document doc = reader.read(file);
            Element root = doc.getRootElement(); //读取根节点 即dao节点
            mapper.setInterfaceName(root.attributeValue("class").trim()); //把dao节点的class值存为接口名
            mapper.setProxyName(root.attributeValue("id").trim());       //把id值设为代理名
            List<Function> list = new ArrayList<Function>(); //用来存储方法的List
            for(Iterator<?> rootIter = root.elementIterator();rootIter.hasNext();) //遍历根节点下所有子节点
            {
                Function fun = new Function();    //用来存储一条方法的信息
                Element e = (Element) rootIter.next();
                String sqltype = e.getName().trim();
                String funcName = e.attributeValue("id").trim();
                String sql = e.getText().trim();
                String resultType = e.attributeValue("resultType").trim();
                String resultOf = "";
                if("java.util.ArrayList".equals(resultType))
                    resultOf = e.attributeValue("resultOf").trim();
                String parameter = e.attributeValue("parameter");
                fun.setSqltype(sqltype);
                fun.setFuncName(funcName);
                fun.setResultType(resultType);
                fun.setSql(sql);
                fun.setResultOf(resultOf);
                fun.setParameter("true".equals(parameter));
                list.add(fun);
            }
            mapper.setList(list);

        } catch (DocumentException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return mapper;
    }

下面写ServiceFactory我自己写了个注解

package com.yck.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
public @interface SelfInject
{
    String value();

}

在Service类中使用了这个注解

package com.yck.service;

import com.yck.annotation.SelfInject;
import com.yck.bean.User;
import com.yck.dao.IUserDao;

public class UserService
{
    @SelfInject(value="userdao")
    private IUserDao userdao;

    public User find(Integer id)
    {
        return userdao.selectById(id);
    }

}

下面是ServiceFactory，先贴配置文件

<?xml version="1.0" encoding="UTF-8"?>
<services>
    <service id="userService" class="com.yck.service.UserService"/>
</services>

package com.yck.yebatis;

import java.io.File;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
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 com.yck.annotation.SelfInject;
import com.yck.bean.ServiceBean;

public class ServiceFactory implements BeanFactory
{
    private static DaoFactory daoFactory = new DaoFactory();
    private static Map<String,Object> beans;

    static
    {
        configService();
    }

    /**
     * 暴露获取方法
     */
    @Override
    public Object getBean(String beanName)
    {
        return beans.get(beanName);
    }

    /**
     * 静态初始化方法
     */
    private static void configService()
    {
        Map<String,Object> map = new HashMap<String,Object>();
        List<ServiceBean> list= getServiceBeans();
        for(ServiceBean service:list)
        {
            map.put(service.getProxyName(), implService(service));
        }
        beans = map;
    }

    /**
     * 读取所有的Service配置
     * @return
     */
    private static List<ServiceBean> getServiceBeans()
    {
        SAXReader reader = new SAXReader();
        List<ServiceBean> list = new ArrayList<ServiceBean>();
        try
        {
            Document doc = reader.read(new File("src/service/service.xml"));
            Element root = doc.getRootElement();
            for(Iterator<?> iter = root.elementIterator("service");iter.hasNext();)
            {
                Element e = (Element) iter.next();
                String proxyName = e.attributeValue("id");
                String className = e.attributeValue("class");
                ServiceBean bean = new ServiceBean(proxyName, className);
                list.add(bean);

            }
        } catch (DocumentException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return list;

    }

    /**
     * 实例化Service并注入所需依赖
     * @param service
     * @return
     */
    public static Object implService(ServiceBean service)
    {
        Object obj = null;
        try
        {
            obj = Class.forName(service.getClassName()).newInstance();
            Class<?> c = obj.getClass();
            Field[] fields = c.getDeclaredFields();
            for(Field f:fields)
            {
                Annotation[] annotations = f.getAnnotations();
                for(Annotation a:annotations)
                {
                    if(a instanceof SelfInject)
                    {
                        Object o = daoFactory.getBean(((SelfInject) a).value());
                        try
                        {
                            f.setAccessible(true);
                            f.set(obj,o);
                            f.setAccessible(false);
                        } catch (IllegalArgumentException | IllegalAccessException e)
                        {
                            // TODO Auto-generated catch block
                            e.printStackTrace();
                        }
                    }
                }
            }

        } catch (ClassNotFoundException | InstantiationException | IllegalAccessException e)
        {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return obj;
    }
}

最后测试有没有注入

package com.yck.test;

import com.yck.bean.User;
import com.yck.dao.IUserDao;
import com.yck.service.UserService;
import com.yck.yebatis.DaoFactory;
import com.yck.yebatis.ServiceFactory;

public class Test2
{

    public static void main(String[] args)
    {
        DaoFactory bf = new DaoFactory();
        IUserDao userdao = (IUserDao) bf.getBean("userdao");
        User u = userdao.selectById(2);
        System.out.println(u);

        ServiceFactory sf = new ServiceFactory();
        UserService userService = (UserService) sf.getBean("userService");
        User us = userService.find(2);

        System.out.println(us);

    }

}

最后查出结果，说明已经通过注解注入了

<*******************************************************************一条分割线*************************************************************************>

我发现还有个功能没有测试。。。我在写这个功能的时候是有考虑到多个dao的实现，所以我读取配置的时候写的是读取src/mapper目录下所有的以.xml结尾的文件，写依赖注入的时候也是读取<Services>下所有<service>子节点，所以我再加了一个表测试这个表如下和一些初始数据如下。

写了个Fruit的bean，我这就不贴了。。。。

下面是dao接口和service

package com.yck.dao;

import java.util.List;

import com.yck.bean.Fruit;

public interface IFruitDao
{
    int insert(String name,Integer numbers);

    int deleteById(Integer id);

    Fruit selectById(Integer id);

    Fruit updateNum(Integer nums,Integer id);

    List<Fruit> selectAll();

}

package com.yck.dao;

import java.util.List;

import com.yck.bean.Fruit;

public interface IFruitDao
{
    int insert(String name,Integer numbers);

    int deleteById(Integer id);

    Fruit selectById(Integer id);

    Fruit updateNum(Integer nums,Integer id);

    List<Fruit> getAll();

}

fruitdao的配置文件如下fruitdao.xml

<?xml version="1.0" encoding="UTF-8"?>
<dao id="fruitdao" class="com.yck.dao.IFruitDao">
    <select id="selectById" resultType ="com.yck.bean.Fruit" parameter="true">
        select * from t_fruit where id = ?
    </select>

    <update id="updateNum" resultType = "java.lang.Integer" parameter="true">
        update t_fruit set name=? where id=?
    </update>

    <delete id="deleteById" resultType = "java.lang.Integer" parameter="true">
        delete from t_fruit where id=?
    </delete>

    <insert id="insert" resultType = "java.lang.Integer" parameter="true">
        insert into t_fruit values(null,?,?)
    </insert>

    <select id="getAll" resultType = "java.util.ArrayList" resultOf="com.yck.bean.Fruit" parameter="false">
        select * from t_fruit;
    </select>
</dao>

在service.xml中注册FruitService教给ServiceFactory管理；（写到这里我突然想写一个配置只配置包的名称，然后自己去扫描生成Bean类的factory）

<?xml version="1.0" encoding="UTF-8"?>
<services>
    <service id="userService" class="com.yck.service.UserService"/>
    <service id="fruitService" class="com.yck.service.FruitService"/>
</services>

然后测试我们的Service，这里我就不测试Dao有没有实现了，因为只要service实现了，dao一定是没问题的

package com.yck.test;

import com.yck.service.FruitService;
import com.yck.service.UserService;
import com.yck.yebatis.ServiceFactory;

public class Test
{
    public static void main(String[] args)
    {
        ServiceFactory sf = new ServiceFactory();
        UserService us = (UserService) sf.getBean("userService");
        FruitService fs = (FruitService) sf.getBean("fruitService");

        System.out.println(us.findAll());
        System.out.println(us.find(2));

        System.out.println(fs.findAll());
        System.out.println(fs.findById(2));
    }

}

测试结果如下，控制台输出

[User [id=, name=二蛋, age=], User [id=, name=王八蛋, age=], User [id=, name=李四, age=], User [id=, name=小明, age=], User [id=, name=小红, age=], User [id=, name=小张, age=], User [id=, name=李三, age=], User [id=, name=大牛, age=], User [id=, name=二牛, age=], User [id=, name=三牛, age=], User [id=, name=四牛, age=], User [id=, name=大牛, age=], User [id=, name=二牛, age=], User [id=, name=三牛, age=], User [id=, name=四牛, age=], User [id=, name=大牛, age=], User [id=, name=二牛, age=], User [id=, name=三牛, age=], User [id=, name=四牛, age=]]
User [id=, name=王八蛋, age=]
[Fruit [id=, name=apple, counts=], Fruit [id=, name=orange, counts=], Fruit [id=, name=葡萄, counts=], Fruit [id=, name=banana, counts=]]
Fruit [id=, name=orange, counts=]

也就是说我之前写的功能完全没有问题，往后我们写的Dao接口只要在src/mapper下面添加一个.xml配置文件，

写Service 只要在 src/service/service.xml里面添加一个<service>子节点就能自动注册到Factory进行管理。这个功能目前为止就这样了

以上内容都是原创，如果转载请标注并标明来源于

大王让我写代码：http://www.cnblogs.com/yeyeck/p/7468644.html