mybatisplus-sql注入器

sql注入器

使用mybatisplus只需要继承BaseMapper接口即可使用；但是有新的需求需要扩展BaseMapper里面的功能时可使用sql注入器。

扩展BaseMapper里面的功能

点击查看代码

public interface UserMapper extends BaseMapper<User> {
    List<User> findAll();
}

实现FindAll方法

点击查看代码

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;

public class FindAll extends AbstractMethod {
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        String sql = "select * from " + tableInfo.getTableName();
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        return this.addSelectMappedStatementForTable(mapperClass,"findAll",sqlSource,tableInfo);
    }
}

定义sql注入器

点击查看代码

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.example.mybatisplusdemo.sample.method.FindAll;

import java.util.List;

public class MyInject extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
        methodList.add(new FindAll());
        return methodList;
    }
}

将sql注入器注册到Spring容器中

点击查看代码

@Bean
    public MyInject inject(){
       return new MyInject();
    }

Mybatis-Plus中使用sql注入器场景：mapper 层 选装件

int alwaysUpdateSomeColumnById(T entity);

int insertBatchSomeColumn(List entityList);

int logicDeleteByIdWithFill(T entity);