| | |
|---|
| | | import com.ruoyi.generator.domain.GenTable;
|
|---|
| | | import com.ruoyi.generator.domain.GenTableColumn;
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 代码生成模板处理
|
|---|
| | | * |
|---|
| | | * @author ruoyi
|
|---|
| | | */
|
|---|
| | | public class VelocityUtils
|
|---|
| | | {
|
|---|
| | | /** 项目空间路径 */
|
|---|
| | | private static final String PROJECT_PATH = "main/java";
|
|---|
| | |
|
|---|
| | | /** mybatis空间路径 */
|
|---|
| | | private static final String MYBATIS_PATH = "main/resources/mapper";
|
|---|
| | | private static final String MYBATIS_PATH = "main/resources/mybatis";
|
|---|
| | |
|
|---|
| | | /** 默认上级菜单，系统工具 */
|
|---|
| | | private static final String DEFAULT_PARENT_MENU_ID = "3";
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 设置模板变量信息
|
|---|
| | |
|---|
| | | 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);
|
|---|
| | | }
|
|---|
| | | 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)
|
|---|
| | |
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 获取上级菜单ID字段
|
|---|
| | | * |
|---|
| | | * @param options 生成其他选项
|
|---|
| | | * @return 上级菜单ID字段
|
|---|
| | | */
|
|---|
| | | public static String getParentMenuId(JSONObject paramsObj)
|
|---|
| | | {
|
|---|
| | | if (paramsObj.containsKey(GenConstants.PARENT_MENU_ID))
|
|---|
| | | {
|
|---|
| | | return paramsObj.getString(GenConstants.PARENT_MENU_ID);
|
|---|
| | | }
|
|---|
| | | return DEFAULT_PARENT_MENU_ID;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 获取树编码
|
|---|
| | | *
|
|---|
| | | * @param options 生成其他选项
|
|---|
| | |
|---|
| | | {
|
|---|
| | | return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_CODE));
|
|---|
| | | }
|
|---|
| | | return "";
|
|---|
| | | return StringUtils.EMPTY;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | |
|---|
| | | {
|
|---|
| | | return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_PARENT_CODE));
|
|---|
| | | }
|
|---|
| | | return "";
|
|---|
| | | return StringUtils.EMPTY;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | |
|---|
| | | {
|
|---|
| | | return StringUtils.toCamelCase(paramsObj.getString(GenConstants.TREE_NAME));
|
|---|
| | | }
|
|---|
| | | return "";
|
|---|
| | | return StringUtils.EMPTY;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
