update 统一代码间隔符

疯狂的狮子li

2022-01-11 26fc652d33fb82a75d6ec7771ca50ee414b21723

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

@@ -21,81 +21,81 @@
 */
public class InsertAll extends AbstractMethod {

   private final static String[] FILL_PROPERTY = {"createTime", "createBy", "updateTime", "updateBy"};
    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>";
      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, sqlMethod, sqlSource, keyGenerator, keyProperty, keyColumn);
   }
    @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);
        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, sqlMethod, sqlSource, keyGenerator, keyProperty, keyColumn);
    }

   private String prepareFieldSql(TableInfo tableInfo) {
      StringBuilder fieldSql = new StringBuilder();
      if (StringUtils.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 prepareFieldSql(TableInfo tableInfo) {
        StringBuilder fieldSql = new StringBuilder();
        if (StringUtils.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 (StringUtils.isNotBlank(tableInfo.getKeyColumn())) {
         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(",");
            }
         }
      }
      valueSql.append("</foreach>");
      return valueSql.toString();
   }
    private String prepareValuesSqlForMysqlBatch(TableInfo tableInfo) {
        final StringBuilder valueSql = new StringBuilder();
        valueSql.append("<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"(\" separator=\"),(\" close=\")\">");
        if (StringUtils.isNotBlank(tableInfo.getKeyColumn())) {
            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(",");
                }
            }
        }
        valueSql.append("</foreach>");
        return valueSql.toString();
    }
}