update 优化 getLoginUser 方法 支持返回多种类型登陆实体

dev
疯狂的狮子Li 10 months ago
parent 8879fc6f1b
commit 0e28aa7509

@ -62,23 +62,25 @@ public class LoginHelper {
/** /**
* () * ()
*/ */
public static LoginUser getLoginUser() { @SuppressWarnings("unchecked cast")
public static <T extends LoginUser> T getLoginUser() {
SaSession session = StpUtil.getTokenSession(); SaSession session = StpUtil.getTokenSession();
if (ObjectUtil.isNull(session)) { if (ObjectUtil.isNull(session)) {
return null; return null;
} }
return (LoginUser) session.get(LOGIN_USER_KEY); return (T) session.get(LOGIN_USER_KEY);
} }
/** /**
* token * token
*/ */
public static LoginUser getLoginUser(String token) { @SuppressWarnings("unchecked cast")
public static <T extends LoginUser> T getLoginUser(String token) {
SaSession session = StpUtil.getTokenSessionByToken(token); SaSession session = StpUtil.getTokenSessionByToken(token);
if (ObjectUtil.isNull(session)) { if (ObjectUtil.isNull(session)) {
return null; return null;
} }
return (LoginUser) session.get(LOGIN_USER_KEY); return (T) session.get(LOGIN_USER_KEY);
} }
/** /**
@ -191,7 +193,11 @@ public class LoginHelper {
* @return * @return
*/ */
public static boolean isTenantAdmin() { public static boolean isTenantAdmin() {
return Convert.toBool(isTenantAdmin(getLoginUser().getRolePermission())); LoginUser loginUser = getLoginUser();
if (loginUser == null) {
return false;
}
return Convert.toBool(isTenantAdmin(loginUser.getRolePermission()));
} }
/** /**

Loading…
Cancel
Save