【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();
    }

结果如下：

这里需要注意：

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

当删除所有数据时，由于有的数据行有外键的约束，不允许你进行物理删除。使用语句删除外键即可，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);
    }

结果如下：

相关阅读:

posted @ 2021-12-02 02:28 牧之丨阅读(450) 评论(0) 编辑收藏举报

会员力量，点亮园子希望

刷新页面返回顶部

【Mybatis-Plus进阶学习（八）】SQL注入器

使用SQL注入器就可以自定义例如selectById的默认方法。

自定义注入器的简单使用

优化

选装件InsertBatchSomeColumn

选装件LogicDeleteByIdWithFill

选装件AlwaysUpdateSomeColumnById

公告