Преглед изворни кода

修改data-mybatis-plus组件,增加配置项com.qmth.mybatis-plus.block-attack,默认为true开启防全表更新/删除插件

Signed-off-by: luoshi <luoshi@qmth.com.cn>
luoshi пре 2 година
родитељ
комит
51db088463

+ 4 - 2
data-mybatis-plus/src/main/java/com/qmth/boot/mybatis/config/MybatisPlusAutoConfiguration.java

@@ -23,10 +23,12 @@ import javax.validation.constraints.NotNull;
 public class MybatisPlusAutoConfiguration {
 
     @Bean
-    public MybatisPlusInterceptor mybatisPlusInterceptor() {
+    public MybatisPlusInterceptor mybatisPlusInterceptor(@NotNull MybatisProperties mybatisProperties) {
         MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
         interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
-        interceptor.addInnerInterceptor(new BlockAttackInnerInterceptor());
+        if (mybatisProperties.isBlockAttack()) {
+            interceptor.addInnerInterceptor(new BlockAttackInnerInterceptor());
+        }
         return interceptor;
     }