From 8ffe1110861ff04ad26ae6203f0866b4dd5350b2 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期三, 25 八月 2021 18:01:52 +0800
Subject: [PATCH] update 重写 InsertAll 方法实现 可完美替代 saveBatch 秒级插入上万数据
---
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java | 2 +-
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/BaseMapperPlus.java | 3 +--
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java | 35 +++++++++++++++++++++++++++++++----
ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestBatchController.java | 2 +-
4 files changed, 34 insertions(+), 8 deletions(-)
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/BaseMapperPlus.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/BaseMapperPlus.java
index ee779a5..ee20abd 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/BaseMapperPlus.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/BaseMapperPlus.java
@@ -14,8 +14,7 @@
public interface BaseMapperPlus<T> extends BaseMapper<T> {
/**
- * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 鏃犺鏁版嵁搴撻粯璁ゅ�� )
- * 閫傜敤浜庢棤鑴戞彃鍏�
+ * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 )
*/
int insertAll(@Param("list") Collection<T> batchList);
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
index e00076e..90a1bd4 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
@@ -113,7 +113,7 @@
}
/**
- * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 鏃犺鏁版嵁搴撻粯璁ゅ�� )
+ * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 )
* 閫傜敤浜庢棤鑴戞彃鍏�
*/
@Override
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java
index 2aba06e..403ea53 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java
+++ b/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;
+
/**
- * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 鏃犺鏁版嵁搴撻粯璁ゅ�� )
+ * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 )
*
* @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();
}
diff --git a/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestBatchController.java b/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestBatchController.java
index 9a66fdd..f3a7f6d 100644
--- a/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestBatchController.java
+++ b/ruoyi-demo/src/main/java/com/ruoyi/demo/controller/TestBatchController.java
@@ -32,7 +32,7 @@
private final ITestDemoService iTestDemoService;
/**
- * 鏂板鎵归噺鏂规硶 ( 鍏ㄩ噺瑕嗙洊濉厖 )
+ * 鏂板鎵归噺鏂规硶 鍙畬缇庢浛浠� saveBatch 绉掔骇鎻掑叆涓婁竾鏁版嵁 (瀵筸ysql璐熻嵎杈冨ぇ)
*/
@ApiOperation(value = "鏂板鎵归噺鏂规硶")
@PostMapping()
--
Gitblit v1.9.3