疯狂的狮子li
2021-09-18 2f18d0d7094b10e169b77a0ea741bd101dbbf1f8
ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java
@@ -16,7 +16,7 @@
/**
 * 模板处理工具类
 *
 *
 * @author ruoyi
 */
public class VelocityUtils
@@ -61,6 +61,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))
        {
@@ -245,7 +246,7 @@
    /**
     * 根据列类型获取导入包
     *
     *
     * @param genTable 业务表对象
     * @return 返回需要导入的包列表
     */
@@ -274,6 +275,27 @@
    }
    /**
     * 根据列类型获取字典组
     *
     * @param genTable 业务表对象
     * @return 返回字典组
     */
    public static String getDicts(GenTable genTable)
    {
        List<GenTableColumn> columns = genTable.getColumns();
        List<String> dicts = new ArrayList<String>();
        for (GenTableColumn column : columns)
        {
            if (!column.isSuperColumn() && StringUtils.isNotEmpty(column.getDictType()) && StringUtils.equalsAny(
                    column.getHtmlType(), new String[] { GenConstants.HTML_SELECT, GenConstants.HTML_RADIO }))
            {
                dicts.add("'" + column.getDictType() + "'");
            }
        }
        return StringUtils.join(dicts, ", ");
    }
    /**
     * 获取权限前缀
     *
     * @param moduleName 模块名称