四、Mybatis 执行流程全解析

知识点:
执行流程解析
Mybatis 插件开发

一、执行流程解析

1. 配置文件解析 configuration

理解解析流程之前先回顾一下mybatis中配置文件的结构:

mybatis-config.xml

<configuration>
  <properties/>
  <settting/>
  <typeHandlers/>
  <..../>
  <mappers/>
</configuration>

mybatis-mapper.xml

<mapper >
  <cache/>
  <resultMap/>
  <select/> 
  <update/> 
  <delete/> 
  <insert/> 
</mapper>

配置文件的解析流程即是将上述XML描述元素转换成对应的Java对象过程,其最终转换对象及其关系如下图:


配置元素解析构建器

>org.apache.ibatis.builder.xml.XMLConfigBuilder
 >org.apache.ibatis.builder.xml.XMLMapperBuilder
  >org.apache.ibatis.builder.xml.XMLStatementBuilder
   >org.apache.ibatis.builder.SqlSourceBuilder
    >org.apache.ibatis.scripting.xmltags.XMLScriptBuilder
 >org.apache.ibatis.builder.annotation.MapperAnnotationBuilder

sql statement 构建流程源码

>org.apache.ibatis.session.SqlSessionFactoryBuilder#build()
//1.Config.xml 文件解析
>org.apache.ibatis.builder.xml.XMLConfigBuilder#parse
 >org.apache.ibatis.builder.xml.XMLConfigBuilder#parseConfiguration
  >org.apache.ibatis.builder.xml.XMLConfigBuilder#mapperElement
//2.Mapper.xml 文件解析
>org.apache.ibatis.builder.xml.XMLMapperBuilder#parse 
 >org.apache.ibatis.builder.xml.XMLMapperBuilder#configurationElement
  >org.apache.ibatis.builder.xml.XMLMapperBuilder#buildStatementFromContext(java.util.List<org.apache.ibatis.parsing.XNode>)
//3.Statemen sql块解析
>org.apache.ibatis.builder.xml.XMLStatementBuilder#parseStatementNode
 >org.apache.ibatis.builder.MapperBuilderAssistant#addMappedStatement
// 4.动态SQL脚本解析
>org.apache.ibatis.scripting.xmltags.XMLLanguageDriver#createSqlSource()
>org.apache.ibatis.scripting.xmltags.XMLScriptBuilder#parseScriptNode()
>org.apache.ibatis.scripting.xmltags.XMLScriptBuilder#parseDynamicTags()   

2. 会话创建SqlSession

首先我们还是先来了解一下会话对像的组成结构如下图:


会话构建源码解析

>org.apache.ibatis.session.defaults.DefaultSqlSessionFactory#openSession(boolean)
 >org.apache.ibatis.session.defaults.DefaultSqlSessionFactory#openSessionFromDataSource
  >org.apache.ibatis.transaction.TransactionFactory#newTransaction
  >org.apache.ibatis.session.Configuration#newExecutor
   >org.apache.ibatis.executor.SimpleExecutor
   >org.apache.ibatis.executor.CachingExecutor
     //执行器插件包装
   >org.apache.ibatis.plugin.InterceptorChain#pluginAll(executor)
  >org.apache.ibatis.session.defaults.DefaultSqlSession#DefaultSqlSession() 

3. 方法执行 StatementHandler

StatementHandler 源码解析

>org.apache.ibatis.session.defaults.DefaultSqlSession#selectList()
 >org.apache.ibatis.executor.CachingExecutor#query()
  >org.apache.ibatis.executor.BaseExecutor#query()
   >org.apache.ibatis.executor.BaseExecutor#queryFromDatabase
>org.apache.ibatis.session.Configuration#newStatementHandler
org.apache.ibatis.executor.statement.BaseStatementHandler#BaseStatementHandler
org.apache.ibatis.session.Configuration#newParameterHandler
org.apache.ibatis.plugin.InterceptorChain#pluginAll(parameterHandler)
org.apache.ibatis.session.Configuration#newResultSetHandler
org.apache.ibatis.plugin.InterceptorChain#pluginAll(resultSetHandler)
>org.apache.ibatis.plugin.InterceptorChain#pluginAll(statementHandler)
>org.apache.ibatis.executor.BaseExecutor#getConnection
>org.apache.ibatis.executor.statement.PreparedStatementHandler#instantiateStatement
>org.apache.ibatis.executor.statement.PreparedStatementHandler#parameterize
>org.apache.ibatis.scripting.defaults.DefaultParameterHandler#setParameters
org.apache.ibatis.type.BaseTypeHandler#setParameter
org.apache.ibatis.type.UnknownTypeHandler#setNonNullParameter
org.apache.ibatis.type.IntegerTypeHandler#setNonNullParameter

二、myBatis插件开发

插件的四大扩展点

  1. Executor
  2. StatementHandler
  3. ParameterHandler
  4. ResultSetHandler

分页插件实现:
用户在接口中声明Page 对像实现后,由插件实现自动分页。使用示例如下:

public class Page implements java.io.Serializable {
   private int szie; // 每页大
   private int number; // 当前页码
}

page参数声明

@Select("select * from user")
List<User> selectByPage(String name, Page page);

客户端调用

mapper.selectByPage("小明", new Page(3, 2))

select * from user  limit 10,20

实现目标分解

  1. 修改 修改SQL 并添加 limit 语句
  2. 判断方法参数中是否有Page对象
  3. 取出Page对象 生成limit 语句
  4. 上述操作必须在PreparedStatement 对像生成前完成

完整代码如下:

package com.niuh.mybatis.dao;

import com.sun.deploy.util.ReflectionUtil;
import org.apache.ibatis.executor.parameter.ParameterHandler;
import org.apache.ibatis.executor.statement.BaseStatementHandler;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.scripting.defaults.DefaultParameterHandler;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Map;
import java.util.Optional;
import java.util.Properties;
/**
 * @author: hejianhui
 * @create: 2019-07-13 10:48
 * @see PagePlugin
 * @since JDK1.8
 */
public class PagePlugin implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        return null;
    }

    @Override
    public Object plugin(Object target) {
        if (target instanceof StatementHandler) {
            return Proxy.newProxyInstance(PagePlugin.class.getClassLoader(),
                    new Class[]{StatementHandler.class},
                    new PageHandler((StatementHandler) target));
        }
        return target;
    }

    @Override
    public void setProperties(Properties properties) {

    }


    private class PageHandler implements InvocationHandler {
        StatementHandler handler;

        public PageHandler(StatementHandler handler) {
            this.handler = handler;
        }

        @Override
        public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
            if (method.getName().equalsIgnoreCase("prepare")) {
                setParametersProxy();
            }
            return method.invoke(handler, args);
        }

        private void setParametersProxy() {
            ;
            if (handler.getBoundSql().getParameterObject() instanceof Map) {
                ((Map) handler.getBoundSql().getParameterObject()).values().stream()
                        .filter(a -> a instanceof Page)
                        .findFirst()
                        .ifPresent(
                                page -> {
                                    appendPageSql((Page) page);
                                }
                        );
            }
        }

        private void appendPageSql(Page page) {
            try {
                BoundSql sql = handler.getBoundSql();
                sql.getSql();
                String limit = String.format(" limit %s,%s", page.getBegin(), page.getSzie());
                setFileValue("sql", sql, sql.getSql() + limit);
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }
    }


    private Object getFileValue(String fileName, Object target) throws NoSuchFieldException, IllegalAccessException {
        Field sqlField = target.getClass().getDeclaredField(fileName);
        sqlField.setAccessible(true);
        return sqlField.get(target);
    }

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

推荐阅读更多精彩内容