疯狂的狮子li
2021-06-16 bb2ade75a82d768db99a46cff54eecc0b67d4de0
ruoyi-generator/src/main/java/com/ruoyi/generator/util/GenUtils.java
@@ -1,16 +1,17 @@
package com.ruoyi.generator.util;
import java.util.Arrays;
import org.apache.commons.lang3.RegExUtils;
import cn.hutool.core.util.StrUtil;
import com.ruoyi.common.constant.GenConstants;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.generator.config.GenConfig;
import com.ruoyi.generator.domain.GenTable;
import com.ruoyi.generator.domain.GenTableColumn;
import org.apache.commons.lang3.RegExUtils;
import java.util.Arrays;
/**
 * 代码生成器 工具类
 *
 *
 * @author ruoyi
 */
public class GenUtils
@@ -39,7 +40,7 @@
        column.setTableId(table.getTableId());
        column.setCreateBy(table.getCreateBy());
        // 设置java字段名
        column.setJavaField(StringUtils.toCamelCase(columnName));
        column.setJavaField(StrUtil.toCamelCase(columnName));
        // 设置默认类型
        column.setJavaType(GenConstants.TYPE_STRING);
@@ -60,7 +61,7 @@
            column.setHtmlType(GenConstants.HTML_INPUT);
            // 如果是浮点型 统一用BigDecimal
            String[] str = StringUtils.split(StringUtils.substringBetween(column.getColumnType(), "(", ")"), ",");
            String[] str = StrUtil.splitToArray(StrUtil.subBetween(column.getColumnType(), "(", ")"), ",");
            if (str != null && str.length == 2 && Integer.parseInt(str[1]) > 0)
            {
                column.setJavaType(GenConstants.TYPE_BIGDECIMAL);
@@ -97,28 +98,33 @@
        }
        // 查询字段类型
        if (StringUtils.endsWithIgnoreCase(columnName, "name"))
        if (StrUtil.endWithIgnoreCase(columnName, "name"))
        {
            column.setQueryType(GenConstants.QUERY_LIKE);
        }
        // 状态字段设置单选框
        if (StringUtils.endsWithIgnoreCase(columnName, "status"))
        if (StrUtil.endWithIgnoreCase(columnName, "status"))
        {
            column.setHtmlType(GenConstants.HTML_RADIO);
        }
        // 类型&性别字段设置下拉框
        else if (StringUtils.endsWithIgnoreCase(columnName, "type")
                || StringUtils.endsWithIgnoreCase(columnName, "sex"))
        else if (StrUtil.endWithIgnoreCase(columnName, "type")
                || StrUtil.endWithIgnoreCase(columnName, "sex"))
        {
            column.setHtmlType(GenConstants.HTML_SELECT);
        }
        // 文件字段设置上传控件
        else if (StringUtils.endsWithIgnoreCase(columnName, "image"))
        // 图片字段设置图片上传控件
        else if (StrUtil.endWithIgnoreCase(columnName, "image"))
        {
            column.setHtmlType(GenConstants.HTML_UPLOAD_IMAGE);
            column.setHtmlType(GenConstants.HTML_IMAGE_UPLOAD);
        }
        // 文件字段设置文件上传控件
        else if (StrUtil.endWithIgnoreCase(columnName, "file"))
        {
            column.setHtmlType(GenConstants.HTML_FILE_UPLOAD);
        }
        // 内容字段设置富文本控件
        else if (StringUtils.endsWithIgnoreCase(columnName, "content"))
        else if (StrUtil.endWithIgnoreCase(columnName, "content"))
        {
            column.setHtmlType(GenConstants.HTML_EDITOR);
        }
@@ -126,7 +132,7 @@
    /**
     * 校验数组是否包含指定值
     *
     *
     * @param arr 数组
     * @param targetValue 值
     * @return 是否包含
@@ -138,7 +144,7 @@
    /**
     * 获取模块名
     *
     *
     * @param packageName 包名
     * @return 模块名
     */
@@ -146,13 +152,13 @@
    {
        int lastIndex = packageName.lastIndexOf(".");
        int nameLength = packageName.length();
        String moduleName = StringUtils.substring(packageName, lastIndex + 1, nameLength);
        String moduleName = StrUtil.sub(packageName, lastIndex + 1, nameLength);
        return moduleName;
    }
    /**
     * 获取业务名
     *
     *
     * @param tableName 表名
     * @return 业务名
     */
@@ -160,13 +166,13 @@
    {
        int lastIndex = tableName.lastIndexOf("_");
        int nameLength = tableName.length();
        String businessName = StringUtils.substring(tableName, lastIndex + 1, nameLength);
        String businessName = StrUtil.sub(tableName, lastIndex + 1, nameLength);
        return businessName;
    }
    /**
     * 表名转换成Java类名
     *
     *
     * @param tableName 表名称
     * @return 类名
     */
@@ -174,17 +180,17 @@
    {
        boolean autoRemovePre = GenConfig.getAutoRemovePre();
        String tablePrefix = GenConfig.getTablePrefix();
        if (autoRemovePre && StringUtils.isNotEmpty(tablePrefix))
        if (autoRemovePre && StrUtil.isNotEmpty(tablePrefix))
        {
            String[] searchList = StringUtils.split(tablePrefix, ",");
            String[] searchList = StrUtil.splitToArray(tablePrefix, ",");
            tableName = replaceFirst(tableName, searchList);
        }
        return StringUtils.convertToCamelCase(tableName);
        return StrUtil.upperFirst(StrUtil.toCamelCase(tableName));
    }
    /**
     * 批量替换前缀
     *
     *
     * @param replacementm 替换值
     * @param searchList 替换列表
     * @return
@@ -205,7 +211,7 @@
    /**
     * 关键字替换
     *
     *
     * @param text 需要被替换的名字
     * @return 替换后的名字
     */
@@ -216,15 +222,15 @@
    /**
     * 获取数据库类型字段
     *
     *
     * @param columnType 列类型
     * @return 截取后的列类型
     */
    public static String getDbType(String columnType)
    {
        if (StringUtils.indexOf(columnType, "(") > 0)
        if (StrUtil.indexOf(columnType, '(') > 0)
        {
            return StringUtils.substringBefore(columnType, "(");
            return StrUtil.subBefore(columnType, "(",false);
        }
        else
        {
@@ -234,15 +240,15 @@
    /**
     * 获取字段长度
     *
     *
     * @param columnType 列类型
     * @return 截取后的列类型
     */
    public static Integer getColumnLength(String columnType)
    {
        if (StringUtils.indexOf(columnType, "(") > 0)
        if (StrUtil.indexOf(columnType, '(') > 0)
        {
            String length = StringUtils.substringBetween(columnType, "(", ")");
            String length = StrUtil.subBetween(columnType, "(", ")");
            return Integer.valueOf(length);
        }
        else