package com.ruoyi.common.utils; import com.ruoyi.common.core.domain.entity.SysRole; import org.springframework.security.core.Authentication; import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder; import com.ruoyi.common.constant.HttpStatus; import com.ruoyi.common.core.domain.model.LoginUser; import com.ruoyi.common.exception.ServiceException; import java.util.List; import java.util.Map; import java.util.concurrent.atomic.AtomicBoolean; /** * 安全服务工具类 * * @author ruoyi */ public class SecurityUtils { /** * 用户ID **/ public static Long getUserId() { try { return getLoginUser().getUserId(); } catch (Exception e) { throw new ServiceException("获取用户ID异常", HttpStatus.UNAUTHORIZED); } } /** * 获取部门ID **/ public static Long getDeptId() { try { return getLoginUser().getDeptId(); } catch (Exception e) { throw new ServiceException("获取部门ID异常", HttpStatus.UNAUTHORIZED); } } /** * 获取用户账户 **/ public static String getUsername() { try { return getLoginUser().getUsername(); } catch (Exception e) { throw new ServiceException("获取用户账户异常", HttpStatus.UNAUTHORIZED); } } /** * 获取用户 **/ public static LoginUser getLoginUser() { try { return (LoginUser) getAuthentication().getPrincipal(); } catch (Exception e) { throw new ServiceException("获取用户信息异常", HttpStatus.UNAUTHORIZED); } } public static List getUserRoles() { try { LoginUser u = (LoginUser) getAuthentication().getPrincipal(); return u.getUser().getRoles(); } catch (Exception e) { throw new ServiceException("获取用户权限异常", HttpStatus.UNAUTHORIZED); } } public static boolean hashRoles(Long roleId){ List userRoles = getUserRoles(); AtomicBoolean flag= new AtomicBoolean(false); userRoles.stream().forEach(e ->{ if (e.getRoleId()==roleId){ flag.set(true); } }); return flag.get(); } /** * 获取Authentication */ public static Authentication getAuthentication() { return SecurityContextHolder.getContext().getAuthentication(); } /** * 生成BCryptPasswordEncoder密码 * * @param password 密码 * @return 加密字符串 */ public static String encryptPassword(String password) { BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder(); return passwordEncoder.encode(password); } /** * 判断密码是否相同 * * @param rawPassword 真实密码 * @param encodedPassword 加密后字符 * @return 结果 */ public static boolean matchesPassword(String rawPassword, String encodedPassword) { BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder(); //密码丢失后,可以用任何密码获取转码,修改数据库登录 // String encode = passwordEncoder.encode(rawPassword); // System.out.println(encode); return passwordEncoder.matches(rawPassword, encodedPassword); } /** * 是否为管理员 * * @param userId 用户ID * @return 结果 */ public static boolean isAdmin(Long userId) { return userId != null && 1L == userId; } public static boolean isSuperAdmin() { if (getUserId()==1l){ return true; } return false; } }