修改ruoyi-改为sgd

fuyongxin

2026-06-03 1fd34350ea68fe820c4b62881b13c9f01082e6c0

 ruoyi-modules/ruoyi-gen/src/main/java/com/ruoyi/gen/util/VelocityUtils.java

@@ -3,8 +3,10 @@
import java.util.ArrayList;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import org.apache.velocity.VelocityContext;

import com.alibaba.fastjson.JSONObject;

import com.alibaba.fastjson2.JSON;

import com.alibaba.fastjson2.JSONObject;

import com.ruoyi.common.core.constant.GenConstants;

import com.ruoyi.common.core.utils.DateUtils;

import com.ruoyi.common.core.utils.StringUtils;

@@ -12,7 +14,7 @@
import com.ruoyi.gen.domain.GenTableColumn;



/**

 * 模板工具类

 * 模板处理工具类

 * 

 * @author ruoyi

 */

@@ -26,6 +28,12 @@


    /** 默认上级菜单，系统工具 */

    private static final String DEFAULT_PARENT_MENU_ID = "3";



    /** Vue3 Element Plus 模版 */

    private static final String ELEMENT_PLUS = "element-plus";



    /** Vue3 Element Plus TypeScript 模版 */

    private static final String ELEMENT_PLUS_TYPESSRIPT = "element-plus-typescript";



    /**

     * 设置模板变量信息

@@ -52,6 +60,7 @@
        velocityContext.put("basePackage", getPackagePrefix(packageName));

        velocityContext.put("packageName", packageName);

        velocityContext.put("author", genTable.getFunctionAuthor());

        velocityContext.put("colSpan", getColSpan(genTable.getFormColNum()));

        velocityContext.put("datetime", DateUtils.getDate());

        velocityContext.put("pkColumn", genTable.getPkColumn());

        velocityContext.put("importList", getImportList(genTable));

@@ -59,6 +68,7 @@
        velocityContext.put("columns", genTable.getColumns());

        velocityContext.put("table", genTable);

        velocityContext.put("dicts", getDicts(genTable));

        setExtensionsContext(velocityContext, genTable.getOptions());

        setMenuVelocityContext(velocityContext, genTable);

        if (GenConstants.TPL_TREE.equals(tplCategory))

        {

@@ -71,10 +81,17 @@
        return velocityContext;

    }



    public static void setExtensionsContext(VelocityContext context, String options)

    {

        JSONObject paramsObj = JSONObject.parseObject(options);

        boolean genView = genView(paramsObj);

        context.put("genView", genView);

    }



    public static void setMenuVelocityContext(VelocityContext context, GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        JSONObject paramsObj = JSON.parseObject(options);

        String parentMenuId = getParentMenuId(paramsObj);

        context.put("parentMenuId", parentMenuId);

    }

@@ -82,7 +99,7 @@
    public static void setTreeVelocityContext(VelocityContext context, GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        JSONObject paramsObj = JSON.parseObject(options);

        String treeCode = getTreecode(paramsObj);

        String treeParentCode = getTreeParentCode(paramsObj);

        String treeName = getTreeName(paramsObj);

@@ -121,11 +138,27 @@


    /**

     * 获取模板信息

     *

     * @param tplCategory 生成的模板

     * @param tplWebType 前端类型

     * @return 模板列表

     */

    public static List<String> getTemplateList(String tplCategory)

    public static List<String> getTemplateList(GenTable table)

    {

        String tplWebType = table.getTplWebType();

        String tplCategory = table.getTplCategory();

        JSONObject paramsObj = JSONObject.parseObject(table.getOptions());

        boolean isView = genView(paramsObj);

        String useWebType = "vm/vue";

        String apiTemplate = "vm/js/api.js.vm";

        if (StringUtils.equals(ELEMENT_PLUS, tplWebType))

        {

            useWebType = "vm/vue/v3";

        }

        else if (StringUtils.equals(ELEMENT_PLUS_TYPESSRIPT, tplWebType))

        {

            useWebType = "vm/vue/v3ts";

            apiTemplate = "vm/ts/api.ts.vm";

        }

        List<String> templates = new ArrayList<String>();

        templates.add("vm/java/domain.java.vm");

        templates.add("vm/java/mapper.java.vm");

@@ -134,19 +167,28 @@
        templates.add("vm/java/controller.java.vm");

        templates.add("vm/xml/mapper.xml.vm");

        templates.add("vm/sql/sql.vm");

        templates.add("vm/js/api.js.vm");

        templates.add(apiTemplate);

        if (StringUtils.equals(ELEMENT_PLUS_TYPESSRIPT, tplWebType))

        {

            templates.add("vm/ts/type.ts.vm");

            templates.add("vm/ts/index.ts.vm");

        }

        if (GenConstants.TPL_CRUD.equals(tplCategory))

        {

            templates.add("vm/vue/index.vue.vm");

            templates.add(useWebType + "/index.vue.vm");

        }

        else if (GenConstants.TPL_TREE.equals(tplCategory))

        {

            templates.add("vm/vue/index-tree.vue.vm");

            templates.add(useWebType + "/index-tree.vue.vm");

        }

        else if (GenConstants.TPL_SUB.equals(tplCategory))

        {

            templates.add("vm/vue/index.vue.vm");

            templates.add(useWebType + "/index.vue.vm");

            templates.add("vm/java/sub-domain.java.vm");

        }

        if (isView)

        {

            templates.add(useWebType + "/view.vue.vm");

        }

        return templates;

    }

