Mybatis源码解析

1.编写一个小demo

导入mybatis源码,demo目录结构如下:


源码的pom文件中需要加入以下依赖:

      <dependency>
          <groupId>mysql</groupId>
          <artifactId>mysql-connector-java</artifactId>
          <version>5.1.38</version>
      </dependency>

UserMapper.java

public interface UserMapper {
   User selectById(String id);
}

User.java

public class User {
  private String id;
  private String name;

  public User(String id, String name) {
    this.id = id;
    this.name = name;
  }

  public String getId() {
    return id;
  }

  public void setId(String id) {
    this.id = id;
  }

  public String getName() {
    return name;
  }

  public void setName(String name) {
    this.name = name;
  }

  @Override
  public String toString() {
    return "User{" +
      "id='" + id + '\'' +
      ", name='" + name + '\'' +
      '}';
  }
}

UserMapper.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">

<!-- 名称空间namespace必须唯一 -->
<mapper namespace="demo.mapper.UserMapper"> //namespace指定映射的接口文件

    <!--resultMap标签:映射实体与表,非主键属性标签 type属性:实体全路径名,可在mybatis.xml文件中统一配置 id属性:为实体与表的映射取一个唯一的编号-->
    <resultMap type="user" id="userMap">
        <!--id标签:映射主键属性 result标签:映射非主键属性 property属性:实体属性名 column属性:表的字段名-->
        <id property="id" column="id"/>
        <result property="name" column="name"/>
    </resultMap>
    
    <select id="selectById" resultMap="userMap"  parameterType="demo.pojo.User">
        SELECT * FROM USER WHERE id=#{id};
    </select>

</mapper>

db.properties

mysql.driver=com.mysql.jdbc.Driver
mysql.url=jdbc:mysql://localhost:3306/mybatisdb
mysql.username=root
mysql.password=12345678

mybatis-config.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration
        PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>

    <properties resource="db.properties"></properties> //定义属性文件

    <!-- 设置实体类的类型别名 -->
    <typeAliases>
        <typeAlias type="demo.pojo.User" alias="user"/>
    </typeAliases>

    <!-- 设置一个默认的连接环境信息,支持多数据源 -->
    <environments default="mysql_env">
        <!-- 连接环境信息,取一个唯一的编号 -->
        <environment id="mysql_env">
            <!-- mybatis使用的jdbc事务管理方式 -->
            <transactionManager type="jdbc">
            </transactionManager>
            <!-- mybatis使用连接池方式来获取链接 -->
            <dataSource type="pooled">
                <!-- 配置与数据库交互的四个属性 -->
                <property name="driver" value="${mysql.driver}"/>
                <property name="url" value="${mysql.url}"/>
                <property name="username" value="${mysql.username}"/>
                <property name="password" value="${mysql.password}"/>
            </dataSource>
        </environment>
    </environments>
    <mappers>
            <mapper resource="sqlMapper/UserMapper.xml"/>  //定义映射XML文件
    </mappers>
</configuration>

MybatisDemo.java

public class MybatisDemo {

  SqlSessionFactory sqlSessionFactory;

  @Before
  public void init() throws IOException {
    String resource = "mybatis-config.xml";
    InputStream inputStream = Resources.getResourceAsStream(resource);
    sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
    inputStream.close();
  }


  @Test
  public void  testAutoMapping(){
    SqlSession sqlSession=sqlSessionFactory.openSession();
    UserMapper mapper = sqlSession.getMapper(UserMapper.class);
    User user = mapper.selectById("1");
    System.out.println(user);
  }
}

运行结果为:

User{id='1', name='xiaoming'}

2.源码的分析

主要分三个阶段:
(1)初始化阶段
主要是对mybatis-config.xml配置文件以及UserMapper的解析,代码跟踪如下:

 sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
 --> return build(parser.parse());
     --> parseConfiguration(parser.evalNode("/configuration"));
     
  private void parseConfiguration(XNode root) {
    try {
      //issue #117 read properties first
      propertiesElement(root.evalNode("properties"));
      Properties settings = settingsAsProperties(root.evalNode("settings"));
      loadCustomVfs(settings);
      loadCustomLogImpl(settings);
      typeAliasesElement(root.evalNode("typeAliases"));
      pluginElement(root.evalNode("plugins"));
      objectFactoryElement(root.evalNode("objectFactory"));
      objectWrapperFactoryElement(root.evalNode("objectWrapperFactory"));
      reflectorFactoryElement(root.evalNode("reflectorFactory"));
      settingsElement(settings);
      // read it after objectFactory and objectWrapperFactory issue #631
      environmentsElement(root.evalNode("environments"));
      databaseIdProviderElement(root.evalNode("databaseIdProvider"));
      typeHandlerElement(root.evalNode("typeHandlers"));
      mapperElement(root.evalNode("mappers"));
    } catch (Exception e) {
      throw new BuilderException("Error parsing SQL Mapper Configuration. Cause: " + e, e);
    }
  }

