|
@@ -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()))
|
|
|
+ .findFirst();
|
|
|
+ dto.setManageMenuId(manageMenu.map(MenuDO::getId).orElse(null));
|
|
|
|
|
|
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) {
|
|
|
+
|
|
|
+ Optional<MenuDO> foundMenu = menuList.stream()
|
|
|
+ .filter(menu -> "企业管理".equals(menu.getName()))
|
|
|
+ .findFirst();
|
|
|
+
|
|
|
+
|
|
|
+ return String.valueOf(foundMenu.map(MenuDO::getId).orElse(null));
|
|
|
+ }
|
|
|
+
|
|
|
}
|