说一下 Mybatis 的工作原理?

 作者简介:大家好,我是码炫码哥,前中兴通讯、美团架构师,现任某互联网公司CTO,兼职码炫课堂主讲源码系列专题


代表作:《jdk源码&多线程&高并发》,《深入tomcat源码解析》,《深入netty源码解析》,《深入dubbo源码解析》,《深入springboot源码解析》,《深入spring源码解析》,《深入redis源码解析》等


联系qq:184480602,加我进群,大家一起学习,一起进步,一起对抗互联网寒冬。码炫课堂的个人空间-码炫码哥个人主页-面试,源码等

回答

Mybatis 是一款 ORM 框架,实现了 SQL 到 Java 对象的映射转换,简化了持久层代码的开发工作。其核心流程分两步,一是创建会话工厂,二是会话运行

  1. 加载配置文件,MyBatis 加载全局配置文件(如 mybatis-config.xml)和映射文件(如XXXMapper.xml),配置数据库连接、映射器、插件等信息,生成一个全局配置对象(Configuration)。
  2. 创建 SqlSessionFactory,MyBatis 基于加载的配置文件创建 SqlSessionFactory(用于生成SqlSession)。
  3. 获取 SqlSession,通过 SqlSessionFactory 获取 SqlSession。SqlSession 是 MyBatis 执行 SQL 操作的核心。
  4. 执行 SQL,Mybatis 内部定义了 Executor 接口来操作数据库,它根据 SqlSession 传递的参数动态地生成执行 SQL,并支持查询缓存的维护。
  5. 处理结果集,MyBatis 将 SQL 执行的结果映射为 Java 对象,返回给调用者。
  6. 关闭会话,操作完成后,需要关闭 SqlSession 以释放资源。

扩展

一、创建会话工厂

1、加载配置文件

  • Mybatis 加载并解析配置文件,创建Configuration 对象。

2、构建会话工厂

  • 基于Configuration 对象,创建会话工厂(DefaultSqlSessionFactory)。Configuration 保存 Mybatis 的全局配置信息。
  public SqlSessionFactory build(InputStream inputStream, String environment, Properties properties) {
    try {
      // 加载 xml & 注解配置  
      XMLConfigBuilder parser = new XMLConfigBuilder(inputStream, environment, properties);
      // 解析配置并创建默认会话工厂
      return build(parser.parse());
    } catch (Exception e) {
      throw ExceptionFactory.wrapException("Error building SqlSession.", e);
    } finally {
      ErrorContext.instance().reset();
      try {
        inputStream.close();
      } catch (IOException e) {
        // Intentionally ignore. Prefer previous error.
      }
    }
  }

  public Configuration parse() {
    if (parsed) {
      throw new BuilderException("Each XMLConfigBuilder can only be used once.");
    }
    parsed = true;
    parseConfiguration(parser.evalNode("/configuration"));
    return configuration;
  }

  public SqlSessionFactory build(Configuration config) {
    return new DefaultSqlSessionFactory(config);
  }

二、会话执行

Mybatis 会话执行是其工作机制的核心之一,涉及到 SqlSessionExecutorStatementHandlerParameterHandlerResultSetHandler等组件。

1、获取 SqlSession

通过 SqlSessionFactory 创建 SqlSession。SqlSession 是 Mybatis 的核心接口,用于执行 SQL,获取映射器,管理事务。

  public SqlSession openSession() {
    // 创建 SqlSession  
    return openSessionFromDataSource(configuration.getDefaultExecutorType(), null, false);
  }

2、获取 MapperdStatement

SqlSession 调用对应的 Mapper 方法时,实际上是通过 Configuration 类获取 MappedStatement。MappedStatement 封装了 SQL 语句的所有信息,如 SQL 语句、输入输出参数类型等。

  public <E> List<E> selectList(String statement, Object parameter, RowBounds rowBounds) {
    try {
      // 获取 Statement 映射器  
      MappedStatement ms = configuration.getMappedStatement(statement);
      return executor.query(ms, wrapCollection(parameter), rowBounds, Executor.NO_RESULT_HANDLER);
    } catch (Exception e) {
      throw ExceptionFactory.wrapException("Error querying database.  Cause: " + e, e);
    } finally {
      ErrorContext.instance().reset();
    }
  }

3、执行 SQL

  • SQL 预编译处理(StatementHandler

    执行器(Executor)通过 Configuration 创建数据库会话处理器(StatementHandler)。

  public <E> List<E> doQuery(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
    Statement stmt = null;
    try {
      // 全局配置  
      Configuration configuration = ms.getConfiguration();
      // 创建会话处理器 
      StatementHandler handler = configuration.newStatementHandler(wrapper, ms, parameter, rowBounds, resultHandler, boundSql);
      // 编译预处理
      stmt = prepareStatement(handler, ms.getStatementLog());
      // 会话执行以及结果集处理
      return handler.query(stmt, resultHandler);
    } finally {
      closeStatement(stmt);
    }
  }

基于 StatementHandler 进行 SQL 编译预处理。StatementHandler 负责处理 JDBC Statement 的创建。

  private Statement prepareStatement(StatementHandler handler, Log statementLog) throws SQLException {
    Statement stmt;
    Connection connection = getConnection(statementLog);
    // 编译预处理  
    stmt = handler.prepare(connection, transaction.getTimeout());
    // 设置参数  
    handler.parameterize(stmt);
    return stmt;
  }
  • SQL 参数处理(ParameterHandler

StatementHandler 编译预处理后通过 ParameterHandler 进行参数设置。

  public void parameterize(Statement statement) throws SQLException {
    // 设置参数  
    parameterHandler.setParameters((PreparedStatement) statement);
  }
  • 打开 DB 连接执行 SQL
  public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
    PreparedStatement ps = (PreparedStatement) statement;
    // 执行 SQL   
    ps.execute(); 
    return resultSetHandler.handleResultSets(ps);
  }

4、结果集处理(ResultSetHandler

结果集处理器通过 handleResultSets(Statement stmt) 方法包装数据库返回的执行结果。

  public List<Object> handleResultSets(Statement stmt) throws SQLException {
    ErrorContext.instance().activity("handling results").object(mappedStatement.getId());

    final List<Object> multipleResults = new ArrayList<>();

    int resultSetCount = 0;
    ResultSetWrapper rsw = getFirstResultSet(stmt);

    List<ResultMap> resultMaps = mappedStatement.getResultMaps();
    int resultMapCount = resultMaps.size();
    validateResultMapsCount(rsw, resultMapCount);
    while (rsw != null && resultMapCount > resultSetCount) {
      ResultMap resultMap = resultMaps.get(resultSetCount);
      handleResultSet(rsw, resultMap, multipleResults, null);
      rsw = getNextResultSet(stmt);
      cleanUpAfterHandlingResultSet();
      resultSetCount++;
    }

    String[] resultSets = mappedStatement.getResultSets();
    if (resultSets != null) {
      while (rsw != null && resultSetCount < resultSets.length) {
        ResultMapping parentMapping = nextResultMaps.get(resultSets[resultSetCount]);
        if (parentMapping != null) {
          String nestedResultMapId = parentMapping.getNestedResultMapId();
          ResultMap resultMap = configuration.getResultMap(nestedResultMapId);
          handleResultSet(rsw, resultMap, null, parentMapping);
        }
        rsw = getNextResultSet(stmt);
        cleanUpAfterHandlingResultSet();
        resultSetCount++;
      }
    }

    return collapseSingleResultList(multipleResults);
  }

5、关闭 SqlSession

操作完成后,必须关闭 SqlSession 以释放资源。

// 关闭会话
session.close();

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值