update 重写 InsertAll 方法实现 可完美替代 saveBatch 秒级插入上万数据

疯狂的狮子li

2021-08-25 8ffe1110861ff04ad26ae6203f0866b4dd5350b2

update 重写 InsertAll 方法实现 可完美替代 saveBatch 秒级插入上万数据

 ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/BaseMapperPlus.java

@@ -14,8 +14,7 @@
public interface BaseMapperPlus<T> extends BaseMapper<T> {

    /**
     * 单sql批量插入( 全量填充 无视数据库默认值 )
     * 适用于无脑插入
     * 单sql批量插入( 全量填充 )
     */
    int insertAll(@Param("list") Collection<T> batchList);


 ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java

@@ -113,7 +113,7 @@
    }

    /**
     * 单sql批量插入( 全量填充 无视数据库默认值 )
     * 单sql批量插入( 全量填充 )
     * 适用于无脑插入
     */
    @Override

 ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java

@@ -2,6 +2,7 @@

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.ruoyi.common.utils.StringUtils;
@@ -11,12 +12,16 @@
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;

import java.util.List;

/**
 * 单sql批量插入( 全量填充 无视数据库默认值 )
 * 单sql批量插入( 全量填充 )
 *
 * @author Lion Li
 */
public class InsertAll extends AbstractMethod {

    private final static String[] FILL_PROPERTY = {"createTime", "createBy", "updateTime", "updateBy"};

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
@@ -63,10 +68,32 @@
        final StringBuilder valueSql = new StringBuilder();
        valueSql.append("<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"(\" separator=\"),(\" close=\")\">");
        if (StringUtils.isNotBlank(tableInfo.getKeyColumn())) {
            valueSql.append("#{item.").append(tableInfo.getKeyProperty()).append("},");
            valueSql.append("\n#{item.").append(tableInfo.getKeyProperty()).append("},\n");
        }
        tableInfo.getFieldList().forEach(x -> valueSql.append("#{item.").append(x.getProperty()).append("},"));
        valueSql.delete(valueSql.length() - 1, valueSql.length());
        List<TableFieldInfo> fieldList = tableInfo.getFieldList();
        int last = fieldList.size() - 1;
        for (int i = 0; i < fieldList.size(); i++) {
            String property = fieldList.get(i).getProperty();
            if (!StringUtils.equalsAny(property, FILL_PROPERTY)) {
                valueSql.append("<if test=\"item.").append(property).append(" != null\">");
                valueSql.append("#{item.").append(property).append("}");
                if (i != last) {
                    valueSql.append(",");
                }
                valueSql.append("</if>");
                valueSql.append("<if test=\"item.").append(property).append(" == null\">");
                valueSql.append("DEFAULT");
                if (i != last) {
                    valueSql.append(",");
                }
                valueSql.append("</if>");
            } else {
                valueSql.append("#{item.").append(property).append("}");
                if (i != last) {
                    valueSql.append(",");
                }
            }
        }
        valueSql.append("</foreach>");
        return valueSql.toString();
    }

 ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestBatchController.java

@@ -32,7 +32,7 @@
    private final ITestDemoService iTestDemoService;

    /**
     * 新增批量方法 ( 全量覆盖填充 )
     * 新增批量方法 可完美替代 saveBatch 秒级插入上万数据 (对mysql负荷较大)
     */
    @ApiOperation(value = "新增批量方法")
    @PostMapping()