Browse Source

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

疯狂的狮子li 3 years ago
parent
commit
8ffe111086

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

@@ -14,8 +14,7 @@ import java.util.Collection;
 public interface BaseMapperPlus<T> extends BaseMapper<T> {
 
 	/**
-	 * 单sql批量插入( 全量填充 无视数据库默认值 )
-	 * 适用于无脑插入
+	 * 单sql批量插入( 全量填充 )
 	 */
 	int insertAll(@Param("list") Collection<T> batchList);
 

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

@@ -113,7 +113,7 @@ public class ServicePlusImpl<M extends BaseMapperPlus<T>, T, K> extends ServiceI
 	}
 
 	/**
-	 * 单sql批量插入( 全量填充 无视数据库默认值 )
+	 * 单sql批量插入( 全量填充 )
 	 * 适用于无脑插入
 	 */
 	@Override

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

@@ -2,6 +2,7 @@ package com.ruoyi.common.core.mybatisplus.methods;
 
 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,13 +12,17 @@ import org.apache.ibatis.executor.keygen.NoKeyGenerator;
 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) {
 		final String sql = "<script>insert into %s %s values %s</script>";
@@ -63,10 +68,32 @@ public class InsertAll extends AbstractMethod {
 		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");
+		}
+		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(",");
+				}
+			}
 		}
-		tableInfo.getFieldList().forEach(x -> valueSql.append("#{item.").append(x.getProperty()).append("},"));
-		valueSql.delete(valueSql.length() - 1, valueSql.length());
 		valueSql.append("</foreach>");
 		return valueSql.toString();
 	}

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

@@ -32,7 +32,7 @@ public class TestBatchController extends BaseController {
     private final ITestDemoService iTestDemoService;
 
     /**
-     * 新增批量方法 ( 全量覆盖填充 )
+     * 新增批量方法 可完美替代 saveBatch 秒级插入上万数据 (对mysql负荷较大)
      */
 	@ApiOperation(value = "新增批量方法")
     @PostMapping()