跟踪一下 environmentsElement(root.evalNode("environments"));

environmentsElement(root.evalNode("environments"));
-->TransactionFactory txFactory = transactionManagerElement(child.evalNode("transactionManager")); //构建 transaction
-->DataSourceFactory dsFactory = dataSourceElement(child.evalNode("dataSource")); //构建dataSource
-->configuration.setEnvironment(environmentBuilder.build()); //设置环境变量

跟踪一下 mapperElement(root.evalNode("mappers"));

mapperElement(root.evalNode("mappers"));
--> XMLMapperBuilder mapperParser = new XMLMapperBuilder(inputStream, configuration, resource, configuration.getSqlFragments());
--> mapperParser.parse(); //解析mappers标签下的文件
  --> configurationElement(parser.evalNode("/mapper"));
    --> buildStatementFromContext(context.evalNodes("select|insert|update|delete"));
       --> buildStatementFromContext(list, null);
          --> statementParser.parseStatementNode();
            -->builderAssistant.addMappedStatement(id, sqlSource, statementType, sqlCommandType,fetchSize, timeout, parameterMap, parameterTypeClass, resultMap, resultTypeClass,resultSetTypeEnum, flushCache, useCache, resultOrdered,keyGenerator, keyProperty, keyColumn, databaseId, langDriver, resultSets);
               --> configuration.addMappedStatement(statement);

可以看出,所有的相关的额配置信息都存储到了Configuration这个类的实例里

(2)代理阶段

主要是这两句代码

    SqlSession sqlSession=sqlSessionFactory.openSession(); //定义sqlSession
    UserMapper mapper = sqlSession.getMapper(UserMapper.class); //关键语句

SqlSession中主要是定义操作数据库的一些方法,如CRUD,重点看一下getMapper方法:

UserMapper mapper = sqlSession.getMapper(UserMapper.class); 
--> return configuration.getMapper(type, this);
   --> return mapperRegistry.getMapper(type, sqlSession);
     -->  return mapperProxyFactory.newInstance(sqlSession); //从Mapper代理工厂创建mapper
        --> final MapperProxy<T> mapperProxy = new MapperProxy<>(sqlSession, mapperInterface, methodCache);//定义了一个Mapper的动态代理,调用Mapper接口时会调用里面的的invoke方法
          --> return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy);//返回代理创建的对象

这里很关键,完成了mapper接口与实际操作sql语句的映射关系。

(3)数据读取阶段
执行User user = mapper.selectById("1")时,会调用MapperProxy类下的invoke方法,过程如下:

 return mapperMethod.execute(sqlSession, args);
 -->switch (command.getType())//根据command字段类型执行相应的操作
   --> result = sqlSession.selectOne(command.getName(), param);//执行sqlSession的selectOne方法
     -->List<T> list = this.selectList(statement, parameter);
        --> return executor.query(ms, wrapCollection(parameter), rowBounds, Executor.NO_RESULT_HANDLER);
           --> return query(ms, parameter, rowBounds, resultHandler, key, boundSql);
              --> list = queryFromDatabase(ms, parameter, rowBounds, resultHandler, key, boundSql);//从数据库查询
                 --> list = doQuery(ms, parameter, rowBounds, resultHandler, boundSql);
                    -->stmt = prepareStatement(handler, ms.getStatementLog());
                       --> Connection connection = getConnection(statementLog); //这里执行的jdbc的操作
                       --> stmt = handler.prepare(connection, transaction.getTimeout());//这里执行的jdbc的操作
                          -->return handler.queryCursor(stmt); //返回游标
                             -->String sql = boundSql.getSql();//获取sql
                             -->statement.execute(sql);//执行sql
                 

3.手写一个简单的mybatis框架

可参考以下链接
https://github.com/xiaohongstudent/mybatis_myself

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 225,124评论 6 523
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 96,453评论 3 404
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 172,386评论 0 368
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 61,136评论 1 301
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 70,142评论 6 400
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 53,593评论 1 315
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 41,958评论 3 429
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 40,944评论 0 279
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 47,477评论 1 324
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 39,512评论 3 346
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 41,639评论 1 355
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 37,227评论 5 351
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 42,971评论 3 340
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 33,397评论 0 25
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 34,550评论 1 277
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 50,203评论 3 381
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 46,713评论 2 366

推荐阅读更多精彩内容