From 5a4419d146e4ac48fd0e3ed89ab64e64cad192d1 Mon Sep 17 00:00:00 2001
From: lbq <1065079612@qq.com>
Date: 星期二, 10 二月 2026 10:54:59 +0800
Subject: [PATCH] 补充登录角色

---
 rsf-server/src/main/java/com/vincent/rsf/server/api/service/impl/MobileServiceImpl.java |   14 ++++++++++++--
 1 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/api/service/impl/MobileServiceImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/api/service/impl/MobileServiceImpl.java
index 99f0a4d..4fc2165 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/api/service/impl/MobileServiceImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/api/service/impl/MobileServiceImpl.java
@@ -129,6 +129,8 @@
     private WarehouseRoleMenuService warehouseRoleMenuService;
     @Autowired
     private BasContainerService basContainerService;
+    @Resource
+    private UserRoleService userRoleService;
 
     /**
      * @return
@@ -157,6 +159,14 @@
         if (!userService.comparePassword(user.getPassword(), param.getPassword())) {
             return R.parse(CodeRes.USER_10003);
         }
+
+        List<Role> roles = userRoleService.listByUserId(user.getId());
+        user.setRoles(roles);
+        Long[] roleIds = new Long[roles.size()];
+        for(int i = 0; i < roles.size(); i++) {
+            roleIds[i] = roles.get(i).getId();
+        }
+        user.setRoleIds(roleIds);
 
         String accessToken = JwtUtil.buildToken(new JwtSubject(param.getUsername(), user.getTenantId()),
                 configProperties.getTokenExpireTime(), configProperties.getTokenKey());
@@ -1026,7 +1036,7 @@
                         .map(WarehouseRoleMenu::getMenuId)
                         .collect(Collectors.toSet());
                 // 鑾峰彇 areaList 骞惰浆鎹负 Long 绫诲瀷鐨� Set
-                List<Integer> areaList = container.getAreas();
+                List<Integer> areaList = container.getAreasIds();
                 Set<Long> areaSet = new HashSet<>();
                 if (areaList != null) {
                     areaList.forEach(area -> areaSet.add(area.longValue()));
@@ -1066,7 +1076,7 @@
                         .map(WarehouseRoleMenu::getMenuId)
                         .collect(Collectors.toSet());
                 // 鑾峰彇 areaList 骞惰浆鎹负 Long 绫诲瀷鐨� Set
-                List<Integer> areaList = container.getAreas();
+                List<Integer> areaList = container.getAreasIds();
                 Set<Long> areaSet = new HashSet<>();
                 if (areaList != null) {
                     areaList.forEach(area -> areaSet.add(area.longValue()));

--
Gitblit v1.9.1