|  |  |  | 
|---|
|  |  |  | import com.moral.api.config.mybatis.wrapper.NullFilterWrapper; | 
|---|
|  |  |  | import com.moral.api.entity.Group; | 
|---|
|  |  |  | import com.moral.api.entity.Menu; | 
|---|
|  |  |  | import com.moral.api.mapper.GroupMenuMapper; | 
|---|
|  |  |  | import com.moral.api.mapper.MenuMapper; | 
|---|
|  |  |  | import com.moral.api.pojo.dto.menu.MenuDTO; | 
|---|
|  |  |  | import com.moral.api.pojo.dto.menu.MenuQueryDTO; | 
|---|
|  |  |  | 
|---|
|  |  |  | MenuMapper menuMapper; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | GroupService groupService; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | GroupMenuMapper groupMenuMapper; | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public MenuQueryDTO queryAllMenus() { | 
|---|
|  |  |  | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.ORGANIZATION_USER_NOT_EXIST.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //根据角色查询对应的所有菜单 | 
|---|
|  |  |  | List<Menu> allMenus = menuMapper.getMenusByGroupId(group.getId()); | 
|---|
|  |  |  | //将菜单转换为父子结构 | 
|---|
|  |  |  | combinationParentChildrenMenus(allMenus); | 
|---|
|  |  |  | //删除所有菜单只保留最底层菜单 | 
|---|
|  |  |  | List<Menu> menus = new ArrayList<>(); | 
|---|
|  |  |  | for (Menu menu : allMenus) { | 
|---|
|  |  |  | getNotParentMenu(menu, menus); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //根据角色查询拥有的所有菜单 | 
|---|
|  |  |  | List<Menu> ownMenus = menuMapper.getMenusByGroupId(group.getId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /*判断每个菜单是否有子菜单,如果有则不传递 | 
|---|
|  |  |  | * 前端无法解决父菜单选中,子菜单没有全部选中回显问题。由后端处理的代码。*/ | 
|---|
|  |  |  | removeMenuWithChildren(ownMenus); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //封装返回结果 | 
|---|
|  |  |  | List<MenuDTO> dtos = new ArrayList<>(); | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | for (Menu menu : ownMenus) { | 
|---|
|  |  |  | MenuDTO menuDTO = new MenuDTO(); | 
|---|
|  |  |  | menuDTO.setMenu(menu); | 
|---|
|  |  |  | dtos.add(menuDTO); | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | @Transactional | 
|---|
|  |  |  | public MenuDTO deleteMenu(MenuDeleteForm form) { | 
|---|
|  |  |  | //创建返回对象 | 
|---|
|  |  |  | MenuDTO dto = new MenuDTO(); | 
|---|
|  |  |  | 
|---|
|  |  |  | //删除菜单 | 
|---|
|  |  |  | existMenu.setIsDelete(Constants.DELETE); | 
|---|
|  |  |  | menuMapper.updateById(existMenu); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //删除角色菜单关系表中对应的菜单 | 
|---|
|  |  |  | UpdateWrapper deleteGroupMenuWrapper = new UpdateWrapper(); | 
|---|
|  |  |  | List<Integer> allMenuIds = new ArrayList<>(); | 
|---|
|  |  |  | allMenuIds.add(id); | 
|---|
|  |  |  | if (!ObjectUtils.isEmpty(childrenIds)) { | 
|---|
|  |  |  | allMenuIds.addAll(childrenIds); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | deleteGroupMenuWrapper.in("menu_id", allMenuIds); | 
|---|
|  |  |  | deleteGroupMenuWrapper.set("is_delete", Constants.DELETE); | 
|---|
|  |  |  | groupMenuMapper.update(null, deleteGroupMenuWrapper); | 
|---|
|  |  |  | //封装返回对象 | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 将传来的菜单集合封装成父子结构的集合 | 
|---|
|  |  |  | * @Param: [menus] | 
|---|
|  |  |  | * @return: java.util.List<com.moral.api.entity.Menu> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/5/13 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | public void combinationParentChildrenMenus(List<Menu> menus){ | 
|---|
|  |  |  | * @Description: 将传来的菜单集合封装成父子结构的集合 | 
|---|
|  |  |  | * @Param: [menus] | 
|---|
|  |  |  | * @return: java.util.List<com.moral.api.entity.Menu> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/5/13 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | public void combinationParentChildrenMenus(List<Menu> menus) { | 
|---|
|  |  |  | //组合menu父子结构 | 
|---|
|  |  |  | Map<Integer, Menu> menuMap = new HashMap<>(); | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 获取menu菜单中的最底层菜单添加到menus中 | 
|---|
|  |  |  | * @Param: [menu, menus] | 
|---|
|  |  |  | * @return: void | 
|---|
|  |  |  | * @Description: 移除含有子菜单的菜单 | 
|---|
|  |  |  | * @Param: [menus] | 
|---|
|  |  |  | * @return: java.util.List<com.moral.api.entity.Menu> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/5/13 | 
|---|
|  |  |  | * @Date: 2021/5/28 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private void getNotParentMenu(Menu menu, List<Menu> menus) { | 
|---|
|  |  |  | if (ObjectUtils.isEmpty(menu.getChildren())) { | 
|---|
|  |  |  | menus.add(menu); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | List<Menu> children = menu.getChildren(); | 
|---|
|  |  |  | for (Menu child : children) { | 
|---|
|  |  |  | getNotParentMenu(child, menus); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | private void removeMenuWithChildren(List<Menu> menus) { | 
|---|
|  |  |  | //查询所有菜单 | 
|---|
|  |  |  | QueryWrapper wrapper = new QueryWrapper(); | 
|---|
|  |  |  | wrapper.eq("is_delete", Constants.NOT_DELETE); | 
|---|
|  |  |  | List<Menu> allMenus = menuMapper.selectList(wrapper); | 
|---|
|  |  |  | //组合menu父子结构 | 
|---|
|  |  |  | Map<Integer, Menu> menuMap = new HashMap<>(); | 
|---|
|  |  |  | for (Menu menu : allMenus) { | 
|---|
|  |  |  | menuMap.put(menu.getId(), menu); | 
|---|
|  |  |  | menu.setChildren(new ArrayList<>());//初始化集合 | 
|---|
|  |  |  | } | 
|---|
|  |  |  | for (Menu menu : allMenus) { | 
|---|
|  |  |  | putMenuToParentMenu(menuMap, menu); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //移除含有子菜单的菜单 | 
|---|
|  |  |  | menus.removeIf(new Predicate<Menu>() { | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public boolean test(Menu menu) { | 
|---|
|  |  |  | if (ObjectUtils.isEmpty(menuMap.get(menu.getId()).getChildren())) | 
|---|
|  |  |  | return false; | 
|---|
|  |  |  | return true; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|