代码生成支持选择前端模板类型

RuoYi

2023-11-30 370a299d05029e085be9e259b103db419144cf87

 src/main/java/com/ruoyi/project/tool/gen/util/VelocityUtils.java

@@ -5,7 +5,8 @@
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.constant.GenConstants;

import com.ruoyi.common.utils.DateUtils;

import com.ruoyi.common.utils.StringUtils;

@@ -75,7 +76,7 @@
    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);

    }

@@ -83,7 +84,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);

@@ -122,11 +123,17 @@


    /**

     * 获取模板信息

     *

     * @param tplCategory 生成的模板

     * @param tplWebType 前端类型

     * @return 模板列表

     */

    public static List<String> getTemplateList(String tplCategory)

    public static List<String> getTemplateList(String tplCategory, String tplWebType)

    {

        String useWebType = "vm/vue";

        if ("element-plus".equals(tplWebType))

        {

            useWebType = "vm/vue/v3";

        }

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

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

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

@@ -138,15 +145,15 @@
        templates.add("vm/js/api.js.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");

        }

        return templates;

@@ -271,6 +278,23 @@
    {

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

        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(

@@ -280,7 +304,6 @@
                dicts.add("'" + column.getDictType() + "'");

            }

        }

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

    }



    /**

@@ -365,7 +388,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())