lyhzzz 2 years ago
parent
commit
3db40b0a9f

+ 4 - 0
src/main/java/com/fdkankan/manage/controller/SysMenuController.java

@@ -76,6 +76,10 @@ public class SysMenuController extends BaseController{
     public ResultData getByRoleId(@RequestParam(required = false) Long roleId){
         return ResultData.ok(menuService.getByRoleId(null,roleId));
     }
+    @GetMapping("/getMenuIdsByRoleId")
+    public ResultData getMenuIdsByRoleId(@RequestParam(required = false) Long roleId){
+        return ResultData.ok(menuService.getMenuIdsByRoleId(roleId));
+    }
 
 }
 

+ 2 - 0
src/main/java/com/fdkankan/manage/service/ISysMenuService.java

@@ -32,4 +32,6 @@ public interface ISysMenuService extends IService<SysMenu> {
     List<SysMenu> getButtonByUserId(Long userId);
 
     HashMap<Long, SysMenu> getByIds(List<Long> menuIds);
+
+    List<Long> getMenuIdsByRoleId(Long roleId);
 }

+ 7 - 0
src/main/java/com/fdkankan/manage/service/impl/SysMenuServiceImpl.java

@@ -164,4 +164,11 @@ public class SysMenuServiceImpl extends ServiceImpl<ISysMenuMapper, SysMenu> imp
 
         return map;
     }
+
+    @Override
+    public List<Long> getMenuIdsByRoleId(Long roleId) {
+        LambdaQueryWrapper<SysRoleMenu> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(SysRoleMenu::getRoleId,roleId);
+        return roleMenuService.list(wrapper).stream().map(SysRoleMenu::getMenuId).collect(Collectors.toList());
+    }
 }