From c12dc71ecea33d5c23e2ceb32bfc3db68e4198e3 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期三, 23 六月 2021 09:44:04 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/dev'
---
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 52 insertions(+), 0 deletions(-)
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
new file mode 100644
index 0000000..9c8c0f7
--- /dev/null
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/methods/InsertAll.java
@@ -0,0 +1,52 @@
+package com.ruoyi.common.core.mybatisplus.methods;
+
+import cn.hutool.core.util.StrUtil;
+import org.apache.ibatis.executor.keygen.NoKeyGenerator;
+import org.apache.ibatis.mapping.MappedStatement;
+import org.apache.ibatis.mapping.SqlSource;
+
+import com.baomidou.mybatisplus.core.injector.AbstractMethod;
+import com.baomidou.mybatisplus.core.metadata.TableInfo;
+
+/**
+ * 鍗晄ql鎵归噺鎻掑叆
+ *
+ * @author Lion Li
+ */
+public class InsertAll extends AbstractMethod {
+
+ @Override
+ public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
+ final String sql = "<script>insert into %s %s values %s</script>";
+ final String fieldSql = prepareFieldSql(tableInfo);
+ final String valueSql = prepareValuesSqlForMysqlBatch(tableInfo);
+ final String sqlResult = String.format(sql, tableInfo.getTableName(), fieldSql, valueSql);
+ SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
+ return this.addInsertMappedStatement(mapperClass, modelClass, "insertAll", sqlSource, new NoKeyGenerator(), null, null);
+ }
+
+ private String prepareFieldSql(TableInfo tableInfo) {
+ StringBuilder fieldSql = new StringBuilder();
+ if (StrUtil.isNotBlank(tableInfo.getKeyColumn())) {
+ fieldSql.append(tableInfo.getKeyColumn()).append(",");
+ }
+ tableInfo.getFieldList().forEach(x -> fieldSql.append(x.getColumn()).append(","));
+ fieldSql.delete(fieldSql.length() - 1, fieldSql.length());
+ fieldSql.insert(0, "(");
+ fieldSql.append(")");
+ return fieldSql.toString();
+ }
+
+ private String prepareValuesSqlForMysqlBatch(TableInfo tableInfo) {
+ final StringBuilder valueSql = new StringBuilder();
+ valueSql.append("<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"(\" separator=\"),(\" close=\")\">");
+ if (StrUtil.isNotBlank(tableInfo.getKeyColumn())) {
+ valueSql.append("#{item.").append(tableInfo.getKeyProperty()).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();
+ }
+}
+
--
Gitblit v1.9.3