From 7d64716cb0d406f21cde03976fd273bd07fdc06f Mon Sep 17 00:00:00 2001
From: whycxzp <perryhsu@163.com>
Date: 星期二, 28 九月 2021 11:44:28 +0800
Subject: [PATCH] 更新用户-角色-菜单-权限

---
 src/main/java/com/whyc/service/RoleMenuService.java |   39 ++++++++++++++++++++++++++++++++++++---
 1 files changed, 36 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/whyc/service/RoleMenuService.java b/src/main/java/com/whyc/service/RoleMenuService.java
index afc7b5d..c25a8ec 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;
 
 /**
  * 瑙掕壊-鑿滃崟
@@ -16,7 +21,35 @@
     @Resource
     private RoleMenuMapper mapper;
 
-    public List<Menu> getUserMenu(Integer id) {
-        return mapper.getUserMenu(id);
+    public List<Menu> getMenuByRoleId(Integer roleId,Integer status) {
+
+        List<Menu> menuList = mapper.getMenuByRoleId(roleId,status);
+
+        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;
     }
+
+    public boolean updateBatch(List<RoleMenu> roleMenus){
+        return mapper.updateBatch(roleMenus);
+    }
+
 }

--
Gitblit v1.9.1