这里写目录标题
- 一 修改pom文件
- 最外层的pom.xml增加如下配置
- common模块的pom文件加入如下配置
- 二 framework模块的config包增加两个类
- MybatisPlusPageConfigurer
- SqlFilterArgumentResolver
- 三 全局搜索com.ruoyi.framework.config.MyBatisConfig类,将此类注释掉
- 四 yml注释掉mybatis,粘贴plus的配置
- 五 修改service和mapper层
- 六 BaseController重载一个getDataTable
- 七 写个plus的分页测试一下
- 八 测试一个plus的修改功能
一 修改pom文件
最外层的pom.xml增加如下配置
若依使用的是3.8.6的
自带的pagehelper用的1.4.6里面的jsqlparser用的4.5.0
mybatis-plus用的3.4.2里面的jsqlparser用的4.0.0
<pagehelper.boot.version>1.4.6</pagehelper.boot.version>
pagehelper需要修改如下,
<dependency>
<groupId>com.github.pagehelper</groupId>
<artifactId>pagehelper-spring-boot-starter</artifactId>
<version>${pagehelper.boot.version}</version>
<exclusions>
<exclusion>
<artifactId>mybatis-spring</artifactId>
<groupId>org.mybatis</groupId>
</exclusion>
<exclusion>
<artifactId>mybatis</artifactId>
<groupId>org.mybatis</groupId>
</exclusion>
<exclusion>
<artifactId>jsqlparser</artifactId>
<groupId>com.github.jsqlparser</groupId>
</exclusion>
</exclusions>
</dependency>
plus包需要新增
hutool工具包和lombok工具包如果不需要的话可以自行删除并修改后面的类里面的代码
<!--mybatis-plus-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.4.2</version>
</dependency>
<!--hutool开发工具包-->
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-all</artifactId>
<version>5.2.5</version>
</dependency>
<!--lombok开发工具包-->
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.22</version>
</dependency>
common模块的pom文件加入如下配置
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
</dependency>
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-all</artifactId>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
</dependency>
jsqlparser包,不剔除是这样的,有个冲突,建议保留plus里面的版本
解决冲突的好办法看这个文章
https://blog.csdn.net/lh155136/article/details/121907377
二 framework模块的config包增加两个类
MybatisPlusPageConfigurer
package cn.chinaunicom.sh.wxy.config;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import java.util.List;
/**
* @author admin
*/
@Configuration(proxyBeanMethods = false)
public class MybatisPlusPageConfigurer implements WebMvcConfigurer {
/**
* SQL 过滤器避免SQL 注入
* @param argumentResolvers
*/
@Override
public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
argumentResolvers.add(new SqlFilterArgumentResolver());
}
/**
* 分页插件
*/
@Bean
public MybatisPlusInterceptor mybatisPlusInterceptor() {
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
return interceptor;
}
}
SqlFilterArgumentResolver
package cn.chinaunicom.sh.wxy.config;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;
import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import java.util.function.Predicate;
import java.util.stream.Collectors;
/**
* 解决Mybatis Plus Order By SQL注入问题
* @author admin
*/
@Slf4j
public class SqlFilterArgumentResolver implements HandlerMethodArgumentResolver {
private final static String[] KEYWORDS = { "master", "truncate", "insert", "select", "delete", "update", "declare",
"alter", "drop", "sleep" };
/**
* 判断Controller是否包含page 参数
* @param parameter 参数
* @return 是否过滤
*/
@Override
public boolean supportsParameter(MethodParameter parameter) {
return parameter.getParameterType().equals(Page.class);
}
/**
* @param parameter 入参集合
* @param mavContainer model 和 view
* @param webRequest web相关
* @param binderFactory 入参解析
* @return 检查后新的page对象
* <p>
* page 只支持查询 GET .如需解析POST获取请求报文体处理
*/
@Override
public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
NativeWebRequest webRequest, WebDataBinderFactory binderFactory) {
HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
String[] ascs = request.getParameterValues("ascs");
String[] descs = request.getParameterValues("descs");
String current = request.getParameter("current");
String size = request.getParameter("size");
Page<?> page = new Page<>();
if (StrUtil.isNotBlank(current)) {
page.setCurrent(Long.parseLong(current));
}
if (StrUtil.isNotBlank(size)) {
page.setSize(Long.parseLong(size));
}
List<OrderItem> orderItemList = new ArrayList<>();
Optional.ofNullable(ascs).ifPresent(s -> orderItemList.addAll(
Arrays.stream(s).filter(sqlInjectPredicate()).map(OrderItem::asc).collect(Collectors.toList())));
Optional.ofNullable(descs).ifPresent(s -> orderItemList.addAll(
Arrays.stream(s).filter(sqlInjectPredicate()).map(OrderItem::desc).collect(Collectors.toList())));
page.addOrder(orderItemList);
return page;
}
/**
* 判断用户输入里面有没有关键字
* @return Predicate
*/
private Predicate<String> sqlInjectPredicate() {
return sql -> {
for (String keyword : KEYWORDS) {
if (StrUtil.containsIgnoreCase(sql, keyword)) {
return false;
}
}
return true;
};
}
}
三 全局搜索com.ruoyi.framework.config.MyBatisConfig类,将此类注释掉
四 yml注释掉mybatis,粘贴plus的配置
如果配置不同,请自行修改
mybatis-plus:
mapper-locations: classpath*:mapper/**/*Mapper.xml
check-config-location: true
type-aliases-package: com.ruoyi.**.domain
configuration:
log-impl: org.apache.ibatis.logging.slf4j.Slf4jImpl
五 修改service和mapper层
mapper接口继承BaseMapper<你的Entity>
public interface PageInfoMapper extends BaseMapper<PageInfoEntity> {
service接口继承IService<你的Entity>接口
public interface IPageInfoService extends IService<PageInfoEntity> {
service类继承ServiceImpl<你的Mapper,你的Entity>类 实现 你的Service接口
public class PageInfoServiceImpl extends ServiceImpl<PageInfoMapper, PageInfoEntity> implements IPageInfoService {
你的Entity类上面需要增加表的名字
@TableName("business_info")
public class BusinessInfoEntity
六 BaseController重载一个getDataTable
protected TableDataInfo getDataTable(List<?> list, Long total)
{
TableDataInfo rspData = new TableDataInfo();
rspData.setCode(HttpStatus.SUCCESS);
rspData.setMsg("查询成功");
rspData.setRows(list);
rspData.setTotal(total);
return rspData;
}
七 写个plus的分页测试一下
@GetMapping("/getPageInfoPage")
public TableDataInfo getPageInfoPage(Page page, InfoReq infoReq) {
Page<InfoEntity> page1 = businessInfoService.page(page, new QueryWrapper<InfoEntity>());
return getDataTable(page1.getRecords(), page1.getTotal());
}
我用的前端写的请求,返回的格式和原来的一样,已兼容
日志如下,到此就分页兼容了
八 测试一个plus的修改功能
InfoEntity entity = new InfoEntity();
entity.setCustName("哈哈哈");
Long id = 1L;
businessInfoService.update(entity, new QueryWrapper<InfoEntity>().lambda().eq(InfoEntity::getId, id));
看日志也没问题