Mybatis Plus使用XML编写动态sql的超简易方法
作者:MadlifeZhou
使用xml编写动态sql
在Resources文件夹下创建一个Mapper文件夹
比如我们需要在User表中使用增删改查,创建UserMapper.xml,对应MybatisPlus中的UserMapper接口
之后我们在application.yml中配置mapper文件夹的路径
mybatis-plus: mapper-locations: classpath:mapper/*.xml
之后在UserMapper中创建函数
@Repository public interface UserMapper extends BaseMapper<User> { // 使函数参数对应xml中的参数wxNickName List<User> selectByName(@Param("wxNickName") String name); }
就可以在UserMapper.xml中写sql语句了
写法和Mybatis一样滴
<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "<http://mybatis.org/dtd/mybatis-3-mapper.dtd>"> <mapper namespace="com.zhou.usercenter.dao.user.UserMapper"> <select id="selectByName" resultType="com.zhou.usercenter.domain.entity.user.User"> select * from user <where> <if test="wxNickName != null and wxNickName != ''"> wx_nickname like CONCAT('%',#{wxNickName},'%'); </if> </where> </select> </mapper>
之后调用即可
@SpringBootTest class UserCenterApplicationTests { @Autowired UserMapper userMapper; @Test void contextLoads() { List<User> users = userMapper.selectByName("杰伦"); System.out.println(users); } }
即可看到正确的输出结果
动态SQL语句的原理
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 顺序: * <p> 1、加载 XML中的 SQL </p> * <p> 2、加载 SqlProvider 中的 SQL </p> * <p> 3、XmlSql 与 SqlProvider不能包含相同的 SQL </p> * <p>调整后的 SQL优先级:XmlSql > sqlProvider > CurdSql </p> */ @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的十多个类,主要如下图所示:
总体上来说,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(); } }
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。