admin/lx-admin-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix 修复 LoginHelper 获取 LoginUser 缺失子类扩展数据问题

Browse Source
This commit is contained in:
疯狂的狮子li 2023-02-20 20:45:45 +08:00
parent a49ed727b6
commit d957914c30
1 changed files with 15 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
ruoyi-common/ruoyi-common-satoken/src/main/java/com/ruoyi/common/satoken/utils/LoginHelper.java
View File

@ -3,8 +3,8 @@ package com.ruoyi.common.satoken.utils;
import cn.dev33.satoken.context.SaHolder; import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.stp.SaLoginModel; import cn.dev33.satoken.stp.SaLoginModel;
import cn.dev33.satoken.stp.StpUtil; import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.ObjectUtil;
import cn.hutool.json.JSONObject;
import com.ruoyi.common.core.constant.TenantConstants; import com.ruoyi.common.core.constant.TenantConstants;
import com.ruoyi.common.core.constant.UserConstants; import com.ruoyi.common.core.constant.UserConstants;
import com.ruoyi.common.core.domain.model.LoginUser; import com.ruoyi.common.core.domain.model.LoginUser;
@ -31,7 +31,8 @@ import java.util.Set;
public class LoginHelper { public class LoginHelper {
public static final String LOGIN_USER_KEY = "loginUser"; public static final String LOGIN_USER_KEY = "loginUser";
public static final String MENU_PERMISSION = "menuPermission"; public static final String TENANT_KEY = "tenantId";
public static final String USER_KEY = "userId";
/** /**
* 登录系统 * 登录系统
@ -50,30 +51,25 @@ public class LoginHelper {
*/ */
public static void loginByDevice(LoginUser loginUser, DeviceType deviceType) { public static void loginByDevice(LoginUser loginUser, DeviceType deviceType) {
SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser); SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser);
Set<String> menuPermission = loginUser.getMenuPermission();
loginUser.setMenuPermission(null);
SaLoginModel model = new SaLoginModel(); SaLoginModel model = new SaLoginModel();
if (ObjectUtil.isNotNull(deviceType)) { if (ObjectUtil.isNotNull(deviceType)) {
model.setDevice(deviceType.getDevice()); model.setDevice(deviceType.getDevice());
} }
StpUtil.login(loginUser.getLoginId(), model.setExtra(LOGIN_USER_KEY, loginUser)); StpUtil.login(loginUser.getLoginId(),
// 解决菜单权限过度 token 臃肿过长问题 model.setExtra(TENANT_KEY, loginUser.getTenantId())
StpUtil.getTokenSession().set(MENU_PERMISSION, menuPermission); .setExtra(USER_KEY, loginUser.getUserId()));
StpUtil.getTokenSession().set(LOGIN_USER_KEY, loginUser);
} }
/** /**
* 获取用户(多级缓存) * 获取用户(多级缓存)
*/ */
@SuppressWarnings("unchecked cast")
public static LoginUser getLoginUser() { public static LoginUser getLoginUser() {
LoginUser loginUser = (LoginUser) SaHolder.getStorage().get(LOGIN_USER_KEY); LoginUser loginUser = (LoginUser) SaHolder.getStorage().get(LOGIN_USER_KEY);
if (loginUser != null) { if (loginUser != null) {
return loginUser; return loginUser;
} }
loginUser = ((JSONObject) StpUtil.getExtra(LOGIN_USER_KEY)).toBean(LoginUser.class); loginUser = (LoginUser) StpUtil.getTokenSession().get(LOGIN_USER_KEY);
// 解决菜单权限过度 token 臃肿过长问题
Set<String> menuPermission = (Set<String>) StpUtil.getTokenSession().get(MENU_PERMISSION);
loginUser.setMenuPermission(menuPermission);
SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser); SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser);
return loginUser; return loginUser;
} }
@ -81,39 +77,34 @@ public class LoginHelper {
/** /**
* 获取用户基于token * 获取用户基于token
*/ */
@SuppressWarnings("unchecked cast")
public static LoginUser getLoginUser(String token) { public static LoginUser getLoginUser(String token) {
LoginUser loginUser = ((JSONObject) StpUtil.getExtra(token, LOGIN_USER_KEY)).toBean(LoginUser.class); return (LoginUser) StpUtil.getTokenSessionByToken(token).get(LOGIN_USER_KEY);
// 解决菜单权限过度 token 臃肿过长问题
Set<String> menuPermission = (Set<String>) StpUtil.getTokenSessionByToken(token).get(MENU_PERMISSION);
loginUser.setMenuPermission(menuPermission);
return loginUser;
} }
/** /**
* 获取用户id * 获取用户id
*/ */
public static Long getUserId() { public static Long getUserId() {
LoginUser loginUser; Long userId;
try { try {
loginUser = getLoginUser(); userId = Convert.toLong(StpUtil.getExtra(USER_KEY));
} catch (Exception e) { } catch (Exception e) {
return null; return null;
} }
return loginUser.getUserId(); return userId;
} }
/** /**
* 获取租户ID * 获取租户ID
*/ */
public static String getTenantId() { public static String getTenantId() {
LoginUser loginUser; String tenantId;
try { try {
loginUser = getLoginUser(); tenantId = (String) StpUtil.getExtra(TENANT_KEY);
} catch (Exception e) { } catch (Exception e) {
return null; return null;
} }
return loginUser.getTenantId(); return tenantId;
} }
/** /**