Mybatisplus详解如何注入自定义的SQL

2024-04-02 19:04:59 427人浏览安东尼

Python 官方文档：入门教程 => 点击学习

摘要

目录sql 注入器第一步重写getMethodList第二步定义自己的SQL方法类第三步定义添加了自定义方法的Mapper类测试调用遗留问题SQL 注入器官方文档提供了一个小案例

SQL 注入器

官方文档提供了一个小案例自定义 Mapper 示例

解读：DefaultSqlInjector就是一个注册类，其中注册了一系列 mybatis-plus 内置的 update,insert,select SQL 语句方法，

并且对表主键是否存在进行了判定：如果设置了主键，那么会注册 DeleteById 等方法，没有则不注册。最终返回所有将要被注入的 SQL 语句 List

public class DefaultSqlInjector extends AbstractSqlInjector {
    public DefaultSqlInjector() {}
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        Builder<AbstractMethod> builder = Stream.builder().add(new Insert()).add(new Delete()).add(new DeleteByMap()).add(new Update()).add(new SelectByMap()).add(new SelectCount()).add(new SelectMaps()).add(new SelectMapsPage()).add(new SelectObjs()).add(new SelectList()).add(new SelectPage());
        if (tableInfo.havePK()) {
            builder.add(new DeleteById()).add(new DeleteBatchByIds()).add(new UpdateById()).add(new SelectById()).add(new SelectBatchByIds());
        } else {
            this.logger.warn(String.fORMat("%s ,Not found @TableId annotation, Cannot use Mybatis-Plus 'xxById' Method.", tableInfo.getEntityType()));
        }
        return (List)builder.build().collect(Collectors.toList());
    }
}

其中，Builder<AbstractMethod> builder就是可以做文章得地方。

一顿操作,最终是返回一个总的 List, 那么可以通过重写 getMethodList 方法，在返回得 List 中添加上我们自己得方法类即可。
不是任意的方法类都可以被添加到这个 List 当中。继续查看，跳入 new Insert()，该方法仅仅做了一个操作：继承了抽象类AbstractMethod。查看其他几个new Update(), new Delete() 操作类发现同样如此
因此，我们要实现清表删库功能 clearOneTable，同样只需要继承抽象类 AbstractMethod 完事。

第一步重写getMethodList

ReGISterSqlInjector.java

package com.example.commonmybatisplus.utils;
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.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.lang.reflect.TypeVariable;
import java.lang.reflect.WildcardType;
import java.util.Collections;
import java.util.List;

 // 基于SpringBoot的注解： @Component
@Component
public class RegisterSqlInjector extends DefaultSqlInjector {
    
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        // 1. 直接调用父类方法，获得mybatis-plus注入SQL方法后的List
        List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
        methodList.forEach(System.out::println);
        // 2. 添加我自定义得类
        methodList.add(new DefineSqlInjector());
        return methodList;
    }
}

第二步定义自己的SQL方法类

DefineSqlInjector.java

package com.example.commonmybatisplus.utils;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.AbstractSqlInjector;
import com.baomidou.mybatisplus.core.injector.methods.*;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;
import org.springframework.stereotype.Component;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

@Component
public class DefineSqlInjector extends AbstractMethod{
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        
        String sql = "delete from " + tableInfo.getTableName();
        
        String method = "clearOneTable";
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        return this.aDDDeleteMappedStatement(mapperClass, method, sqlSource);
    }
}

第三步定义添加了自定义方法的Mapper类

package com.example.commonmybatisplus.utils;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.example.commonmybatisplus.entity.UserEntity;

public interface BaseDao<T> extends BaseMapper<T> {
    
    int clearOneTable();
    
    int addBackDoor();
}

测试调用

package com.example.commonmybatisplus;
import com.example.commonmybatisplus.dao.UserDao;
import com.example.commonmybatisplus.entity.UserEntity;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
@SpringBootTest
class CommonMybatisPlusApplicationTests {
    @Autowired
    private UserDao userDao;
    @Test
    void contextLoads() {
        userDao.clearOneTable();
    }
}