Mybatis源码和设计模式(SqlSession模块与策略模式)

最新推荐文章于 2024-06-01 10:10:10 发布

smart哥

最新推荐文章于 2024-06-01 10:10:10 发布

阅读量824

点赞数 9

分类专栏： mybatis专题文章标签： mybatis 设计模式策略模式

本文链接：https://blog.csdn.net/smart_an/article/details/138517913

版权

mybatis专题专栏收录该内容

44 篇文章 1 订阅

订阅专栏

作者简介：大家好，我是smart哥，前中兴通讯、美团架构师，现某互联网公司CTO

联系qq：184480602，加我进群，大家一起学习，一起进步，一起对抗互联网寒冬

学习必须往深处挖，挖的越深，基础越扎实！

阶段1、深入多线程

 阶段2、深入多线程设计模式

 阶段3、深入juc源码解析

阶段4、深入jdk其余源码解析

阶段5、深入jvm源码解析

码哥源码部分

码哥讲源码-原理源码篇【2024年最新大厂关于线程池使用的场景题】

码哥讲源码【炸雷啦！炸雷啦！黄光头他终于跑路啦！】

码哥讲源码-【jvm课程前置知识及c/c++调试环境搭建】

码哥讲源码-原理源码篇【揭秘join方法的唤醒本质上决定于jvm的底层析构函数】

码哥源码-原理源码篇【Doug Lea为什么要将成员变量赋值给局部变量后再操作？】

码哥讲源码【你水不是你的错,但是你胡说八道就是你不对了！】

码哥讲源码【谁再说Spring不支持多线程事务，你给我抽他！】

终结B站没人能讲清楚红黑树的历史，不服等你来踢馆！

打脸系列【020-3小时讲解MESI协议和volatile之间的关系，那些将x86下的验证结果当作最终结果的水货们请闭嘴】

SqlSession模块与策略模式

一、策略模式

策略模式

二、功能

SqlSession是Mybatis的对外最核心的接口，通过该接口来实现数据库的读写命令，事物管理等。SqlSession本质是提供数据库操作的，包括数据库的四大操作。如下是ibatis的编程方式，我们直接通过Sqlsession来访问数据库。

    @Test
        public void query() {
            SqlSession sqlSession = SqlSessionFactoryUtil.getSqlSessionFactoryInstace().openSession();
            List<Object> list = sqlSession.selectList("com.intellif.mozping.dao.ItemMapper.findItemById", 1);
            for (Object obj : list) {
                System.out.println(((Item) obj).toString());
            }
        }

这里我们不分析面向接口编程后，通过接口调用如何最后又回到sqlsession的方法，这里有兴趣可以阅读参考文献[2]。本文主要分析SqlSession本身的设计方式，包括其子类，工厂类等。

三、源码

SqlSession继承图

3.1 实现类

SqlSession接口的主要实现类是DefaultSqlSession，SqlSessionFactory的主要实现是DefaultSqlSessionFactory。不过从类图我们看到有一个SqlSessionManager类，他继承了2个接口，一方面具备生产SqlSession的能力，另一方面又具备SqlSession的能力，我们比较一下2个实现类的区别。

SqlSession实现类	区别
DefaultSqlSession	每次访问数据库创建一个新的sqlsession，一个sqlsession代表一次数据库连接(默认，也是我们重点分析的)
SqlsessionManager	内部通过ThreadLocal来保证一个线程只创建一个sqlsession,但是也可以每次创建一个新的(和DefaultSqlSession一样，支持2种模式，了解)

3.2 DefaultSqlSession源码分析

