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
int logicDeleteByIdWithFill(T entity);