Mybatis Plus 实现动态SQL语句的原理
Mybatis-Plus(简称MP)是一个 Mybatis 的增强工具,那么它是怎么增强的呢?其实就是它已经封装好了一些crud方法,开发就不需要再写xml了,直接调用这些方法就行,就类似于JPA。那么这篇文章就来阅读以下MP的具体实现,看看是怎样实现这些增强的。
入口类:MybatisSqlSessionFactoryBuilder
通过在入口类
MybatisSqlSessionFactoryBuilder#build
方法中, 在应用启动时, 将mybatis plus(简称MP)自定义的动态配置xml文件注入到Mybatis中。publicclassMybatisSqlSessionFactoryBuilderextendsSqlSessionFactoryBuilder
public SqlSessionFactory build(Configuration configuration)
// ... 省略若干行
if
new
// ... 省略若干行
return
{
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动态脚本的功能。
publicclassMybatisConfigurationextendsConfiguration
protectedfinal
// ....
publicMybatisConfiguration()
super
this
@Override
publicvoidaddMappedStatement(MappedStatement ms)
// ...
// ... 省略若干行
@Override
public
// .... 省略若干行
{
/**
* Mapper 注册
*/
protectedfinal
MybatisMapperRegistry mybatisMapperRegistry =
new MybatisMapperRegistry(
this);
// ....
/**
* 初始化调用
*/
publicMybatisConfiguration()
{
super
();
this
.mapUnderscoreToCamelCase =
true;
languageRegistry.setDefaultDriverClass(MybatisXMLLanguageDriver
.class);
}
/**
* MybatisPlus 加载 SQL 顺序:
* <p> 1、加载 XML中的 SQL </p>
* <p> 2、加载 SqlProvider 中的 SQL </p>
* <p> 3、XmlSql 与 SqlProvider不能包含相同的 SQL </p>
* <p>调整后的 SQL优先级:XmlSql > sqlProvider > CurdSql </p>
*/
@Override
publicvoidaddMappedStatement(MappedStatement ms)
{
// ...
}
// ... 省略若干行
/**
* 使用自己的 MybatisMapperRegistry
*/
@Override
public
<T>
voidaddMapper(Class<T> type){
mybatisMapperRegistry.addMapper(type);
}
// .... 省略若干行
}
在MybatisMapperRegistry中,MP将mybatis的MapperAnnotationBuilder替换为MP自己的MybatisMapperAnnotationBuilder
publicclassMybatisMapperRegistryextendsMapperRegistry
@Override
public
// ... 省略若干行
// ... 省略若干行
{
@Override
public
<T>
voidaddMapper(Class<T> type){
// ... 省略若干行
MybatisMapperAnnotationBuilder parser =
new MybatisMapperAnnotationBuilder(config, type);
parser.parse();
// ... 省略若干行
}
}
在MybatisMapperRegistry类的addMapper方法中,真正进入到MP的核心类MybatisMapperAnnotationBuilder,MybatisMapperAnnotationBuilder这个类是MP实现动态脚本的关键类。
MybatisMapperAnnotationBuilder动态构造
在MP的核心类MybatisMapperAnnotationBuilder的parser方法中,MP逐一遍历要加载的Mapper类,加载的方法包括下面几个
publicclassMybatisMapperAnnotationBuilderextendsMapperAnnotationBuilder
@Override
publicvoidparse()
//... 省略若干行
for
/** for循环代码, MP判断method方法是否是@Select@Insert等mybatis注解方法**/
/** 这2行代码, MP注入默认的方法列表**/
if
//... 省略若干行
@Override
publicvoidinspectInject(MapperBuilderAssistant builderAssistant, Class<?> mapperClass)
//... 省略若干行
// 循环注入自定义方法
publicclassDefaultSqlInjectorextendsAbstractSqlInjector
@Override
public List<AbstractMethod> getMethodList(Class<?> mapperClass)
return
new
//... 省略若干行
new
{
@Override
publicvoidparse()
{
//... 省略若干行
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
publicvoidinspectInject(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);
}
}
publicclassDefaultSqlInjectorextendsAbstractSqlInjector
{
@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 查询一条数据
*/
publicclassSelectByIdextendsAbstractMethod
{
@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中 **/
returnthis
.addSelectMappedStatementForTable(mapperClass, getMethod(sqlMethod), sqlSource, tableInfo);
}
}
至此,MP完成了在启动时加载自定义的方法xml配置的过程,后面的就是mybatis
${变量}
#{变量}
的动态替换和预编译,已经进入mybatis自有功能。总结一下
MP总共改写和替换了mybatis的十多个类,主要如下图所示:
总体上来说,MP实现mybatis的增强,手段略显繁琐和不够直观,其实根据MybatisMapperAnnotationBuilder构造出自定义方法的xml文件,将其转换为mybatis的Resource资源,可以只继承重写一个Mybatis类:SqlSessionFactoryBean
比如如下:
publicclassYourSqlSessionFactoryBeanextendsSqlSessionFactoryBeanimplementsApplicationContextAware
private
@Override
publicvoidsetMapperLocations(Resource... mapperLocations)
super
/** 暂存使用mybatis原生定义的mapper xml文件路径**/
this
@Override
publicvoidafterPropertiesSet()throws Exception
/** 只需要通过将自定义的方法构造成xml resource和原生定义的Resource一起注入到mybatis中即可, 这样就可以实现MP的自定义动态SQL和原生SQL的共生关系**/
this
super
{
private
Resource[] mapperLocations;
@Override
publicvoidsetMapperLocations(Resource... mapperLocations)
{
super
.setMapperLocations(mapperLocations);
/** 暂存使用mybatis原生定义的mapper xml文件路径**/
this
.mapperLocations = mapperLocations;
}
/**
* {
@inheritDoc}
*/
@Override
publicvoidafterPropertiesSet()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学习完整路线,回复 路线
缺Java入门视频,回复: 视频
要Java面试经验,回复 面试
缺Java项目,回复: 项目
进Java粉丝群: 加群
PS:如果觉得我的分享不错,欢迎大家随手点赞、在看。
请备注:666,不然不通过~
最近好文
关键词
语句
就是
SQL语句
功能
核心
最新评论
推荐文章
作者最新文章
你可能感兴趣的文章
Copyright Disclaimer: The copyright of contents (including texts, images, videos and audios) posted above belong to the User who shared or the third-party website which the User shared from. If you found your copyright have been infringed, please send a DMCA takedown notice to [email protected]. For more detail of the source, please click on the button "Read Original Post" below. For other communications, please send to [email protected].
版权声明:以上内容为用户推荐收藏至CareerEngine平台,其内容(含文字、图片、视频、音频等)及知识版权均属用户或用户转发自的第三方网站,如涉嫌侵权,请通知[email protected]进行信息删除。如需查看信息来源,请点击“查看原文”。如需洽谈其它事宜,请联系[email protected]。
版权声明:以上内容为用户推荐收藏至CareerEngine平台,其内容(含文字、图片、视频、音频等)及知识版权均属用户或用户转发自的第三方网站,如涉嫌侵权,请通知[email protected]进行信息删除。如需查看信息来源,请点击“查看原文”。如需洽谈其它事宜,请联系[email protected]。