package cn.stylefeng.roses.kernel.db.mp.injector;
|
|
import com.baomidou.mybatisplus.annotation.FieldFill;
|
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
|
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
|
import com.baomidou.mybatisplus.core.metadata.TableInfo;
|
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
|
|
import java.util.List;
|
|
/**
|
* 自定义，针对saveBatch的优化，改为一次拼接多个values插入到库
|
*
|
* @author fengshuonan
|
* @date 2022/9/17 14:23
|
*/
|
public class CustomInsertBatchSqlInjector extends DefaultSqlInjector {
|
|
@Override
|
public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
|
List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
|
// 例: 不要指定了 update 填充的字段
|
methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));
|
return methodList;
|
}
|
|
}
|
