Merge remote-tracking branch 'origin/dev'

疯狂的狮子li

2021-06-23 c12dc71ecea33d5c23e2ceb32bfc3db68e4198e3

 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();
   }
}