优化代码生成模板

RuoYi

2021-07-30 e46007c15f1e8bbf80e4002ae2b31aeb604a1e9f

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

@@ -11,6 +11,11 @@
import com.ruoyi.project.tool.gen.domain.GenTable;

import com.ruoyi.project.tool.gen.domain.GenTableColumn;



/**

 * 模板处理工具类

 * 
 * @author ruoyi

 */

public class VelocityUtils

{

    /** 项目空间路径 */

@@ -18,6 +23,9 @@


    /** mybatis空间路径 */

    private static final String MYBATIS_PATH = "main/resources/mybatis";



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

    private static final String DEFAULT_PARENT_MENU_ID = "3";



    /**

     * 设置模板变量信息

@@ -46,15 +54,28 @@
        velocityContext.put("author", genTable.getFunctionAuthor());

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

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

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

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

        velocityContext.put("permissionPrefix", getPermissionPrefix(moduleName, businessName));

        velocityContext.put("columns", genTable.getColumns());

        velocityContext.put("table", genTable);

        setMenuVelocityContext(velocityContext, genTable);

        if (GenConstants.TPL_TREE.equals(tplCategory))

        {

            setTreeVelocityContext(velocityContext, genTable);

        }

        if (GenConstants.TPL_SUB.equals(tplCategory))

        {

            setSubVelocityContext(velocityContext, genTable);

        }

        return velocityContext;

    }



    public static void setMenuVelocityContext(VelocityContext context, GenTable genTable)

    {

        String options = genTable.getOptions();

        JSONObject paramsObj = JSONObject.parseObject(options);

        String parentMenuId = getParentMenuId(paramsObj);

        context.put("parentMenuId", parentMenuId);

    }



    public static void setTreeVelocityContext(VelocityContext context, GenTable genTable)

@@ -79,6 +100,23 @@
        }

    }



    public static void setSubVelocityContext(VelocityContext context, GenTable genTable)

    {

        GenTable subTable = genTable.getSubTable();

        String subTableName = genTable.getSubTableName();

        String subTableFkName = genTable.getSubTableFkName();

        String subClassName = genTable.getSubTable().getClassName();

        String subTableFkClassName = StringUtils.convertToCamelCase(subTableFkName);



        context.put("subTable", subTable);

        context.put("subTableName", subTableName);

        context.put("subTableFkName", subTableFkName);

        context.put("subTableFkClassName", subTableFkClassName);

        context.put("subTableFkclassName", StringUtils.uncapitalize(subTableFkClassName));

        context.put("subClassName", subClassName);

        context.put("subclassName", StringUtils.uncapitalize(subClassName));

        context.put("subImportList", getImportList(genTable.getSubTable()));

    }

    /**

     * 获取模板信息

     * 

@@ -102,6 +140,11 @@
        else if (GenConstants.TPL_TREE.equals(tplCategory))

        {

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

        }

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

        {

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

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

        }

        return templates;

    }

@@ -129,6 +172,10 @@
        if (template.contains("domain.java.vm"))

        {

            fileName = StringUtils.format("{}/domain/{}.java", javaPath, className);

        }

        if (template.contains("sub-domain.java.vm") && StringUtils.equals(GenConstants.TPL_SUB, genTable.getTplCategory()))

        {

            fileName = StringUtils.format("{}/domain/{}.java", javaPath, genTable.getSubTable().getClassName());

        }

        else if (template.contains("mapper.java.vm"))

        {

@@ -185,12 +232,18 @@
    /**

     * 根据列类型获取导入包

     * 

     * @param column 列集合

     * @param genTable 业务表对象

     * @return 返回需要导入的包列表

     */

    public static HashSet<String> getImportList(List<GenTableColumn> columns)

    public static HashSet<String> getImportList(GenTable genTable)

    {

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

        GenTable subGenTable = genTable.getSubTable();

        HashSet<String> importList = new HashSet<String>();

        if (StringUtils.isNotNull(subGenTable))

        {

            importList.add("java.util.List");

        }

        for (GenTableColumn column : columns)

        {

            if (!column.isSuperColumn() && GenConstants.TYPE_DATE.equals(column.getJavaType()))

@@ -216,7 +269,22 @@
    public static String getPermissionPrefix(String moduleName, String businessName)

    {

        return StringUtils.format("{}:{}", moduleName, businessName);

    }



    /**

     * 获取上级菜单ID字段

     * 
     * @param options 生成其他选项

     * @return 上级菜单ID字段

     */

    public static String getParentMenuId(JSONObject paramsObj)

    {

        if (StringUtils.isNotEmpty(paramsObj) && paramsObj.containsKey(GenConstants.PARENT_MENU_ID)

                && StringUtils.isNotEmpty(paramsObj.getString(GenConstants.PARENT_MENU_ID)))

        {

            return paramsObj.getString(GenConstants.PARENT_MENU_ID);

        }

        return DEFAULT_PARENT_MENU_ID;

    }



    /**

@@ -231,7 +299,7 @@
        {

            return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_CODE));

        }

        return "";

        return StringUtils.EMPTY;

    }



    /**

@@ -246,7 +314,7 @@
        {

            return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_PARENT_CODE));

        }

        return "";

        return StringUtils.EMPTY;

    }



    /**

@@ -261,7 +329,7 @@
        {

            return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_NAME));

        }

        return "";

        return StringUtils.EMPTY;

    }



    /**