|
@@ -39,6 +39,7 @@ import javax.servlet.http.HttpServletRequest;
|
|
|
import javax.validation.Valid;
|
|
|
import java.util.Collections;
|
|
|
import java.util.List;
|
|
|
+import java.util.Optional;
|
|
|
import java.util.Set;
|
|
|
|
|
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
|
@@ -148,7 +149,10 @@ public class AuthController {
|
|
|
Set<Long> menuIds = permissionService.getRoleMenuListByRoleId(convertSet(roles, RoleDO::getId));
|
|
|
List<MenuDO> menuList = menuService.getMenuList(menuIds);
|
|
|
menuList.removeIf(menu -> !CommonStatusEnum.ENABLE.getStatus().equals(menu.getStatus())); // 移除禁用的菜单
|
|
|
-
|
|
|
+ Optional<MenuDO> manageMenu = menuList.stream()
|
|
|
+ .filter(menu -> "企业管理".equals(menu.getName())) // 查找name为“企业管理”的MenuDO
|
|
|
+ .findFirst(); // 返回第一个找到的MenuDO(如果有的话)
|
|
|
+ dto.setManageMenuId(manageMenu.map(MenuDO::getId).orElse(null));
|
|
|
// 2. 拼接结果返回
|
|
|
return success(AuthConvert.INSTANCE.convert(dto, roles, menuList));
|
|
|
}
|
|
@@ -214,4 +218,14 @@ public class AuthController {
|
|
|
return success(authService.getCurrentTenantIdByUsername(username));
|
|
|
}
|
|
|
|
|
|
+ private String getEnterpriseManagementMenuId(List<MenuDO> menuList) {
|
|
|
+ // 使用Java 8的Stream API来找到第一个匹配的MenuDO
|
|
|
+ Optional<MenuDO> foundMenu = menuList.stream()
|
|
|
+ .filter(menu -> "企业管理".equals(menu.getName())) // 查找name为“企业管理”的MenuDO
|
|
|
+ .findFirst(); // 返回第一个找到的MenuDO(如果有的话)
|
|
|
+
|
|
|
+ // 如果找到了,就返回它的id,否则返回null
|
|
|
+ return String.valueOf(foundMenu.map(MenuDO::getId).orElse(null));
|
|
|
+ }
|
|
|
+
|
|
|
}
|