123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138 |
- package com.ruoyi.common.helper;
- import cn.dev33.satoken.context.SaHolder;
- import cn.dev33.satoken.stp.SaLoginModel;
- import cn.dev33.satoken.stp.StpUtil;
- import cn.hutool.core.util.ObjectUtil;
- import cn.hutool.json.JSONObject;
- import com.ruoyi.common.constant.UserConstants;
- import com.ruoyi.common.core.domain.model.LoginUser;
- import com.ruoyi.common.enums.DeviceType;
- import com.ruoyi.common.enums.UserType;
- import com.ruoyi.common.exception.UtilException;
- import com.ruoyi.common.utils.StringUtils;
- import lombok.AccessLevel;
- import lombok.NoArgsConstructor;
- /**
- * 登录鉴权助手
- * <p>
- * user_type 为 用户类型 同一个用户表 可以有多种用户类型 例如 pc,app
- * deivce 为 设备类型 同一个用户类型 可以有 多种设备类型 例如 web,ios
- * 可以组成 用户类型与设备类型多对多的 权限灵活控制
- * <p>
- * 多用户体系 针对 多种用户类型 但权限控制不一致
- * 可以组成 多用户类型表与多设备类型 分别控制权限
- *
- * @author Lion Li
- */
- @NoArgsConstructor(access = AccessLevel.PRIVATE)
- public class LoginHelper {
- public static final String JOIN_CODE = ":";
- public static final String LOGIN_USER_KEY = "loginUser";
- /**
- * 登录系统
- *
- * @param loginUser 登录用户信息
- */
- public static void login(LoginUser loginUser) {
- SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser);
- StpUtil.login(loginUser.getLoginId(), new SaLoginModel().setExtra(LOGIN_USER_KEY, loginUser));
- }
- /**
- * 登录系统 基于 设备类型
- * 针对相同用户体系不同设备
- *
- * @param loginUser 登录用户信息
- */
- public static void loginByDevice(LoginUser loginUser, DeviceType deviceType) {
- SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser);
- StpUtil.login(loginUser.getLoginId(),
- new SaLoginModel()
- .setDevice(deviceType.getDevice())
- .setExtra(LOGIN_USER_KEY, loginUser));
- }
- /**
- * 获取用户(多级缓存)
- */
- public static LoginUser getLoginUser() {
- LoginUser loginUser = (LoginUser) SaHolder.getStorage().get(LOGIN_USER_KEY);
- if (loginUser != null) {
- return loginUser;
- }
- loginUser = ((JSONObject) StpUtil.getExtra(LOGIN_USER_KEY)).toBean(LoginUser.class);
- SaHolder.getStorage().set(LOGIN_USER_KEY, loginUser);
- return loginUser;
- }
- /**
- * 获取用户基于token
- */
- public static LoginUser getLoginUser(String token) {
- return ((JSONObject) StpUtil.getExtra(token, LOGIN_USER_KEY)).toBean(LoginUser.class);
- }
- /**
- * 获取用户id
- */
- public static Long getUserId() {
- LoginUser loginUser = getLoginUser();
- if (ObjectUtil.isNull(loginUser)) {
- String loginId = StpUtil.getLoginIdAsString();
- String userId = null;
- for (UserType value : UserType.values()) {
- if (StringUtils.contains(loginId, value.getUserType())) {
- String[] strs = StringUtils.split(loginId, JOIN_CODE);
- // 用户id在总是在最后
- userId = strs[strs.length - 1];
- }
- }
- if (StringUtils.isBlank(userId)) {
- throw new UtilException("登录用户: LoginId异常 => " + loginId);
- }
- return Long.parseLong(userId);
- }
- return loginUser.getUserId();
- }
- /**
- * 获取部门ID
- */
- public static Long getDeptId() {
- return getLoginUser().getDeptId();
- }
- /**
- * 获取用户账户
- */
- public static String getUsername() {
- return getLoginUser().getUsername();
- }
- /**
- * 获取用户类型
- */
- public static UserType getUserType() {
- String loginId = StpUtil.getLoginIdAsString();
- return UserType.getUserType(loginId);
- }
- /**
- * 是否为管理员
- *
- * @param userId 用户ID
- * @return 结果
- */
- public static boolean isAdmin(Long userId) {
- return UserConstants.ADMIN_ID.equals(userId);
- }
- public static boolean isAdmin() {
- return isAdmin(getUserId());
- }
- }
|