update 日常字符串校验 统一重构到 StringUtils 便于维护扩展

疯狂的狮子li

2021-08-02 f8ab5663ef7220f8adc0fd8fcbd7d0cf84c6d77d

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

@@ -1,15 +1,18 @@
package com.ruoyi.common.core.mybatisplus.methods;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.ruoyi.common.utils.StringUtils;
import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator;
import org.apache.ibatis.executor.keygen.KeyGenerator;
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;

/**
 * 单sql批量插入
 * 单sql批量插入( 全量填充 无视数据库默认值 )
 *
 * @author Lion Li
 */
@@ -20,14 +23,33 @@
      final String sql = "<script>insert into %s %s values %s</script>";
      final String fieldSql = prepareFieldSql(tableInfo);
      final String valueSql = prepareValuesSqlForMysqlBatch(tableInfo);
      KeyGenerator keyGenerator = new NoKeyGenerator();
      String sqlMethod = "insertAll";
      String keyProperty = null;
      String keyColumn = null;
      // 表包含主键处理逻辑,如果不包含主键当普通字段处理
      if (StringUtils.isNotBlank(tableInfo.getKeyProperty())) {
         if (tableInfo.getIdType() == IdType.AUTO) {
            /** 自增主键 */
            keyGenerator = new Jdbc3KeyGenerator();
            keyProperty = tableInfo.getKeyProperty();
            keyColumn = tableInfo.getKeyColumn();
         } else {
            if (null != tableInfo.getKeySequence()) {
               keyGenerator = TableInfoHelper.genKeyGenerator(sqlMethod, tableInfo, builderAssistant);
               keyProperty = tableInfo.getKeyProperty();
               keyColumn = tableInfo.getKeyColumn();
            }
         }
      }
      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);
      return this.addInsertMappedStatement(mapperClass, modelClass, sqlMethod, sqlSource, keyGenerator, keyProperty, keyColumn);
   }

   private String prepareFieldSql(TableInfo tableInfo) {
      StringBuilder fieldSql = new StringBuilder();
      if (StrUtil.isNotBlank(tableInfo.getKeyColumn())) {
      if (StringUtils.isNotBlank(tableInfo.getKeyColumn())) {
         fieldSql.append(tableInfo.getKeyColumn()).append(",");
      }
      tableInfo.getFieldList().forEach(x -> fieldSql.append(x.getColumn()).append(","));
@@ -40,7 +62,7 @@
   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())) {
      if (StringUtils.isNotBlank(tableInfo.getKeyColumn())) {
         valueSql.append("#{item.").append(tableInfo.getKeyProperty()).append("},");
      }
      tableInfo.getFieldList().forEach(x -> valueSql.append("#{item.").append(x.getProperty()).append("},"));