Mybatis学习之自定义持久层框架（七） 自定义持久层框架优化

前言

接上文，这里只是出于强迫症，凭借着半年前的笔记来把之前没写完的文章写完，这里是最后一篇了。

前面自定义的持久层框架存在的问题

Dao层若使用实现类，会存在代码重复，整个操作的过程模版重复（加载配置文件、创建sqlSessionFactory、生产sqlSession）

解决思路

使用代理模式生成Dao层接口的代理实现类（去掉原Dao层的代理实现类）

代码实现

在IPersistence_test项目中添加Dao层及接口类：

package com.hardy.dao;

import com.hardy.pojo.User;

import java.util.List;

public interface IUserDao {

    // 查询所有用户
    public List<User> findAll() throws Exception;

    // 根据条件进行用户查询
    public User findByCondition(User user) throws Exception;

}

在SqlSession接口类中添加getMapper接口：

package com.hardy.sqlSession;

import java.util.List;

public interface SqlSession {

    /* 查询所有：
    根据statementId，找到Mapper.xml文件中对应的sql语句
    Object...Parameter 表示支持传递多个参数值进行查询
     */
    public <E> List<E> selectList(String statementId, Object... Parameter) throws Exception;

    // 根据条件查询单个
    public <T> T selectOne(String statementId, Object... Parameter) throws Exception;

    //为Dao接口生成代理实现类
    public <T> T getMapper(Class<?> mapperClass);

}

在SqlSession的实现类中重写getMapper接口：

package com.hardy.sqlSession;

import com.hardy.pojo.Configuration;
import com.hardy.pojo.MappedStatement;

import java.lang.reflect.*;
import java.lang.reflect.Proxy;
import java.sql.SQLException;
import java.util.List;

public class DefaultSqlSession implements SqlSession {

    private Configuration configuration;

    public DefaultSqlSession(Configuration configuration) {
        this.configuration = configuration;
    }

    // 处理器对象
    private Executor simpleExecutor = new SimpleExecutor();

    @Override
    public <E> List<E> selectList(String statementId, Object... params) throws Exception {
        // 对SimpleExecutor里的query方法的调用 7
        SimpleExecutor simpleExecutor = new SimpleExecutor();
        MappedStatement mappedStatement = configuration.getMappedStatementMap().get(statementId);
        List<Object> list = simpleExecutor.query(configuration, mappedStatement, params);

        return (List<E>) list;
    }

    @Override
    public <T> T selectOne(String statementId, Object... params) throws Exception {
        List<Object> objects = selectList(statementId, params);
        if (objects.size() == 1) {
            return (T) objects.get(0);
        } else {
            throw new RuntimeException("查无此数据或查询结果过多");
        }

    }

    @Override
    public <T> T getMapper(Class<?> mapperClass) {
        // 使用JDK动态代理来为Dao接口生成代理对象，并返回
        Object proxyInstance = Proxy.newProxyInstance(DefaultSqlSession.class.getClassLoader(), new Class[]{mapperClass}, new InvocationHandler() {
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                // 底层都还是去执行JDBC代码
                // 根据不同情况，来调用selctList或者selectOne
                // 准备参数 1：statmentid :sql语句的唯一标识：namespace.id = 接口全限定名.方法名
                String methodName = method.getName();
                String className = method.getDeclaringClass().getName();

                String statementId = className + "." + methodName;

                // 准备参数2：params:args
                // 获取被调用方法的返回值类型
                Type genericReturnType = method.getGenericReturnType();
                // 判断是否进行了 泛型类型参数化
                if (genericReturnType instanceof ParameterizedType) {
                    List<Object> objects = selectList(statementId, args);
                    return objects;
                }

                return selectOne(statementId, args);
            }
        });

        return (T) proxyInstance;
    }

    public void close() throws SQLException {
        simpleExecutor.close();
    }

}

修改测试类：

package com.hardy.test;

import com.hardy.dao.IUserDao;
import com.hardy.io.Resources;
import com.hardy.pojo.User;
import com.hardy.sqlSession.SqlSession;
import com.hardy.sqlSession.SqlSessionFactory;
import com.hardy.sqlSession.SqlSessionFactoryBuilder;
import org.junit.Test;

import java.io.InputStream;
import java.util.List;

public class IPersistenceTest {

    @Test
    public void test() throws Exception {
        InputStream resourceAsStream = Resources.getResourceAsStream("sqlMapConfig.xml");
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().builder(resourceAsStream);
        SqlSession sqlSession = sqlSessionFactory.openSession();

        System.out.println("自定义持久层框架，GO！");
        // 调用
//        User user = new User();
//        user.setId(1);
//        user.setUsername("hardy");
//        User user2 = sqlSession.selectOne("User.selectOne", user);
//        System.out.println("查询单条记录："+ user2);
//
//        System.out.println("-----------------------------------------");
//
//        List<User> users = sqlSession.selectList("User.selectList");
//        System.out.println("查询多条记录：");
//        for (User user1 : users) {
//            System.out.println(user1);
//        }

        IUserDao userDao = sqlSession.getMapper(IUserDao.class);

        List<User> allUsers = userDao.findAll();
        for (User user1 : allUsers) {
            System.out.println(user1);
        }

    }
}

运行结果如下所示：

总结

这半年多来，经过工作中的磨练及空闲时间的自学，虽然我依旧是个菜鸟，但是开发能力还是比之前强了不少，回想半年多以前自己写这个系列文章的时候，对当中很多概念都不太清楚，现在倒是觉得很简单了。

之前是挺想一直坚持写下去的，不过由于那段时间项目太赶，工作强度比较大，导致学习积极性下降了很多。另一方面，也是因为自己的心态没有调整好，对于未来没有明确的目标。

我还算比较幸运，后面有机会换了一份比较好的工作，工作强度不那么大，这几个月自学的时间相当多，技术方面进步了不少。

可惜一直没有下定决心来好好写技术博客，新年新气象，正好最近自学也快告一段落了，是时候通过输出倒逼输入了，今年的目标之一就是坚持每周写一点博客，每月发布几篇。

也许我在很长的一段时间内都写不出什么好文章，但是 管它呢，自己喜欢就好，有个平台可以免费写点东西，顺便还能复习下自己学到的知识，这本身就是一种回报。

话虽如此，还是要精益求精，尽可能写多点有技术含量的东西，所以，要好好学习咯。