diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/service/SysLoginService.java b/ruoyi-admin/src/main/java/com/ruoyi/web/service/SysLoginService.java index 57e9c48..357dac1 100644 --- a/ruoyi-admin/src/main/java/com/ruoyi/web/service/SysLoginService.java +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/service/SysLoginService.java @@ -5,7 +5,6 @@ import cn.dev33.satoken.stp.StpUtil; import cn.hutool.core.bean.BeanUtil; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.util.ObjectUtil; -import com.mybatisflex.core.tenant.TenantManager; import com.ruoyi.common.core.constant.*; import com.ruoyi.common.core.core.domain.dto.RoleDTO; import com.ruoyi.common.core.enums.LoginType; @@ -21,21 +20,15 @@ import com.ruoyi.common.tenant.helper.TenantHelper; import com.ruoyi.system.domain.SysUser; import com.ruoyi.system.domain.bo.SysSocialBo; import com.ruoyi.system.domain.bo.SysUserBo; -import com.ruoyi.system.domain.vo.SysSocialVo; -import com.ruoyi.system.domain.vo.SysTenantVo; -import com.ruoyi.system.domain.vo.SysUserVo; -import com.ruoyi.system.service.ISysPermissionService; -import com.ruoyi.system.service.ISysSocialService; -import com.ruoyi.system.service.ISysTenantService; +import com.ruoyi.system.domain.vo.*; +import com.ruoyi.system.service.*; import jakarta.annotation.Resource; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import me.zhyd.oauth.model.AuthUser; -import org.springframework.beans.factory.annotation.Autowired; import com.ruoyi.common.core.core.domain.model.LoginUser; import com.ruoyi.common.core.utils.DateUtils; import com.ruoyi.common.core.utils.MessageUtils; -import com.ruoyi.system.service.ISysUserService; import org.springframework.stereotype.Service; import org.springframework.beans.factory.annotation.Value; @@ -70,6 +63,12 @@ public class SysLoginService { @Resource private ISysUserService userService; + @Resource + private ISysDeptService deptService; + + @Resource + private ISysRoleService roleService; + @Resource private ISysTenantService tenantService; @@ -198,9 +197,12 @@ public class SysLoginService { loginUser.setUserType(user.getUserType()); loginUser.setMenuPermission(permissionService.getMenuPermission(user.getUserId())); loginUser.setRolePermission(permissionService.getRolePermission(user.getUserId())); - loginUser.setDeptName(ObjectUtil.isNull(user.getDept()) ? "" : user.getDept().getDeptName()); - List roles = BeanUtil.copyToList(user.getRoles(), RoleDTO.class); - loginUser.setRoles(roles); + TenantHelper.dynamic(user.getTenantId(), () -> { + SysDeptVo dept = deptService.selectDeptById(user.getDeptId()); + loginUser.setDeptName(ObjectUtil.isNull(dept) ? "" : dept.getDeptName()); + List roles = roleService.selectRolesByUserId(user.getUserId()); + loginUser.setRoles(BeanUtil.copyToList(roles, RoleDTO.class)); + }); return loginUser; }