优化代码生成模板

RuoYi

2021-07-30 e46007c15f1e8bbf80e4002ae2b31aeb604a1e9f

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

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

@@ -62,6 +62,10 @@
        if (GenConstants.TPL_TREE.equals(tplCategory))

        {

            setTreeVelocityContext(velocityContext, genTable);

        }

        if (GenConstants.TPL_SUB.equals(tplCategory))

        {

            setSubVelocityContext(velocityContext, genTable);

        }

        return velocityContext;

    }

@@ -96,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()));

    }

    /**

     * 获取模板信息

     * 

@@ -119,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;

    }

@@ -146,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"))

        {

@@ -202,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()))

@@ -243,7 +279,8 @@
     */

    public static String getParentMenuId(JSONObject paramsObj)

    {

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

        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);

        }