diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/res/menu/MenuQueryRes.java b/src/main/java/com/cczsa/xinghe/codegen/entity/res/menu/MenuQueryRes.java index af92d31..67d583c 100644 --- a/src/main/java/com/cczsa/xinghe/codegen/entity/res/menu/MenuQueryRes.java +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/res/menu/MenuQueryRes.java @@ -49,6 +49,9 @@ public class MenuQueryRes implements Serializable { @Schema(description = "排序") private Integer sortOrder; + @Schema(description = "功能id列表") + private List funIdList; + @Schema(description = "子菜单") private List children; diff --git a/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java b/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java index 80bdfa5..63e53fd 100644 --- a/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java +++ b/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java @@ -59,6 +59,15 @@ public class MenuServiceImpl implements MenuService { .eq(MenuEntity::getParentId, req.getParentId(), req.getParentId() != null) .orderBy(MenuEntity::getSortOrder, false); List menuEntities = menuMapper.selectListByQueryAs(query, MenuQueryRes.class); + for (MenuQueryRes menuEntity : menuEntities) { + MenuBindFunDef menuBindFunDef = MenuBindFunDef.MENU_BIND_FUN_ENTITY; + QueryWrapper queryMenuBindFun = new QueryWrapper(); + queryMenuBindFun.select(menuBindFunDef.FUN_ID) + .from(menuBindFunDef) + .eq(MenuBindFunEntity::getMenuId, menuEntity.getId()); + List funIds = menuBindFunMapper.selectListByQueryAs(queryMenuBindFun, Long.class); + menuEntity.setFunIdList(funIds); + } // 构建树形结构 List treeList = buildMenuTree(menuEntities); return XResult.ok(treeList);