fix bug

liulingling.177216

2024-08-26 349f1cfc5fa77fbc636d542df0d8050fddec48c2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package com.dingzhuo.energy.common.utils;
 
import com.dingzhuo.energy.common.constant.HttpStatus;
import com.dingzhuo.energy.common.exception.CustomException;
import com.dingzhuo.energy.framework.security.LoginUser;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
 
/**
 * 安全服务工具类
 *
 * @author ruoyi
 */
public class SecurityUtils {
 
  /**
   * 获取用户账户
   **/
  public static String getUsername() {
    try {
      return getLoginUser().getUsername();
    } catch (Exception e) {
      throw new CustomException("获取用户账户异常", HttpStatus.UNAUTHORIZED);
    }
  }
 
  /**
   * 获取用户
   **/
  public static Long getUserId() {
    try {
      return getLoginUser().getUser().getUserId();
    } catch (Exception e) {
      throw new CustomException("获取用户信息异常", HttpStatus.UNAUTHORIZED);
    }
  }
 
  /**
   * 获取用户
   **/
  public static LoginUser getLoginUser() {
    try {
      return (LoginUser) getAuthentication().getPrincipal();
    } catch (Exception e) {
      throw new CustomException("获取用户信息异常", HttpStatus.UNAUTHORIZED);
    }
  }
 
  /**
   * 获取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();
    return passwordEncoder.matches(rawPassword, encodedPassword);
  }
 
  /**
   * 是否为管理员
   *
   * @param userId 用户ID
   * @return 结果
   */
  public static boolean isAdmin(Long userId) {
    return userId != null && 1L == userId;
  }
}