【107期】面试官问：Mybatis Plus 实现动态 SQL 语句的原理，你知道吗？

最新推荐文章于 2024-06-11 15:13:13 发布

2401_83916326

最新推荐文章于 2024-06-11 15:13:13 发布

阅读量747

点赞数 8

分类专栏：程序员文章标签： mybatis sql 数据库

本文链接：https://blog.csdn.net/2401_83916326/article/details/137899696

版权

程序员专栏收录该内容

343 篇文章 0 订阅

订阅专栏

文章详细介绍了MybatisPlus如何在初始化时加载SQL，重点阐述了MybatisMapperAnnotationBuilder在构建自定义动态SQL中的关键作用，以及MapperRegistry的修改。作者还提出了一个更便捷的方法，通过继承SqlSessionFactoryBean来实现动态SQL与原生SQL的结合。

摘要由CSDN通过智能技术生成

/**

* 初始化调用

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 void addMapper(Class type) {

mybatisMapperRegistry.addMapper(type);

}

// … 省略若干行

}

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

public class MybatisMapperRegistry extends MapperRegistry {

@Override

public void addMapper(Class type) {

// … 省略若干行

MybatisMapperAnnotationBuilder parser = new MybatisMapperAnnotationBuilder(config, type);

parser.parse();

// … 省略若干行

}

在MybatisMapperRegistry类的addMapper方法中，真正进入到MP的核心类MybatisMapperAnnotationBuilder，MybatisMapperAnnotationBuilder这个类是MP实现动态脚本的关键类。更多面试题，见微信小程序Java精选面试题，内涵大量初级、中级、高级面试题，支持随时随地刷题。

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 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 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自有功能。更多面试题，见微信小程序Java精选面试题，内涵大量初级、中级、高级面试题，支持随时随地刷题。

总结一下

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

Spring Boot 短信验证码登录

总体上来说，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 {