MyBatisPlus（二十）防全表更新与删除

梦里梦外; 2024-02-21 09:35 19阅读 0赞

## 说明 ##

针对 update 和 delete 语句，阻止恶意的全表更新和全表删除。

## 实现方式 ##

配置`BlockAttackInnerInterceptor`拦截器

![在这里插入图片描述][e2517478434d4a9ea00164e1ffb6c5c1.png]

## 代码 ##

package com.example.core.config;
    
    import com.baomidou.mybatisplus.annotation.DbType;
    import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
    import com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor;
    import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
    import org.mybatis.spring.annotation.MapperScan;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    
    @Configuration
    @MapperScan("com.example.web")
    public class MybatisPlusConfig {
        
    
        /**
         * 添加拦截器
         */
        @Bean
        public MybatisPlusInterceptor mybatisPlusInterceptor() {
        
            MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
            interceptor.addInnerInterceptor(new BlockAttackInnerInterceptor()); // 针对 update 和 delete 语句 作用: 阻止恶意的全表更新删除
            interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));// 如果配置多个插件,切记分页最后添加
            // interceptor.addInnerInterceptor(new PaginationInnerInterceptor()); 如果有多数据源可以不配具体类型 否则都建议配上具体的DbType
            return interceptor;
        }
    }

## 测试 ##

### 更新全表 ###

/**
         * 更新全表
         */
        @Test
        public void updateAll() {
        
            User user = new User();
            user.setGender(GenderEnum.MALE);
    
            mapper.update(user, null);
        }

![在这里插入图片描述][9e8023aee6ac436093e32be64faf3105.png]

### 删除全表 ###

/**
         * 删除全表
         */
        @Test
        public void deleteAll() {
        
            mapper.delete(null);
        }

![在这里插入图片描述][2f04670c9984447b908046afff9a18a8.png]

### 正常更新 ###

正常更新，不受影响。

/**
         * 更新一条数据
         */
        @Test
        public void update() {
        
            User user = new User();
            user.setId(7L);
            user.setGender(GenderEnum.MALE);
    
            mapper.updateById(user);
        }

![在这里插入图片描述][4b7273ef622045258af73ee2050c8a36.png]

## 未开启防护前 ##

未开启防护前，可以更新全表，或删除全表。

### 全表更新 ###

![在这里插入图片描述][89a01c9d4cc94fcea316ec4b8a536814.png]  
![在这里插入图片描述][1a90b3bacb864387b5833f1717782c22.png]

### 全表删除 ###

![在这里插入图片描述][26f0186c5817464f959ce2bdde0a6902.png]  
![在这里插入图片描述][60230af08bcd4c29a406d4f13aa17633.png]

[e2517478434d4a9ea00164e1ffb6c5c1.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/fd0d5b51f11a46bd95453cb834bc69e2.png
[9e8023aee6ac436093e32be64faf3105.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/ff10eeccf3ee41c9a26ec1baf06d9775.png
[2f04670c9984447b908046afff9a18a8.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/54baf3daf5164d02925fd9525decfc03.png
[4b7273ef622045258af73ee2050c8a36.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/166aa844687c48fd93ef368bcb42adbe.png
[89a01c9d4cc94fcea316ec4b8a536814.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/74b5b95354f6485da32ef5831480daf5.png
[1a90b3bacb864387b5833f1717782c22.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/16c04eb18de34c40a4f347e2d9e354f9.png
[26f0186c5817464f959ce2bdde0a6902.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/9a573f76bac24d7abae6686c4771ac22.png
[60230af08bcd4c29a406d4f13aa17633.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/21/b85f3fad97fc4a9fba46f29a3b3c5a2e.png