| | |
| | | package com.ruoyi.framework.config; |
| | | |
| | | import com.baomidou.mybatisplus.annotation.DbType; |
| | | import com.baomidou.mybatisplus.autoconfigure.ConfigurationCustomizer; |
| | | 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.ruoyi.common.handler.LocalDateTimeTypeHandler; |
| | | import org.apache.ibatis.type.JdbcType; |
| | | import org.springframework.context.annotation.Bean; |
| | | import org.springframework.context.annotation.Configuration; |
| | | import org.springframework.transaction.annotation.EnableTransactionManagement; |
| | | |
| | | /** |
| | | * Mybatis Plus 配置 |
| | | * |
| | | * @author ruoyi |
| | | */ |
| | | @EnableTransactionManagement(proxyTargetClass = true) |
| | | @Configuration |
| | | public class MybatisPlusConfig { |
| | | |
| | | @Bean |
| | | public MybatisPlusInterceptor mybatisPlusInterceptor() { |
| | | MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); |
| | | // 分页插件 |
| | | interceptor.addInnerInterceptor(paginationInnerInterceptor()); |
| | | // 乐观锁插件 |
| | | interceptor.addInnerInterceptor(optimisticLockerInnerInterceptor()); |
| | | // 阻断插件 |
| | | interceptor.addInnerInterceptor(blockAttackInnerInterceptor()); |
| | | // 1. 分页插件 - 指定 PostgreSQL 数据库类型 |
| | | PaginationInnerInterceptor pagination = new PaginationInnerInterceptor(DbType.POSTGRE_SQL); |
| | | pagination.setOptimizeJoin(true); // 优化 JOIN 查询 |
| | | pagination.setMaxLimit(500L); // 设置最大单页限制 |
| | | interceptor.addInnerInterceptor(pagination); |
| | | |
| | | // 2. 乐观锁插件 |
| | | interceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor()); |
| | | |
| | | // 3. 防止全表更新/删除插件 |
| | | interceptor.addInnerInterceptor(new 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; |
| | | } |
| | | // 4. 添加全局类型处理器配置 |
| | | @Bean |
| | | public ConfigurationCustomizer configurationCustomizer() { |
| | | return configuration -> { |
| | | // 注册 LocalDateTime 类型处理器 |
| | | configuration.getTypeHandlerRegistry().register(new LocalDateTimeTypeHandler()); |
| | | |
| | | /** |
| | | * 乐观锁插件 https://baomidou.com/guide/interceptor-optimistic-locker.html |
| | | */ |
| | | public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor() { |
| | | return new OptimisticLockerInnerInterceptor(); |
| | | } |
| | | // 设置空值处理策略 |
| | | configuration.setJdbcTypeForNull(JdbcType.NULL); |
| | | |
| | | /** |
| | | * 如果是对全表的删除或更新操作,就会终止该操作 https://baomidou.com/guide/interceptor-block-attack.html |
| | | */ |
| | | public BlockAttackInnerInterceptor blockAttackInnerInterceptor() { |
| | | return new BlockAttackInnerInterceptor(); |
| | | // 启用自动驼峰命名规则映射 |
| | | configuration.setMapUnderscoreToCamelCase(true); |
| | | }; |
| | | } |
| | | } |