Mybatis Plus实现动态SQL语句的原理，你知道吗？

￡神魔★判官ぃ 2022-09-07 14:42 230阅读 0赞

**点击关注公众号，利用****碎片时间学习****![3594d79c4a8f0a95c90a07804de4ef34.png][]**

Mybatis-Plus（简称MP）是一个 Mybatis 的增强工具,那么它是怎么增强的呢？其实就是它已经封装好了一些crud方法，开发就不需要再写xml了，直接调用这些方法就行，就类似于JPA。那么这篇文章就来阅读以下MP的具体实现，看看是怎样实现这些增强的。

![d82ee15119c4706e18a8860236908d52.png][]

### 入口类：MybatisSqlSessionFactoryBuilder ###

通过在入口类 `MybatisSqlSessionFactoryBuilder#build`方法中, 在应用启动时, 将mybatis plus(简称MP)自定义的动态配置xml文件注入到Mybatis中。

public class MybatisSqlSessionFactoryBuilder extends SqlSessionFactoryBuilder {
        public SqlSessionFactory build(Configuration configuration) {
                // ... 省略若干行 
                if (globalConfig.isEnableSqlRunner()) {
                    new SqlRunnerInjector().inject(configuration);
                }
                // ... 省略若干行 
                return sqlSessionFactory;
            }
    }

这里涉及到2个MP2个功能类

*  扩展继承自Mybatis的MybatisConfiguration类: MP动态脚本构建，注册，及其它逻辑判断。
 *  SqlRunnerInjector: MP默认插入一些动态方法的xml 脚本方法。

### MybatisConfiguration类 ###

这里我们重点剖析MybatisConfiguration类，在MybatisConfiguration中，MP初始化了其自身的MybatisMapperRegistry，而MybatisMapperRegistry是MP加载自定义的SQL方法的注册器。

MybatisConfiguration中很多方法是使用MybatisMapperRegistry进行重写实现

其中有3个重载方法addMapper实现了注册MP动态脚本的功能。

public class MybatisConfiguration extends Configuration {
     /**
      * Mapper 注册
      */
     protected final MybatisMapperRegistry mybatisMapperRegistry = new MybatisMapperRegistry(this);
     // ....
 
     /**
      * 初始化调用
      */
     public MybatisConfiguration() {
         super();
         this.mapUnderscoreToCamelCase = true;
         languageRegistry.setDefaultDriverClass(MybatisXMLLanguageDriver.class);
     }
 
     /**
      * MybatisPlus 加载 SQL 顺序：
      *  1、加载 XML中的 SQL 
      *  2、加载 SqlProvider 中的 SQL 
      *  3、XmlSql 与 SqlProvider不能包含相同的 SQL 
      * 调整后的 SQL优先级：XmlSql > sqlProvider > CurdSql 
      */
     @Override
     public void addMappedStatement(MappedStatement ms) {
         // ...
     }
     
     // ... 省略若干行 
     /**
      * 使用自己的 MybatisMapperRegistry
      */
     @Override
     public <T> void addMapper(Class<T> type) {
         mybatisMapperRegistry.addMapper(type);
     }
     // .... 省略若干行 
 }

在MybatisMapperRegistry中，MP将mybatis的MapperAnnotationBuilder替换为MP自己的MybatisMapperAnnotationBuilder

public class MybatisMapperRegistry extends MapperRegistry {
     @Override
     public <T> void addMapper(Class<T> type) {
         // ... 省略若干行 
         MybatisMapperAnnotationBuilder parser = new MybatisMapperAnnotationBuilder(config, type);
         parser.parse();
         // ... 省略若干行 
     }
 }

在MybatisMapperRegistry类的addMapper方法中，真正进入到MP的核心类MybatisMapperAnnotationBuilder，MybatisMapperAnnotationBuilder这个类是MP实现动态脚本的关键类。

### MybatisMapperAnnotationBuilder动态构造 ###

在MP的核心类MybatisMapperAnnotationBuilder的parser方法中，MP逐一遍历要加载的Mapper类，加载的方法包括下面几个

public class MybatisMapperAnnotationBuilder extends MapperAnnotationBuilder {
     @Override
     public void parse() {
         //... 省略若干行 
         for (Method method : type.getMethods()) {
             /** for循环代码, MP判断method方法是否是@Select @Insert等mybatis注解方法**/
             parseStatement(method);
             InterceptorIgnoreHelper.initSqlParserInfoCache(cache, mapperName, method);
             SqlParserHelper.initSqlParserInfoCache(mapperName, method);
         }
         /** 这2行代码, MP注入默认的方法列表**/
         if (GlobalConfigUtils.isSupperMapperChildren(configuration, type)) {
             GlobalConfigUtils.getSqlInjector(configuration).inspectInject(assistant, type);
         }
         //... 省略若干行 
     }
 
     @Override
     public void inspectInject(MapperBuilderAssistant builderAssistant, Class<?> mapperClass) {
         Class<?> modelClass = extractModelClass(mapperClass);
         //... 省略若干行 
         List<AbstractMethod> methodList = this.getMethodList(mapperClass);
         TableInfo tableInfo = TableInfoHelper.initTableInfo(builderAssistant, modelClass);
         // 循环注入自定义方法
         methodList.forEach(m -> m.inject(builderAssistant, mapperClass, modelClass, tableInfo));
         mapperRegistryCache.add(className);
     }
 }
 public class DefaultSqlInjector extends AbstractSqlInjector {
 
     @Override
     public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
         return Stream.of(
             new Insert(),
             //... 省略若干行 
             new SelectPage()
         ).collect(toList());
     }
 }

