四、持久层框架设计实现及MyBatis源码分析-自定义持久层框架优化（四）

前面我们完成了自定义持久层框架的编写，接下来我们分析自定义持久层框架存在的问题，首先我们先来查看我们编写的测试类中的代码

package study.lagou.com.test;

import org.junit.Test;
import study.lagou.com.io.Resources;
import study.lagou.com.persistence.test.pojo.User;
import study.lagou.com.sqlSession.SqlSession;
import study.lagou.com.sqlSession.SqlSessionFactory;
import study.lagou.com.sqlSession.SqlSessionFactoryBuilder;

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

/**
 * @Description: 功能描述
 * @Author houjh
 * @Email: happyxiaohou@gmail.com
 * @Date: 2021-1-28 21:19
 */
public class PersistenceTest {

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


        User user = new User();
        user.setId(1);
        user.setUsername("zhangsan");
        User u = sqlSession.selectOne("user.selectOne", user);
        System.out.println(u);
    }

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


        List<User> userList = sqlSession.selectList("user.selectList", null);
        for (User user : userList) {
            System.out.println(user);
        }
    }
}

我们分析测试类中的代码如果是在实际的项目当中，此段测试代码会存在什么地方呢？
是不是会将此部分代码编写到我们的持久层中，之前我们使用端的项目当中没有实现持久层，现在我们先来实现一下持久层具体的代码

首先定义一个接口IUserDao

package study.lagou.com.persistence.test.dao;

import study.lagou.com.persistence.test.pojo.User;

import java.util.List;

public interface IUserDao {

    /**
     * 查询所有用户信息
     * @return
     */
    List<User> findAll();

    /**
     * 根据条件查询单个用户信息
     * @param user
     * @return
     */
    User findByCondition(User user);
}

接着实现具体的接口实现类UserDaoImpl

package study.lagou.com.persistence.test.dao.impl;

import org.dom4j.DocumentException;
import study.lagou.com.io.Resources;
import study.lagou.com.persistence.test.dao.IUserDao;
import study.lagou.com.persistence.test.pojo.User;
import study.lagou.com.sqlSession.SqlSession;
import study.lagou.com.sqlSession.SqlSessionFactory;
import study.lagou.com.sqlSession.SqlSessionFactoryBuilder;

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

/**
 * @Description: 功能描述
 * @Author houjh
 * @Email: happyxiaohou@gmail.com
 * @Date: 2021-2-25 22:32
 */
public class UserDaoImpl implements IUserDao {

    @Override
    public List<User> findAll() throws Exception {
        InputStream resourceAsStream = Resources.getResourceAsStream("sqlMapConfig.xml");
        SqlSessionFactory build = new SqlSessionFactoryBuilder().build(resourceAsStream);
        SqlSession sqlSession = build.openSession();
        List<User> userList = sqlSession.selectList("user.selectList", null);
        return userList;
    }

    @Override
    public User findByCondition(User user) throws Exception {
        InputStream resourceAsStream = Resources.getResourceAsStream("sqlMapConfig.xml");
        SqlSessionFactory build = new SqlSessionFactoryBuilder().build(resourceAsStream);
        SqlSession sqlSession = build.openSession();
        return sqlSession.selectOne("user.selectOne", null);
    }

}

我们分析UserDaoImpl这个实现类中的代码，该类中的代码主要存在2个问题

1、Dao层使用自定义持久层框架本身存在代码重，整个操作的过程模板重复（主要是加载配置文件、创建SqlSessionFactory、生产SqlSession这几步重复）

重复代码

2、代码中的statementId还是存在硬编码问题

硬编码

以上2个问题，我们有什么方式可以进行解决呢？

解决方案还是有的：我们可以基于JDK动态代理，使用代理模式来动态生成Dao层接口的代理实现类，不再进行持久层实现类的具体编写

具体实现步骤如下：

1、在自定义持久层框架本身项目SqlSession当中添加getMapper(Class<?> mapperInterfaceClass)方法，通过JDK动态代理通过接口类直接生成对应的代理实现类

SqlSession接口的调整

package study.lagou.com.sqlSession;

import java.util.List;

public interface SqlSession {

    /**
     * 查询列表信息
     * @param statementId
     * @param params
     * @param <E>
     * @return
     */
    <E> List<E> selectList(String statementId, Object... params) throws Exception;

    /**
     * 查询单个数据信息
     * @param statementId
     * @param params
     * @param <T>
     * @return
     */
    <T> T selectOne(String statementId,Object... params) throws Exception;

    /**
     * 删除信息
     * @param statementId
     * @param params
     */
    void delete(String statementId,Object... params);

    /**
     * 更新数据信息
     * @param statementId
     * @param params
     */
    void update(String statementId,Object... params);

    /**
     * 通过接口
     * @param mapperInterfaceClass
     * @param <T>
     * @return
     */
    <T> T getMapper(Class<?> mapperInterfaceClass);
}

DefaultSqlSession的调整

package study.lagou.com.sqlSession;

import study.lagou.com.pojo.Configuration;
import study.lagou.com.pojo.MappedStatement;

import java.lang.reflect.*;
import java.util.List;

/**
 * @Description: 功能描述
 * @Author houjh
 * @Email: happyxiaohou@gmail.com
 * @Date: 2021-2-22 21:24
 */
public class DefaultSqlSession implements SqlSession {

    private Configuration configuration;

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

    @Override
    public <E> List<E> selectList(String statementId, Object... params) throws Exception {
        SimpleExecutor simpleExecutor = new SimpleExecutor();
        MappedStatement mappedStatement = configuration.getMappedStatementMap().get(statementId);
        List<Object> query = simpleExecutor.query(configuration, mappedStatement, params);
        return (List<E>) query;
    }