@@ -207,6 +249,18 @@
        {

            fileName = StringUtils.format("{}/api/{}/{}.js", vuePath, moduleName, businessName);

        }

        else if (template.contains("api.ts.vm"))

        {

            fileName = StringUtils.format("{}/api/{}/{}.ts", vuePath, moduleName, businessName);

        }

        else if (template.contains("type.ts.vm"))

        {

            fileName = StringUtils.format("{}/types/api/{}/{}.ts", vuePath, moduleName, businessName);

        }

        else if (template.contains("index.ts.vm"))

        {

            fileName = StringUtils.format("{}/types/api/index-bak.ts", vuePath);

        }

        else if (template.contains("index.vue.vm"))

        {

            fileName = StringUtils.format("{}/views/{}/{}/index.vue", vuePath, moduleName, businessName);

@@ -214,6 +268,10 @@
        else if (template.contains("index-tree.vue.vm"))

        {

            fileName = StringUtils.format("{}/views/{}/{}/index.vue", vuePath, moduleName, businessName);

        }

        else if (template.contains("view.vue.vm"))

        {

            fileName = StringUtils.format("{}/views/{}/{}/view.vue", vuePath, moduleName, businessName);

        }

        return fileName;

    }

@@ -227,8 +285,7 @@
    public static String getPackagePrefix(String packageName)

    {

        int lastIndex = packageName.lastIndexOf(".");

        String basePackage = StringUtils.substring(packageName, 0, lastIndex);

        return basePackage;

        return StringUtils.substring(packageName, 0, lastIndex);

    }



    /**

@@ -270,16 +327,33 @@
    public static String getDicts(GenTable genTable)

    {

        List<GenTableColumn> columns = genTable.getColumns();

        List<String> dicts = new ArrayList<String>();

        Set<String> dicts = new HashSet<String>();

        addDicts(dicts, columns);

        if (StringUtils.isNotNull(genTable.getSubTable()))

        {

            List<GenTableColumn> subColumns = genTable.getSubTable().getColumns();

            addDicts(dicts, subColumns);

        }

        return StringUtils.join(dicts, ", ");

    }



    /**

     * 添加字典列表

     * 
     * @param dicts 字典列表

     * @param columns 列集合

     */

    public static void addDicts(Set<String> dicts, List<GenTableColumn> columns)

    {

        for (GenTableColumn column : columns)

        {

            if (!column.isSuperColumn() && StringUtils.isNotEmpty(column.getDictType()) && StringUtils.equalsAny(

                    column.getHtmlType(), new String[] { GenConstants.HTML_SELECT, GenConstants.HTML_RADIO }))

                    column.getHtmlType(),

                    new String[] { GenConstants.HTML_SELECT, GenConstants.HTML_RADIO, GenConstants.HTML_CHECKBOX }))

            {

                dicts.add("'" + column.getDictType() + "'");

            }

        }

        return StringUtils.join(dicts, ", ");

    }



    /**

@@ -341,6 +415,21 @@
    }



    /**

     * 扩展功能/生成详情页

     * 
     * @param paramsObj 生成其他选项

     * @return 是否生成详细页

     */

    public static boolean genView(JSONObject paramsObj)

    {

        if (StringUtils.isNotNull(paramsObj) && paramsObj.containsKey(GenConstants.GEN_VIEW))

        {

            return paramsObj.getBoolean(GenConstants.GEN_VIEW);

        }

        return false;

    }



    /**

     * 获取树名称

     *

     * @param paramsObj 生成其他选项

@@ -364,7 +453,7 @@
    public static int getExpandColumn(GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        JSONObject paramsObj = JSON.parseObject(options);

        String treeName = paramsObj.getString(GenConstants.TREE_NAME);

        int num = 0;

        for (GenTableColumn column : genTable.getColumns())

@@ -381,4 +470,23 @@
        }

        return num;

    }



    /**

     * 获取表单 el-col span

     * 
     * @param formColNum 表单布局方式（1单列 2双列 3三列）

     * @return span 数值字符串

     */

    public static String getColSpan(int formColNum)

    {

        if (formColNum == 2)

        {

            return "12";

        }

        else if (formColNum == 3)

        {

            return "8";

        }

        return "24";

    }

}