package com.ruoyi.framework.config;
|
|
import com.baomidou.mybatisplus.annotation.DbType;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.ParameterUtils;
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
|
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
|
import com.ruoyi.common.handler.CustomTenantLineHandler;
|
import org.apache.ibatis.executor.Executor;
|
import org.apache.ibatis.mapping.BoundSql;
|
import org.apache.ibatis.mapping.MappedStatement;
|
import org.apache.ibatis.session.ResultHandler;
|
import org.apache.ibatis.session.RowBounds;
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
import org.springframework.transaction.annotation.EnableTransactionManagement;
|
|
import java.sql.SQLException;
|
|
/**
|
* Mybatis Plus 配置
|
*
|
* @author ruoyi
|
*/
|
@EnableTransactionManagement(proxyTargetClass = true)
|
@Configuration
|
public class MybatisPlusConfig
|
{
|
@Bean
|
public MybatisPlusInterceptor mybatisPlusInterceptor()
|
{
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
// 租户插件
|
TenantLineInnerInterceptor tenantLineInnerInterceptor = new TenantLineInnerInterceptor(new CustomTenantLineHandler());
|
interceptor.addInnerInterceptor(tenantLineInnerInterceptor);
|
// 分页插件
|
interceptor.addInnerInterceptor(paginationInnerInterceptor());
|
// 乐观锁插件
|
interceptor.addInnerInterceptor(optimisticLockerInnerInterceptor());
|
// 阻断插件
|
interceptor.addInnerInterceptor(blockAttackInnerInterceptor());
|
return interceptor;
|
}
|
|
/**
|
* 分页插件，自动识别数据库类型 https://baomidou.com/guide/interceptor-pagination.html
|
*/
|
// public PaginationInnerInterceptor paginationInnerInterceptor()
|
// {
|
// PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
|
// // 设置数据库类型为mysql
|
// paginationInnerInterceptor.setDbType(DbType.MYSQL);
|
// // 设置最大单页限制数量，默认 500 条，-1 不受限制
|
// paginationInnerInterceptor.setMaxLimit(-1L);
|
// return paginationInnerInterceptor;
|
// }
|
|
public PaginationInnerInterceptor paginationInnerInterceptor() {
|
PaginationInnerInterceptor interceptor = new PaginationInnerInterceptor(DbType.MYSQL) {
|
@Override
|
public void beforeQuery(Executor executor, MappedStatement ms, Object parameter,
|
RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
|
IPage<?> page = ParameterUtils.findPage(parameter).orElse(null);
|
if (page != null && page.getSize() <= 0) {
|
// 当size<=0时，不进行分页
|
return;
|
}
|
super.beforeQuery(executor, ms, parameter, rowBounds, resultHandler, boundSql);
|
}
|
};
|
interceptor.setMaxLimit(1000L); // 建议设置合理的最大值
|
return interceptor;
|
}
|
|
/**
|
* 乐观锁插件 https://baomidou.com/guide/interceptor-optimistic-locker.html
|
*/
|
public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor()
|
{
|
return new OptimisticLockerInnerInterceptor();
|
}
|
|
/**
|
* 如果是对全表的删除或更新操作，就会终止该操作 https://baomidou.com/guide/interceptor-block-attack.html
|
*/
|
public BlockAttackInnerInterceptor blockAttackInnerInterceptor()
|
{
|
return new BlockAttackInnerInterceptor();
|
}
|
}
|
