From b7226efbe9f5fdcfe33b152baf91eac6ab44b30a Mon Sep 17 00:00:00 2001
From: whycxzp <perryhsu@163.com>
Date: 星期五, 24 九月 2021 17:25:51 +0800
Subject: [PATCH] 更新角色,菜单接口
---
src/main/java/com/whyc/service/RoleMenuService.java | 34 +++++++++++++++++++++++++++++++++-
1 files changed, 33 insertions(+), 1 deletions(-)
diff --git a/src/main/java/com/whyc/service/RoleMenuService.java b/src/main/java/com/whyc/service/RoleMenuService.java
index afc7b5d..546e89a 100644
--- a/src/main/java/com/whyc/service/RoleMenuService.java
+++ b/src/main/java/com/whyc/service/RoleMenuService.java
@@ -1,11 +1,16 @@
package com.whyc.service;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.whyc.dto.Response;
import com.whyc.mapper.RoleMenuMapper;
import com.whyc.pojo.Menu;
+import com.whyc.pojo.RoleMenu;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
-import java.util.List;
+import java.util.*;
+import java.util.stream.Collectors;
/**
* 瑙掕壊-鑿滃崟
@@ -19,4 +24,31 @@
public List<Menu> getUserMenu(Integer id) {
return mapper.getUserMenu(id);
}
+
+ public List<Menu> getMenuByRoleId(Integer roleId) {
+
+ List<Menu> menuList = mapper.getMenuByRoleId(roleId);
+
+ for (int i = 0; i < menuList.size(); i++) {
+ Integer parentId = menuList.get(i).getParentId();
+ if(parentId !=0){
+ //灏嗚繖涓猰enu鎻掑叆鍒皃arent涓�
+ for (int j = 0; j < menuList.size(); j++) {
+ if(menuList.get(j).getId()==parentId){
+ List<Menu> menuListTemp = menuList.get(j).getMenuList();
+ if(menuListTemp==null) {
+ menuListTemp = new LinkedList<>();
+ }
+ menuListTemp.add(menuList.get(i));
+ menuList.get(j).setMenuList(menuListTemp);
+ menuList.remove(i);
+ i--;
+ break;
+ }
+ }
+ }
+
+ }
+ return menuList;
+ }
}
--
Gitblit v1.9.1