在MybatisMapperAnnotationBuilder中，MP真正将框架自定义的动态SQL语句注册到Mybatis引擎中。而AbstractMethod则履行了具体方法的SQL语句构造。

### 具体的AbstractMethod实例类，构造具体的方法SQL语句 ###

以 SelectById 这个类为例说明下

/**
  * 根据ID 查询一条数据
  */
 public class SelectById extends AbstractMethod {
     @Override
     public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
         /** 定义 mybatis xml method id, 对应 <id="xyz"> **/
         SqlMethod sqlMethod = SqlMethod.SELECT_BY_ID;
         /** 构造id对应的具体xml片段 **/
         SqlSource sqlSource = new RawSqlSource(configuration, String.format(sqlMethod.getSql(),
             sqlSelectColumns(tableInfo, false),
             tableInfo.getTableName(), tableInfo.getKeyColumn(), tableInfo.getKeyProperty(),
             tableInfo.getLogicDeleteSql(true, true)), Object.class);
         /** 将xml method方法添加到mybatis的MappedStatement中 **/
         return this.addSelectMappedStatementForTable(mapperClass, getMethod(sqlMethod), sqlSource, tableInfo);
     }
 }

至此，MP完成了在启动时加载自定义的方法xml配置的过程，后面的就是mybatis `${变量}` `#{变量}`的动态替换和预编译，已经进入mybatis自有功能。

### 总结一下 ###

MP总共改写和替换了mybatis的十多个类，主要如下图所示:

![18f6616656795a8ceb50a663f2d392b4.png][]

总体上来说，MP实现mybatis的增强，手段略显繁琐和不够直观，其实根据MybatisMapperAnnotationBuilder构造出自定义方法的xml文件，将其转换为mybatis的Resource资源，可以只继承重写一个Mybatis类：SqlSessionFactoryBean 比如如下：

public class YourSqlSessionFactoryBean extends SqlSessionFactoryBean implements ApplicationContextAware {
    
        private Resource[] mapperLocations;
    
        @Override
        public void setMapperLocations(Resource... mapperLocations) {
            super.setMapperLocations(mapperLocations);
            /** 暂存使用mybatis原生定义的mapper xml文件路径**/
            this.mapperLocations = mapperLocations;
        }
    
        /**
         * {@inheritDoc}
         */
        @Override
        public void afterPropertiesSet() throws Exception {
            ConfigurableListableBeanFactory beanFactory = getBeanFactory();
            /** 只需要通过将自定义的方法构造成xml resource和原生定义的Resource一起注入到mybatis中即可, 这样就可以实现MP的自定义动态SQL和原生SQL的共生关系**/
            this.setMapperLocations(InjectMapper.getMapperResource(this.dbType, beanFactory, this.mapperLocations));
            super.afterPropertiesSet();
        }
    }

在这边文章中，简单介绍了MP实现动态语句的实现过程，并且给出一个可能的更便捷方法。

*（感谢阅读，希望对你所有帮助)*

*来源：juejin.cn/post/6883081187103866894*

推荐好文

[**主流Java进阶技术（学习资料分享）**][Java]

[**分享一套基于SpringBoot和Vue的企业级中后台开源项目，代码很规范！**][SpringBoot_Vue]

[**能挣钱的，开源 SpringBoot 商城系统，功能超全，超漂亮**][SpringBoot]![8bf792a2b8e37ca27aed56848ed62153.png][]

[3594d79c4a8f0a95c90a07804de4ef34.png]: /images/20220829/2d8d60e6d1c84013ad275e311c972f61.png
[d82ee15119c4706e18a8860236908d52.png]: /images/20220829/daf0e1b1019a44818c55d05af2f9a651.png
[18f6616656795a8ceb50a663f2d392b4.png]: /images/20220829/311cb37dc3b64a6186967fa5631d0ddb.png
[Java]: http://mp.weixin.qq.com/s?__biz=MzU2MTI4MjI0MQ%3D%3D&chksm=fc79b1b2cb0e38a498933a034926df45b51524b044762583488d498ae2112c06cf35d412fb3b&idx=2&mid=2247506972&scene=21&sn=790959d9ee2b74e0d0c8a0a5395bc665#wechat_redirect
[SpringBoot_Vue]: http://mp.weixin.qq.com/s?__biz=MzU2MTI4MjI0MQ%3D%3D&chksm=fc799be6cb0e12f0b58479f8b031bc5ae79efdca1880ff9b19f0a107c3a88c37b9596dc73203&idx=2&mid=2247497288&scene=21&sn=65d7111a19e920df1e9416959594b2f4#wechat_redirect
[SpringBoot]: http://mp.weixin.qq.com/s?__biz=MzU2MTI4MjI0MQ%3D%3D&chksm=fc799a9ccb0e138a64a2671f31b8d4be1b3c061a288f7b0d1d4af02a60c45a72231446b91e00&idx=2&mid=2247497522&scene=21&sn=c5b4e596be4ac7d7f293e7e2f0ea0769#wechat_redirect
[8bf792a2b8e37ca27aed56848ed62153.png]: /images/20220829/14a13e75d12849f8b2726587854ec86b.png