| | |
| | | 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 com.ruoyi.common.interceptor.DataScopeSqlInterceptor; |
| | | import org.apache.ibatis.executor.Executor; |
| | | import org.apache.ibatis.mapping.BoundSql; |
| | |
| | | import org.springframework.context.annotation.Configuration; |
| | | import org.springframework.transaction.annotation.EnableTransactionManagement; |
| | | |
| | | import java.sql.SQLException; |
| | | |
| | | /** |
| | | * Mybatis Plus 配置 |
| | | * |
| | | * @author ruoyi |
| | | * MyBatis Plus config. |
| | | */ |
| | | @EnableTransactionManagement(proxyTargetClass = true) |
| | | @Configuration |
| | |
| | | private DataScopeSqlInterceptor dataScopeSqlInterceptor; |
| | | |
| | | @Bean |
| | | public MybatisPlusInterceptor mybatisPlusInterceptor() |
| | | { |
| | | 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()); |
| | | // 数据权限插件 |
| | | // Rewrite the original SQL before pagination generates the count query. |
| | | interceptor.addInnerInterceptor(dataScopeSqlInterceptor); |
| | | interceptor.addInnerInterceptor(paginationInnerInterceptor()); |
| | | interceptor.addInnerInterceptor(optimisticLockerInnerInterceptor()); |
| | | interceptor.addInnerInterceptor(blockAttackInnerInterceptor()); |
| | | return interceptor; |
| | | } |
| | | |
| | | /** |
| | | * 分页插件,自动识别数据库类型 https://baomidou.com/guide/interceptor-pagination.html |
| | | * Pagination interceptor. |
| | | */ |
| | | // 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 { |
| | | RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) { |
| | | 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); // 建议设置合理的最大值 |
| | | interceptor.setMaxLimit(1000L); |
| | | return interceptor; |
| | | } |
| | | |
| | | /** |
| | | * 乐观锁插件 https://baomidou.com/guide/interceptor-optimistic-locker.html |
| | | * Optimistic lock interceptor. |
| | | */ |
| | | public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor() |
| | | { |
| | | public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor() { |
| | | return new OptimisticLockerInnerInterceptor(); |
| | | } |
| | | |
| | | /** |
| | | * 如果是对全表的删除或更新操作,就会终止该操作 https://baomidou.com/guide/interceptor-block-attack.html |
| | | * Block full-table update and delete. |
| | | */ |
| | | public BlockAttackInnerInterceptor blockAttackInnerInterceptor() |
| | | { |
| | | public BlockAttackInnerInterceptor blockAttackInnerInterceptor() { |
| | | return new BlockAttackInnerInterceptor(); |
| | | } |
| | | } |
| | | } |