面试官问:Mybatis Plus 是如何实现动态 SQL 语句的?原理你懂吗?

程序员的成长之路

共 6351字,需浏览 13分钟

 ·

2021-12-01 01:19

程序员的成长之路
互联网/程序员/技术/资料共享 
关注


阅读本文大概需要 4 分钟。

来源:juejin.cn/post/6883081187103866894


Mybatis-Plus(简称MP)是一个 Mybatis 的增强工具,那么它是怎么增强的呢?其实就是它已经封装好了一些crud方法,开发就不需要再写xml了,直接调用这些方法就行,就类似于JPA。

那么这篇文章就来阅读以下MP的具体实现,看看是怎样实现这些增强的。


# 入口类: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  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实现动态脚本的关键类。

# MybatisMapperAnnotationBuilder动态构造


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

public class MybatisMapperAnnotationBuilder extends MapperAnnotationBuilder {    @Overrde    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, 对应  **/ 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(truetrue)), Object.class);        /** 将xml method方法添加到mybatis的MappedStatement中 **/        return this.addSelectMappedStatementForTable(mapperClass, getMethod(sqlMethod), sqlSource, tableInfo);    }}

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

# 总结一下


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



总体上来说,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实现动态语句的实现过程,并且给出一个可能的更便捷方法。
推荐阅读:
妙用Java 8中的 Function接口 消灭if...else(非常新颖的写法)
大文件上传服务器、支持超大文件HTTP断点续传实践总结

最近面试BAT,整理一份面试资料《Java面试BATJ通关手册》,覆盖了Java核心技术、JVM、Java并发、SSM、微服务、数据库、数据结构等等。

获取方式:点个「在看」,点击上方小卡片,进入公众号后回复「面试题」领取,更多内容陆续奉上。

朕已阅 

浏览 30
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报