SqlSession的默认实现是DefaultSqlSession。SqlSession提供的数据库操作在底层都是调用Executor来执行。SqlSession提供很多增删改查的方法，在DefaultSqlSession中看方法执行流程，所有的查询方法最后都会走到一个方法，底层调用Executor的query，更新和删除也类似，都会走到Executor的更新方法。简单抽取部分代码如下，更多源码可以阅读参考文章[6]：

      // -> selectMap(String statement, Object parameter, String mapKey, RowBounds rowBounds)
      @Override
      public <K, V> Map<K, V> selectMap(String statement, String mapKey) {
        return this.selectMap(statement, null, mapKey, RowBounds.DEFAULT);
      }
    
        // -> selectMap(String statement, Object parameter, String mapKey, RowBounds rowBounds)
      @Override
      public <K, V> Map<K, V> selectMap(String statement, Object parameter, String mapKey) {
        return this.selectMap(statement, parameter, mapKey, RowBounds.DEFAULT);
      }
    
        // -> selectList(String statement, Object parameter, RowBounds rowBounds)
      @Override
      public <K, V> Map<K, V> selectMap(String statement, Object parameter, String mapKey, RowBounds rowBounds) {
          //selectMap底层走的还是selectList(String statement, Object parameter, RowBounds rowBounds)方法
        final List<? extends V> list = selectList(statement, parameter, rowBounds);
        //因为是按照resultMap的形式返回，因此拿到结果集之后，需要对结果进行处理，通过mapResultHandler处理
        final DefaultMapResultHandler<K, V> mapResultHandler = new DefaultMapResultHandler<K, V>(mapKey,
            configuration.getObjectFactory(), configuration.getObjectWrapperFactory(), configuration.getReflectorFactory());
        final DefaultResultContext<V> context = new DefaultResultContext<V>();
        for (V o : list) {
          context.nextResultObject(o);
          mapResultHandler.handleResult(context);
        }
        //返回处理后的结果集
        return mapResultHandler.getMappedResults();
      }

逻辑图如下：

四、SqlSessionFactory

SqlSessionFactory是创建SqlSession的工厂。

4.1 SqlSessionFactory接口

    /**
     * Creates an {@link SqlSession} out of a connection or a DataSource
     * 创建SqlSession，通过数据库连接创建或者从数据源创建
     * @author Clinton Begin
     */
    public interface SqlSessionFactory {
    
      SqlSession openSession();
    
      SqlSession openSession(boolean autoCommit);
      SqlSession openSession(Connection connection);
      SqlSession openSession(TransactionIsolationLevel level);
    
      SqlSession openSession(ExecutorType execType);
      SqlSession openSession(ExecutorType execType, boolean autoCommit);
      SqlSession openSession(ExecutorType execType, TransactionIsolationLevel level);
      SqlSession openSession(ExecutorType execType, Connection connection);
    
      Configuration getConfiguration();
    }

4.2 实现类DefaultSqlSessionFactory

DefaultSqlSessionFactory是SqlSessionFactory的重要实现类，它是创建SqlSession的工厂。创建SqlSession有2种方式，从数据源获取或者从数据库连接获取，具体看下面代码和注释:

    /**
         * 方式一：从数据源获取SqlSession
         * */
        private SqlSession openSessionFromDataSource(ExecutorType execType, TransactionIsolationLevel level, boolean autoCommit) {
            Transaction tx = null;
            try {
                final Environment environment = configuration.getEnvironment();
                final TransactionFactory transactionFactory = getTransactionFactoryFromEnvironment(environment);
                //1.从数据源DataSource获取tx，这是和方式二最大的区别，其他的都差不多
                tx = transactionFactory.newTransaction(environment.getDataSource(), level, autoCommit);
                final Executor executor = configuration.newExecutor(tx, execType);
                return new DefaultSqlSession(configuration, executor, autoCommit);
            } catch (Exception e) {
                closeTransaction(tx); // may have fetched a connection so lets call close()
                throw ExceptionFactory.wrapException("Error opening session.  Cause: " + e, e);
            } finally {
                ErrorContext.instance().reset();
            }
        }
    
        /**
         * 方式二：从数据库连接获取SqlSession
         * */
        private SqlSession openSessionFromConnection(ExecutorType execType, Connection connection) {
            try {
                //1.确认是否自动提交
                boolean autoCommit;
                try {
                    autoCommit = connection.getAutoCommit();
                } catch (SQLException e) {
                    // Failover to true, as most poor drivers
                    // or databases won't support transactions
                    autoCommit = true;
                }
                final Environment environment = configuration.getEnvironment();
                final TransactionFactory transactionFactory = getTransactionFactoryFromEnvironment(environment);
                //2.从一个连接connection获取tx，这是和方式一最大的区别，其他的都差不多
                final Transaction tx = transactionFactory.newTransaction(connection);
                final Executor executor = configuration.newExecutor(tx, execType);
                return new DefaultSqlSession(configuration, executor, autoCommit);
            } catch (Exception e) {
                throw ExceptionFactory.wrapException("Error opening session.  Cause: " + e, e);
            } finally {
                ErrorContext.instance().reset();
            }
        }

