深入理解Mybatis工作原理

基本原理
源码解析

基本原理

本文中Mybatis基本原理切入，同时摘录Mybatis的源码，理解Mybatis框架的工作原理。首先看一段JDBC代码：

Class.forName("com.mysql.jdbc.Driver");
String url = "jdbc:mysql://localhost:3306/localdb";
Connection con = DriverManager.getConnection(url, "root","password");			
String sql = "SELECT name FROM t_test_employee WHERE id = ?";
PreparedStatement ps = con.prepareStatement(sql);
ps.setString(1, "123");
ResultSet rs = ps.executeQuery();
while(rs.next()){
	System.out.println("name=" + rs.getString(1));
}
con.close();

大家应该比较熟悉这段代码，典型的jdbc流程: 建立连接->传递sql->传递参数->sql执行->处理结果->关闭连接。ORM框架的本质是抽取共性、封装逻辑。现在，如果我们来设计一个ORM框架该如何设计？
仔细考虑JDBC流程，可以发现流程中需要用户配置的输入和输出是变化的(如下，括号中部分)，而其他部分则是固定不变的逻辑。

建立连接(数据源配置信息)->传递sql(sql语句)->传递参数(sql参数)->sql执行->处理结果(映射关系)->关闭连接

数据源配置信息:配置文件，在启动时从配置文件中读取并建立数据源对象
Sql语句：配置文件(代码)，在启动时抽取解析，以<Key,SQL>的形式存储
Sql参数：java bean对象，在执行前从对象中抽取
映射关系：配置文件(代码)，根据映射关系读取结果集并创建java bean返回。

不难想象，Mybatis会在启动时”读取配置数据(builder)”，”建立数据源(datasource)”; 然后程序开始执行，传递sql语句的”Key”和java bean; 最后”执行SQL，处理结果(executor)”。当然，Mybatis也要负责”连接管理(session)”和”事务管理(transaction)”等内容。

源码解析

Mybatis的配置分为两种：基于XML配置和基于注解配置。两者的原理大同小异，在此仅以“基于注解配置”为例解析Mybatis。

启动配置解析

“基于注解配置”一般使用Mybatis-Spring，其工作原理详见深入理解Mybatis-Spring工作原理，可以看到启动后所有Mapper的实现类被设置为了MapperFactoryBean，。MapperFactoryBean是一个工厂类，它通过SqlSession-Configuration-MapperRegistry得到Mapper的代理类MapperProxy，以实例化Mapper属性。

SQL(参数)解析与执行

Mapper执行时调用MapperProxy的invoke方法，MapperProxy创建MapperMethod对象用于执行具体SQL，通过annotation获取得到参数化的sql语句，通过传递的参数获取SQL参数。

 public MapperMethod(Class<?> mapperInterface, Method method, Configuration config) {
//解析annotation中的SQL语句
   this.command = new SqlCommand(config, mapperInterface, method);
//处理sql，建立①param数组
   this.method = new MethodSignature(config, method);
 }
 
  public Object execute(SqlSession sqlSession, Object[] args) {
   Object result;
   if (SqlCommandType.INSERT == command.getType()) { //判断sql类型
  //使用①param数组建立param-Object对应关系
     Object param = method.convertArgsToSqlCommandParam(args);
  //执行sql并处理结果集的映射
     result = rowCountResult(sqlSession.insert(command.getName(), param));
   }
...

//XXXExecutor.java
public int doUpdate(MappedStatement ms, Object parameter) throws SQLException {
   Statement stmt = null;
   try {
     Configuration configuration = ms.getConfiguration();
     StatementHandler handler = configuration.newStatementHandler(this, ms, parameter, RowBounds.DEFAULT, null, null);
  //使用TypeHandler处理参数
     stmt = prepareStatement(handler, ms.getStatementLog());
     return handler.update(stmt);
   } finally {
     closeStatement(stmt);
   }
 }

结果集映射

略。

以上即是Mybatis的大致原理，更详细的当然就是RTFC咯。Mybatis的实现很明了，功能相对简单，还有更多的东西可以去扩展，譬如：多数据源、数据库切分、缓存、连接管理等等。

(完)

本文作者 ： cyningsun
本文地址 ： https://www.cyningsun.com/10-15-2014/reading-mybatis-source-code.html
版权声明 ：本博客所有文章除特别声明外，均采用 CC BY-NC-ND 3.0 CN 许可协议。转载请注明出处！