Java教程

【Mybatis-Plus进阶学习(八)】SQL注入器

本文主要是介绍【Mybatis-Plus进阶学习(八)】SQL注入器,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
  • 使用SQL注入器就可以自定义例如selectById的默认方法。

    实现步骤
    Step1:创建定义方法的类;
    Step2:创建注入器;
    Step3:在Mapper中加入自定义方法。

    自定义注入器的简单使用

    第一步:创建定义方法的类

    public class DeleteAllMethod  extends AbstractMethod {
        @Override
        public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
            //执行的SQL
            String sql = "delete from " + tableInfo.getTableName();
            //mapper接口方法名
            String method = "deleteAll";
            SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
            return addDeleteMappedStatement(mapperClass, method, sqlSource);
        }
    }
    

    第二步:创建注入器

    @Component
    public class MySqlInjector extends DefaultSqlInjector {
    
        @Override
        public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
            //MP自定义的SQL语句,如果不添加,MP自定义的语句就不能用了
            List<AbstractMethod> methodList = super.getMethodList(mapperClass);
            methodList.add(new DeleteAllMethod());
            return methodList;
        }
    
    }
    

    DefaultSqlInjector、AbstractSqlInjector、ISqlInjector都能继承。

    第三步:在UserMapper中加入自定义方法deleteAll

        /**
         * 删除所有数据
         * @return 影响行数
         */
        int deleteAll();
    

    第四步:测试

        @Test
        public void deleteAll() {
            userMapper.deleteAll();
        }
    

    结果如下:

    这里需要注意:

    1. 不知道大家还记不记得之前逻辑删除,3.1.2之前的版本配置了一个SQL注入器。自定义SQL注入器不可以和逻辑删除一起配置,因为不能同时制定两个sql注入器,所以报错。有一个办法就是MySqlinjector不要继承DefaultSqlInjector直接继承LogicSqlInjector就可以了,这样配置这一个sql注入器,就既能使用逻辑删除又能加入自定义方法了。

    2. 如果报以下错误

    当删除所有数据时,由于有的数据行有外键的约束,不允许你进行物理删除。使用语句删除外键即可,alter table user drop foreign key manager_fk;。或者把外键字段原来默认的Restrict改成CASCADE。

    同时附上初始化数据语句,可以多次尝试

    #初始化数据:
    INSERT INTO user (id, name, age, email, manager_id
    	, create_time)
    VALUES (1087982257332887553, '大boss', 40, 'boss@baomidou.com', NULL
    		, '2019-01-11 14:20:20'),
    	(1088248166370832385, '王天风', 25, 'wtf@baomidou.com', 1087982257332887553
    		, '2019-02-05 11:12:22'),
    	(1088250446457389058, '李艺伟', 28, 'lyw@baomidou.com', 1088248166370832385
    		, '2019-02-14 08:31:16'),
    	(1094590409767661570, '张雨琪', 31, 'zjq@baomidou.com', 1088248166370832385
    		, '2019-01-14 09:15:15'),
    	(1094592041087729666, '刘红雨', 32, 'lhm@baomidou.com', 1088248166370832385
    		, '2019-01-14 09:48:16');
    

    优化

    如果每张表都需要删除全部,那我们为每个Mapper都写一个deleteAll()方法,非常的繁琐。这时候我们可以自己建立一个MyMapper接口,MyMapper接口继承BaseMapper,而类似于UserMapper的自定义Mapper全部继承MyMapper,这样我们就只需要在MyMapper中编写一个deleteAll方法,再由UserMpper和其他Mapper继承就可以实现deleteAll()方法的重用。

    选装件InsertBatchSomeColumn

    选择件其实也是自定义SQL注入器,但是是由官方提供的。InsertBatchSomeColumn的主要功能是批量新增数据,自选字段insert。接下来让我们看看是如何实现的吧。
    第一步:注入器中加入方法

        @Override
        public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
            //MP自定义的SQL语句,如果不添加,MP自定义的语句就不能用了
            List<AbstractMethod> methodList = super.getMethodList(mapperClass);
            methodList.add(new DeleteAllMethod());
            //加入选装件InsertBatchSomeColumn,同时排除逻辑删除字段
            methodList.add(new InsertBatchSomeColumn(t->!t.isLogicDelete()));
            return methodList;
        }
    

    第二步:在MyMapper中加入insertBatchSomeColumn方法

    public interface MyMapper<T> extends BaseMapper<T> {
        /**
         * 删除所有数据
         * @return 影响行数
         */
        int deleteAll();
    
        int insertBatchSomeColumn(List<T> list);
    }
    

    第三步:测试

     @Test
        public void insertBatchSomeColumn() {
    
            User user = new User();
            user.setName("王聚义");
            user.setAge(26);
            user.setEmail("wjy@163.com");
            user.setManagerId(1088248166370832385L);
    
            User user2 = new User();
            user2.setName("王收义");
            user2.setAge(28);
            user2.setEmail("wsy@163.com");
            user2.setManagerId(1088248166370832385L);
    
            List<User> list = new ArrayList<>();
            list.add(user);
            list.add(user2);
    
            userMapper.insertBatchSomeColumn(list);
        }
    

    可以看出批量插入成功。

    需要注意的是,User对象中为Null的值会覆盖没有被排除字段的默认值。例如version的默认值为1,但是我们插入的User的version为null,如果version这个字段在注入器组件中没有被排除,结果数据库中的值应该为null。

    同时作者在注解中也提示,选装器只在mysql中测试过,所以慎用。

    选装件LogicDeleteByIdWithFill

    LogicDeleteByIdWithFill的主要功能是根据id逻辑删除数据并带字段填充功能。例如在删除的时候需要添加操作人。

    第一步:注入器中加入方法

     methodList.add(new LogicDeleteByIdWithFill());
    

    第二步:在MyMapper中加入方法

        int deleteByIdWithFill(T entity);
    

    第三步:测试

     @Test
        public void deleteByIdWithFill() {
            User user = new User();
            user.setId(1346432120618147841L);
            user.setUpdateTime(LocalDateTime.now());
            userMapper.deleteByIdWithFill(user);
        }
    

    注意,如果想在删除的同时修改数据,被修改的数据必须有自动填充标识,不然无法完成更新。结果如下:

    选装件AlwaysUpdateSomeColumnById

    AlwaysUpdateSomeColumnById的主要功能是根据id更新固定的某些字段。

    第一步:注入器中加入方法

            methodList.add(new AlwaysUpdateSomeColumnById(t->!t.getColumn().equals("name")));
    

    第二步:在MyMapper中加入方法

     int alwaysUpdateSomeColumnById(@Param(Constants.ENTITY) T entity);
    

    第三步:测试

        @Test
        public void alwaysUpdateSomeColumnById() {
            User user = new User();
            user.setId(1346432120618147841L);
            user.setName("王地风");
            user.setAge(16);
            userMapper.alwaysUpdateSomeColumnById(user);
        }
    

    结果如下:

  • 相关阅读:
这篇关于【Mybatis-Plus进阶学习(八)】SQL注入器的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!