我们看到2种创建方式的区别只在transactionFactory.newTransaction(connection);这一行，只是构造Transaction有区别，这里稍微跟进去其实区别在中2个构造方法，这就不深入了。

      //通过Connection构造ManagedTransaction
      public ManagedTransaction(Connection connection, boolean closeConnection) {
        this.connection = connection;
        this.closeConnection = closeConnection;
      }
      //通过DataSource构造ManagedTransaction
      public ManagedTransaction(DataSource ds, TransactionIsolationLevel level, boolean closeConnection) {
        this.dataSource = ds;
        this.level = level;
        this.closeConnection = closeConnection;
      }

DefaultSqlSessionFactory提供了很多创建的方法，底层最后都是走上面的2个方法，逻辑图如下：

SqlSession中的策略模式体现在：创建数据源有3种方式，POOL、UNPOOL和JNDI。在创建Sqlsession的时候，是根据环境创建的，在Environment里面会指定数据源的方式，对于Sqlsession的使用代码来说，不管底层是如何创建Sqlsession的，都没有关系。只需要改配置，数据源模块就能够生产出3种不同类型的SqlSession。在上面的2个方法中，configuration.newExecutor(tx, execType)说明传到DefaultSqlSession的executor都是根据具体的类型execType创建的，我们看看Configuration#newExecutor()方法

        //Configuration.java
        
        //是否开启2级缓存
        protected boolean cacheEnabled = true;
        
        /**
         * 配置对象创建Executor组件，BatchExecutor/ReuseExecutor/SimpleExecutor三种
         * 该方法在SqlSessionFactory的实现类DefaultSqlSessionFactory中会调用，得到的Executor组件会传到SqlSession中，
         * 因为SqlSession对数据库的访问需要使用Executor来实现
         * */
        public Executor newExecutor(Transaction transaction, ExecutorType executorType) {
            //1.如果executorType是null，那就使用defaultExecutorType = ExecutorType.SIMPLE
            executorType = executorType == null ? defaultExecutorType : executorType;
            executorType = executorType == null ? ExecutorType.SIMPLE : executorType;
            Executor executor;
            //2.BATCH
            if (ExecutorType.BATCH == executorType) {
                executor = new BatchExecutor(this, transaction);
            //3.REUSE
            } else if (ExecutorType.REUSE == executorType) {
                executor = new ReuseExecutor(this, transaction);
            //4.SIMPLE
            } else {
                executor = new SimpleExecutor(this, transaction);
            }
            //5.如果开启了二级缓存，那么就装饰一下
            if (cacheEnabled) {
                executor = new CachingExecutor(executor);
            }
            //6.处理插件
            executor = (Executor) interceptorChain.pluginAll(executor);
            return executor;
        }

关于CachingExecutor的装饰器模式，有兴趣阅读参考文章[4]的4.4小节
关于插件处理，有兴趣阅读参考文章[5]

五、小结

SqlSession是用户使用Mybatis编程的关键接口，我们通过SqlSessionFactory获取SqlSession。
SqlSessionFactory的初始化过程在配置初始化阶段完成，之后就可以通过SqlSessionFactory获取SqlSession了(可以阅读参考文章[1],其中6.2 Mybais的初始化流程图给出了DefaultSqlSessionFactory的实例化过程)
通过SqlSession，我们可以获取接口的代理对象，继而进行增删改查操作，经过一层代理之后，真正的查询入口还是SqlSession的方法(ibatis编程模型，关于代理，可以阅读参考文章[2])
SqlSession中很多数据库读取的重载方法，这些方法最后都走到一个方法里面，并且会使用Executor组件去访问数据库(Executor会引出其他三大对象来完成数据库的读写)，整个SqlSession查询逻辑有兴趣可以阅读参考文章[3]