疯狂的狮子Li
2021-12-09 da247c61b715dafaba58e8dd4f106f0954f6cbf4
ruoyi-framework/src/main/java/com/ruoyi/framework/config/MybatisPlusConfig.java
@@ -1,24 +1,30 @@
package com.ruoyi.framework.config;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.autoconfigure.ConfigurationCustomizer;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.baomidou.mybatisplus.core.incrementer.IKeyGenerator;
import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.injector.ISqlInjector;
import com.baomidou.mybatisplus.extension.incrementer.H2KeyGenerator;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.IllegalSQLInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.ruoyi.common.core.mybatisplus.methods.InsertAll;
import com.ruoyi.framework.handler.CreateAndUpdateMetaObjectHandler;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import java.util.List;
/**
 * mybatis-plus配置类(下方注释有插件介绍)
 *
 * @author Lion Li
 */
@EnableTransactionManagement(proxyTargetClass = true)
@Configuration
@MapperScan("${mybatis-plus.mapperPackage}")
public class MybatisPlusConfig {
   @Bean
@@ -28,101 +34,65 @@
      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);
      // 分页合理化
      paginationInnerInterceptor.setOverflow(true);
      return paginationInnerInterceptor;
   }
   /**
    * 新的分页插件,一缓和二缓遵循mybatis的规则,需要设置 MybatisConfiguration#useDeprecatedExecutor = false 避免缓存出现问题(该属性会在旧插件移除后一同移除)
    */
   @Bean
   public ConfigurationCustomizer configurationCustomizer() {
      return configuration -> configuration.setUseDeprecatedExecutor(false);
   }
   /**
    * 乐观锁插件
    * 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();
   @Bean
   public MetaObjectHandler metaObjectHandler() {
      return new CreateAndUpdateMetaObjectHandler();
   }
   /**
    * sql性能规范插件(垃圾SQL拦截)
    * 如有需要可以启用
    */
//   public IllegalSQLInnerInterceptor illegalSQLInnerInterceptor() {
//      return new IllegalSQLInnerInterceptor();
//   }
   /**
    * Sequence主键策略 IdType.INPUT 时使用
    * 内置支持:
    *
    * DB2KeyGenerator
    * H2KeyGenerator
    * KingbaseKeyGenerator
    * OracleKeyGenerator
    * PostgreKeyGenerator
    * https://baomidou.com/guide/sequence.html
    */
//   @Bean
//   public IKeyGenerator keyGenerator() {
//      return new H2KeyGenerator();
//   }
   /**
    * 自定义主键策略
    * https://baomidou.com/guide/id-generator.html
    */
//   @Bean
//   public IdentifierGenerator idGenerator() {
//      return new CustomIdGenerator();
//   }
   /**
    * 元对象字段填充控制器
    * https://baomidou.com/guide/auto-fill-metainfo.html
    */
//   @Bean
//   public MetaObjectHandler metaObjectHandler() {
//      return new MyMetaObjectHandler();
//   }
   /**
    * sql注入器配置
    * https://baomidou.com/guide/sql-injector.html
    */
//   @Bean
//   public ISqlInjector sqlInjector() {
//      return new DefaultSqlInjector();
//   }
   @Bean
   public ISqlInjector sqlInjector() {
      return new DefaultSqlInjector() {
         @Override
         public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
            List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
            methodList.add(new InsertAll());
            return methodList;
         }
      };
   }
   /**
    * PaginationInnerInterceptor 分页插件,自动识别数据库类型
    * https://baomidou.com/guide/interceptor-pagination.html
    * OptimisticLockerInnerInterceptor 乐观锁插件
    * https://baomidou.com/guide/interceptor-optimistic-locker.html
    * MetaObjectHandler 元对象字段填充控制器
    * https://baomidou.com/guide/auto-fill-metainfo.html
    * ISqlInjector sql注入器
    * https://baomidou.com/guide/sql-injector.html
    * BlockAttackInnerInterceptor 如果是对全表的删除或更新操作,就会终止该操作
    * https://baomidou.com/guide/interceptor-block-attack.html
    * IllegalSQLInnerInterceptor sql性能规范插件(垃圾SQL拦截)
    * IdentifierGenerator 自定义主键策略
    * https://baomidou.com/guide/id-generator.html
    * TenantLineInnerInterceptor 多租户插件
    * https://baomidou.com/guide/interceptor-tenant-line.html
    * DynamicTableNameInnerInterceptor 动态表名插件