From 5bfa6ac57681f0b07649c2f1a54d6afbd3dbc1ba Mon Sep 17 00:00:00 2001
From: nidapeng <jp@doumee.com>
Date: 星期三, 20 三月 2024 11:37:28 +0800
Subject: [PATCH] 提交一把

---
 server/service/src/main/java/com/doumee/core/model/LoginUserInfo.java |   11 ++++++++---
 1 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/server/service/src/main/java/com/doumee/core/model/LoginUserInfo.java b/server/service/src/main/java/com/doumee/core/model/LoginUserInfo.java
index 127f16e..e73d456 100644
--- a/server/service/src/main/java/com/doumee/core/model/LoginUserInfo.java
+++ b/server/service/src/main/java/com/doumee/core/model/LoginUserInfo.java
@@ -44,8 +44,9 @@
     private List<String> roles;
 
     private List<String> permissions;
+    private List<Integer> companyIdList;
 
-    public static LoginUserInfo from(SystemUser user, List<SystemRole> roles, List<SystemPermission> permissions) {
+    public static LoginUserInfo from(SystemUser user, List<SystemRole> roles, List<SystemPermission> permissions ) {
         if (user == null) {
             return null;
         }
@@ -60,11 +61,15 @@
         }
         loginUserInfo.setRoles(rs);
         // 璁剧疆鏉冮檺淇℃伅
+
         List<String> pms = new ArrayList<>();
-        for (SystemPermission permission : permissions) {
-            pms.add(permission.getCode());
+        if(permissions!=null){
+            for (SystemPermission permission : permissions) {
+                pms.add(permission.getCode());
+            }
         }
         loginUserInfo.setPermissions(pms);
+        loginUserInfo.setCompanyIdList(user.getCompanyIdList());
         return loginUserInfo;
     }
 }

--
Gitblit v1.9.3