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