Selaa lähdekoodia

feat: 用户菜单权限增加菜单ID列表返回

chenjianhua 6 kuukautta sitten
vanhempi
commit
f31fe72f7d

+ 14 - 1
service-manage/src/main/java/com/simuwang/manage/dto/UserInfoVO.java

@@ -32,16 +32,29 @@ public class UserInfoVO {
      * 菜单树
      */
     private MenuTreeDTO menuTree;
+    /**
+     * 菜单权限ID集合
+     */
+    private List<Integer> menuIds;
 
     public UserInfoVO() {
     }
 
-    public UserInfoVO(Integer userId, String username, List<UserRoleDTO> roles, MenuTreeDTO menuTree) {
+    public UserInfoVO(Integer userId, String username, List<UserRoleDTO> roles, MenuTreeDTO menuTree,List<Integer> menuIds) {
         this();
         this.userId = userId;
         this.username = username;
         this.roles = roles;
         this.menuTree = menuTree;
+        this.menuIds = menuIds;
+    }
+
+    public List<Integer> getMenuIds() {
+        return menuIds;
+    }
+
+    public void setMenuIds(List<Integer> menuIds) {
+        this.menuIds = menuIds;
     }
 
     public Integer getUserId() {

+ 2 - 1
service-manage/src/main/java/com/simuwang/manage/service/LoginService.java

@@ -107,7 +107,8 @@ public class LoginService {
                 MenuTreeDTO::getChildren, MenuTreeDTO::setChildren, 0, MenuTreeDTO::getSort);
         MenuTreeDTO root = new MenuTreeDTO(0, "全部菜单", trees);
         // 返回的数据结构构建
-        UserInfoVO vo = new UserInfoVO(userId, username, roles, root);
+        List<Integer> menuIds = tempList.stream().map(e -> e.getId()).collect(Collectors.toList());
+        UserInfoVO vo = new UserInfoVO(userId, username, roles, root,menuIds);
         vo.setEmail(userInfo.getEmail());
         vo.setPhonenumber(userInfo.getPhonenumber());
         return vo;