mybatis-plus自定义sql注入器
mybatis-plus:自定义sql注入器
虽然mybats-plus的BaseMapper提供了非常好用，非常多的方法，但是如果我们需要扩充BaseMapper中的方法又该如何实现呢
以findAll方法为例
新建一个spring-boot项目 引入mybatis-plus依赖
在com.mp.Mapper包中新建MyBaseMapper接口
/**
*创建自己的BaseMapper，指定泛型；添加一个findAll方法；
*/
public interface MyBaseMapper<T> extends BaseMapper<T> {
List<T> findAll();
}新建一个AutoMapper接口继承MyBaseMapper接口
/**
* Mapper接口
*/
@Mapper
public interface AutoMapper extends MyBaseMapper<Car> {
}该接口作为mapper接口与Controller交互
新建sqlInjector包编写自定义的sql注入器：MySqlInjector
/**
* 继承DefaultSqlInjector类，可以获取到BaseMapper原有的方法；
* 重写getMethodList方法；添加自己的方法；
*/
public class MySqlInjector extends DefaultSqlInjector {
@Override
public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
List<AbstractMethod> methodList = super.getMethodList(mapperClass);
//把自己的方法实例添加到集合；
// 需要是参照MP的方法写FindAll类(sql语句封装为MappedStatement)；
methodList.add(new FindAll());
return methodList;
}
}在sqlInjector包下创建FindAll类(参考SelectList类来写)
**
* 自定义findAll查询方法的封装类；
* 参照MP的方法写FindAll类(sql语句封装为MappedStatement)；
*/
public class FindAll extends AbstractMethod {
@Override
public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
String sqlMethod = "findAll";//方法名
String sql = "select * from " + tableInfo.getTableName();//sql语句
SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
return this.addSelectMappedStatementForTable(mapperClass, sqlMethod, sqlSource, tableInfo);
}
}在spring-boot启动类中注册到spring容器
/**
* 自定义sql注入器
* @return
*/
@Bean
public MySqlInjector mySqlInjector(){
return new MySqlInjector();
}在测试类中测试
//测试自己定义的findAll方法
@Test
void testFindAll(){
List<Car> all = autoMapper.findAll();
all.forEach(System.out::println);
}本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2020-07-12 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读