    @Override
    public <T> T selectOne(String statementId, Object... params) throws Exception {
        //selectOne方法和selectList方法操作数据库的方式一致，所以我们此处直接调用selectList方法即可
        List<Object> objects = selectList(statementId, params);
        if(objects.size() == 1){
            return (T) objects.get(0);
        } else {
            throw new RuntimeException("查询到的结果集为空或返回的结果集较多!");
        }
    }

    @Override
    public void delete(String statementId, Object... params) {

    }

    @Override
    public void update(String statementId, Object... params) {

    }

    @Override
    public <T> T getMapper(Class<?> mapperInterfaceClass) {
        Object proxyInstance = Proxy.newProxyInstance(DefaultSqlSession.class.getClassLoader(), new Class[]{mapperInterfaceClass}, new InvocationHandler() {
            //proxy 为当前代理对象
            //method 当前被调用方法的引用
            //args 传递的参数信息
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                //底层还是去执行JDBC代码，只是我们需要根据不同情况，来选择调用selectList或是selectOne方法，如果是查询
                //我们可以根据返回值类型是否进行了 泛型类型参数化（通俗讲就是判断返回值是否是一个泛型）进行判断，如果是就
                //说明当前返回对象是一个集合，则调用selectList方法，否则就调用selectOne

                //1、准备参数statementId，在xml文件中statementId是由namespace.id来组成，但是我们此处只能获取到对应的
                //接口信息，并不能获取到xml文件，所以此处约定让namespace.id = 接口全限定名.方法名的方式来进行处理，这
                //也是Mybatis的Mapper.xml文件中namespace为什么要和接口路径保持一致的原因

                //获取方法名称
                String methodName = method.getName();
                //通过方法获取到接口类的全限定名
                String className = method.getDeclaringClass().getName();
                //拼接生成statementId
                String statementId = className+"."+methodName;


                //2、根据被调用方法的返回值类型，判断应该调用selectOne还是selectList方法
                Type genericReturnType = method.getGenericReturnType();
                if(genericReturnType instanceof ParameterizedType){
                    return selectList(statementId,args);
                } else {
                    return selectOne(statementId,args);
                }
            }
        });
        return (T) proxyInstance;
    }
}

需要注意：此次的改造我们对Mapper.xml也有对应的调整，主要调整的地方就是namespace要和接口的全限定类型保持一致，具体SQL语句的id需要和接口中的方法名称保持一致（需要做此调整的主要原因是我们在使用JDK动态代理的时候只能拿到接口的全限定路径和方法名称，无法获取到Mapper.xml配置文件中的信息，但是需要执行底层的JDBC代码又必须通过statementId参数获取到对应的MappedStatement对象才能执行，为了解决此问题，我们约定通过让Mapper.xml文件中的 namespace.id = 接口全限定名.方法名的方式来进行处理）

使用端UserMapper.xml的改造

<mapper namespace="study.lagou.com.persistence.test.dao.IUserDao">

    <!-- 查询所有的用户信息 -->
    <select id="findAll" resultType="study.lagou.com.persistence.test.pojo.User">
        select * from user
    </select>

    <!-- 根据条件查询单个用户信息 -->
    <select id="findByCondition" resultType="study.lagou.com.persistence.test.pojo.User"
            paramterType="study.lagou.com.persistence.test.pojo.User">
        select * from user where id = #{id} and username = #{username}
    </select>


</mapper>

接着编写测试类对改造优化后的框架进行测试

package study.lagou.com.test;

import org.junit.Test;
import study.lagou.com.io.Resources;
import study.lagou.com.persistence.test.dao.IUserDao;
import study.lagou.com.persistence.test.pojo.User;
import study.lagou.com.sqlSession.SqlSession;
import study.lagou.com.sqlSession.SqlSessionFactory;
import study.lagou.com.sqlSession.SqlSessionFactoryBuilder;

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

/**
 * @Description: 功能描述
 * @Author houjh
 * @Email: happyxiaohou@gmail.com
 * @Date: 2021-1-28 21:19
 */
public class PersistenceMapperTest {

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

        IUserDao userDao = sqlSession.getMapper(IUserDao.class);
        User user = new User();
        user.setId(1);
        user.setUsername("zhangsan");
        User resultUser = userDao.findByCondition(user);
        System.out.println(resultUser);
    }

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


        IUserDao userDao = sqlSession.getMapper(IUserDao.class);
        List<User> userList = userDao.findAll();
        for (User user : userList) {
            System.out.println(user);
        }
    }
}

至此，一个简单的MyBatis自定义持久层框架基本实现，下一节我们对增、删、改方法进行详细的实现

具体代码对应下载地址:https://gitee.com/happymima/mybatis.git

最后编辑于：2021.07.22 19:45:33

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 218,451评论 6赞 506
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 93,172评论 3赞 394
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 164,782评论 0赞 354
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 58,709评论 1赞 294
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 67,733评论 6赞 392
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 51,578评论 1赞 305
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 40,320评论 3赞 418
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 39,241评论 0赞 276
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 45,686评论 1赞 314
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 37,878评论 3赞 336
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 39,992评论 1赞 348
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 35,715评论 5赞 346
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 41,336评论 3赞 330
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 31,912评论 0赞 22
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 33,040评论 1赞 270
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 48,173评论 3赞 370
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 44,947评论 2赞 355

四、持久层框架设计实现及MyBatis源码分析-自定义持久层框架优化（四）

1、在自定义持久层框架本身项目SqlSession当中添加getMapper(Class<?> mapperInterfaceClass)方法，通过JDK动态代理通过接口类直接生成对应的代理实现类

推荐阅读更多精彩内容