huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2024-11-23 07:41:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【修复】SYSTEM：获取菜单精简信息列表接口没有排除父 ID 非 0 的节点

Browse Source
This commit is contained in:
YunaiV 2024-06-18 09:20:49 +08:00
parent cb73e7feb0
commit 3da2449a15
2 changed files with 20 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/permission/MenuService.java
View File

@ -56,7 +56,7 @@ public interface MenuService {
* 过滤掉关闭的菜单及其子菜单 * 过滤掉关闭的菜单及其子菜单
* *
* @param list 菜单列表 * @param list 菜单列表
* @return List<MenuDO> 过滤后的菜单列表 * @return 过滤后的菜单列表
*/ */
List<MenuDO> filterDisableMenus(List<MenuDO> list); List<MenuDO> filterDisableMenus(List<MenuDO> list);

48
yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/permission/MenuServiceImpl.java
View File

@ -1,6 +1,7 @@
package cn.iocoder.yudao.module.system.service.permission; package cn.iocoder.yudao.module.system.service.permission;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils; import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuListReqVO; import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuListReqVO;
@ -14,7 +15,6 @@ import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.Lists; import com.google.common.collect.Lists;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.cache.annotation.CacheEvict; import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable; import org.springframework.cache.annotation.Cacheable;
import org.springframework.context.annotation.Lazy; import org.springframework.context.annotation.Lazy;
@ -22,11 +22,10 @@ import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
import java.util.*; import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList; import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMap;
import static cn.iocoder.yudao.module.system.dal.dataobject.permission.MenuDO.ID_ROOT; import static cn.iocoder.yudao.module.system.dal.dataobject.permission.MenuDO.ID_ROOT;
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*; import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*;
@ -116,56 +115,47 @@ public class MenuServiceImpl implements MenuService {
return menus; return menus;
} }
/**
* 过滤关闭的菜单节点及其子节点
*
* @param menuList 所有菜单列表
* @return List<MenuDO> 过滤后的菜单列表
*/
@Override @Override
public List<MenuDO> filterDisableMenus(List<MenuDO> menuList) { public List<MenuDO> filterDisableMenus(List<MenuDO> menuList) {
if(CollectionUtils.isEmpty(menuList)){ if (CollUtil.isEmpty(menuList)){
return Collections.emptyList(); return Collections.emptyList();
} }
Map<Long, MenuDO> menuMap = convertMap(menuList, MenuDO::getId);
Map<Long, MenuDO> menuMap = new HashMap<>(); // 遍历 menu 菜单，查找不是禁用的菜单，添加到 enabledMenus 结果
for (MenuDO menuDO : menuList) {
menuMap.put(menuDO.getId(),menuDO);
}
// 存下递归搜索过被禁用的菜单，防止重复的搜索
Set<Long> disabledMenuIds = new HashSet<>();
List<MenuDO> enabledMenus = new ArrayList<>(); List<MenuDO> enabledMenus = new ArrayList<>();
Set<Long> disabledMenuCache = new HashSet<>(); // 存下递归搜索过被禁用的菜单，防止重复的搜索
for (MenuDO menu : menuList) { for (MenuDO menu : menuList) {
if (!isMenuDisabled(menu, menuMap, disabledMenuIds)) { if (isMenuDisabled(menu, menuMap, disabledMenuCache)) {
enabledMenus.add(menu); continue;
} }
enabledMenus.add(menu);
} }
return enabledMenus; return enabledMenus;
} }
private boolean isMenuDisabled(MenuDO node, Map<Long, MenuDO> menuMap, Set<Long> disabledMenuIds) { private boolean isMenuDisabled(MenuDO node, Map<Long, MenuDO> menuMap, Set<Long> disabledMenuCache) {
if (disabledMenuIds.contains(node.getId())) { // 如果已经判定是禁用的节点，直接结束
if (disabledMenuCache.contains(node.getId())) {
return true; return true;
} }
// 1. 遍历到 parentId 为根节点，则无需判断
Long parentId = node.getParentId(); Long parentId = node.getParentId();
if (parentId == 0) { if (ObjUtil.equal(parentId, ID_ROOT)) {
if (!node.getStatus().equals(CommonStatusEnum.ENABLE.getStatus())) { if (CommonStatusEnum.isDisable(node.getStatus())) {
disabledMenuIds.add(node.getId()); disabledMenuCache.add(node.getId());
return true; return true;
} }
return false; return false;
} }
// 2. 继续遍历 parent 节点
MenuDO parent = menuMap.get(parentId); MenuDO parent = menuMap.get(parentId);
if (parent == null || isMenuDisabled(parent, menuMap, disabledMenuIds)) { if (parent == null || isMenuDisabled(parent, menuMap, disabledMenuCache)) {
disabledMenuIds.add(node.getId()); disabledMenuCache.add(node.getId());
return true; return true;
} }
return false; return false;
} }