代码优化

稚屿

2022-02-09 c99eb98001e22f671ce514268231edcd27fc73ef

 ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java

@@ -3,6 +3,7 @@
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.ruoyi.common.constant.GenConstants;

@@ -58,6 +59,7 @@
        velocityContext.put("permissionPrefix", getPermissionPrefix(moduleName, businessName));

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

        velocityContext.put("table", genTable);

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

        setMenuVelocityContext(velocityContext, genTable);

        if (GenConstants.TPL_TREE.equals(tplCategory))

        {

@@ -226,8 +228,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);

    }



    /**

@@ -261,6 +262,28 @@
    }



    /**

     * 根据列类型获取字典组

     * 
     * @param genTable 业务表对象

     * @return 返回字典组

     */

    public static String getDicts(GenTable genTable)

    {

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

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

        for (GenTableColumn column : columns)

        {

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

                    column.getHtmlType(),

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

            {

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

            }

        }

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

    }



    /**

     * 获取权限前缀

     *

     * @param moduleName 模块名称

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

        }