package com.rongyichuang.common.util;
|
|
import com.rongyichuang.auth.util.JwtUtil;
|
import com.rongyichuang.employee.entity.Employee;
|
import com.rongyichuang.employee.repository.EmployeeRepository;
|
import com.rongyichuang.judge.entity.Judge;
|
import com.rongyichuang.judge.repository.JudgeRepository;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.security.core.Authentication;
|
import org.springframework.security.core.context.SecurityContextHolder;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.context.request.RequestContextHolder;
|
import org.springframework.web.context.request.ServletRequestAttributes;
|
|
import jakarta.servlet.http.HttpServletRequest;
|
import java.util.Optional;
|
|
/**
|
* 用户上下文工具类
|
* 用于获取当前登录用户信息和关联的评委信息
|
*/
|
@Component
|
public class UserContextUtil {
|
|
private static final Logger logger = LoggerFactory.getLogger(UserContextUtil.class);
|
|
@Autowired
|
private JudgeRepository judgeRepository;
|
|
@Autowired
|
private EmployeeRepository employeeRepository;
|
|
@Autowired
|
private JwtUtil jwtUtil;
|
|
/**
|
* 获取当前登录用户ID
|
* 从JWT token中解析用户ID
|
*
|
* @return 用户ID
|
*/
|
public Long getCurrentUserId() {
|
try {
|
// 首先尝试从HTTP请求头中获取JWT token
|
String token = getTokenFromRequest();
|
if (token != null && jwtUtil.validateToken(token)) {
|
Long userId = jwtUtil.getUserIdFromToken(token);
|
logger.debug("从JWT token中获取到用户ID: {}", userId);
|
return userId;
|
}
|
|
if (token == null) {
|
logger.debug("未能从请求头获取到JWT token");
|
} else {
|
logger.debug("从请求头获取到token但校验失败");
|
}
|
|
// 如果没有有效的JWT token，尝试从Spring Security上下文获取
|
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
|
if (authentication != null && authentication.isAuthenticated() &&
|
!"anonymousUser".equals(authentication.getPrincipal())) {
|
logger.debug("获取到认证用户: {}", authentication.getName());
|
// 在开发环境下，返回一个有效的评委用户ID
|
// 查找第一个有效的评委记录并返回其user_id
|
try {
|
Optional<Judge> firstJudge = judgeRepository.findAll().stream().findFirst();
|
if (firstJudge.isPresent() && firstJudge.get().getUserId() != null) {
|
Long userId = firstJudge.get().getUserId();
|
logger.debug("开发环境：使用评委用户ID: {}", userId);
|
return userId;
|
}
|
} catch (Exception e) {
|
logger.warn("查找评委用户ID时发生异常: {}", e.getMessage());
|
}
|
// 如果没有找到评委，返回固定用户ID
|
return 1L;
|
}
|
} catch (Exception e) {
|
logger.warn("获取当前用户ID时发生异常: {}", e.getMessage());
|
}
|
|
// 如果没有认证信息，返回null表示未登录
|
logger.debug("未找到有效的认证信息");
|
return null;
|
}
|
|
/**
|
* 从HTTP请求中获取JWT token
|
*/
|
private String getTokenFromRequest() {
|
try {
|
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
if (attributes == null) {
|
logger.warn("RequestContextHolder中无ServletRequestAttributes，可能为异步执行或非Servlet环境");
|
} else {
|
HttpServletRequest request = attributes.getRequest();
|
String authHeader = request.getHeader("Authorization");
|
logger.debug("读取到Authorization头: {}", authHeader);
|
if (authHeader != null && authHeader.startsWith("Bearer ")) {
|
String token = authHeader.substring(7);
|
logger.debug("从Authorization头提取到Bearer token，长度: {}", token != null ? token.length() : 0);
|
return token;
|
} else {
|
logger.debug("Authorization头不存在或不以Bearer开头");
|
}
|
}
|
} catch (Exception e) {
|
logger.debug("获取JWT token时发生异常: {}", e.getMessage());
|
}
|
return null;
|
}
|
|
/**
|
* 获取当前用户关联的员工信息
|
*
|
* @return 员工信息，如果当前用户不是员工则返回空
|
*/
|
public Optional<Employee> getCurrentEmployee() {
|
Long userId = getCurrentUserId();
|
if (userId == null) {
|
logger.warn("无法获取当前用户ID");
|
return Optional.empty();
|
}
|
|
try {
|
Optional<Employee> employee = employeeRepository.findByUserId(userId);
|
if (employee.isPresent()) {
|
logger.debug("找到当前用户关联的员工: {}", employee.get().getName());
|
} else {
|
logger.debug("当前用户(ID: {})不是员工", userId);
|
}
|
return employee;
|
} catch (Exception e) {
|
logger.error("查询员工信息时发生异常: {}", e.getMessage(), e);
|
return Optional.empty();
|
}
|
}
|
|
/**
|
* 获取当前用户关联的评委信息
|
*
|
* @return 评委信息，如果当前用户不是评委则返回空
|
*/
|
public Optional<Judge> getCurrentJudge() {
|
Long userId = getCurrentUserId();
|
if (userId == null) {
|
logger.warn("无法获取当前用户ID");
|
return Optional.empty();
|
}
|
|
try {
|
Optional<Judge> judge = judgeRepository.findByUserId(userId);
|
if (judge.isPresent()) {
|
logger.debug("找到当前用户关联的评委: {}", judge.get().getName());
|
} else {
|
logger.debug("当前用户(ID: {})不是评委", userId);
|
}
|
return judge;
|
} catch (Exception e) {
|
logger.error("查询评委信息时发生异常: {}", e.getMessage(), e);
|
return Optional.empty();
|
}
|
}
|
|
/**
|
* 获取当前用户关联的员工ID
|
*
|
* @return 员工ID，如果当前用户不是员工则返回null
|
*/
|
public Long getCurrentEmployeeId() {
|
return getCurrentEmployee().map(Employee::getId).orElse(null);
|
}
|
|
/**
|
* 获取当前用户关联的评委ID
|
*
|
* @return 评委ID，如果当前用户不是评委则返回null
|
*/
|
public Long getCurrentJudgeId() {
|
return getCurrentJudge().map(Judge::getId).orElse(null);
|
}
|
|
/**
|
* 检查当前用户是否为员工
|
*
|
* @return true如果当前用户是员工，否则false
|
*/
|
public boolean isCurrentUserEmployee() {
|
return getCurrentEmployee().isPresent();
|
}
|
|
/**
|
* 检查当前用户是否为评委
|
*
|
* @return true如果当前用户是评委，否则false
|
*/
|
public boolean isCurrentUserJudge() {
|
return getCurrentJudge().isPresent();
|
}
|
|
/**
|
* 检查当前用户是否为指定活动的评委
|
*
|
* @param activityId 活动ID
|
* @return true如果当前用户是该活动的评委，否则false
|
*/
|
public boolean isCurrentUserJudgeForActivity(Long activityId) {
|
Optional<Judge> judge = getCurrentJudge();
|
if (judge.isEmpty()) {
|
return false;
|
}
|
|
try {
|
// 通过ActivityJudge表检查当前评委是否参与指定活动
|
return judgeRepository.existsByIdAndActivityId(judge.get().getId(), activityId);
|
} catch (Exception e) {
|
logger.error("检查评委活动权限时发生异常: {}", e.getMessage(), e);
|
return false;
|
}
|
}
|
|
/**
|
* 获取当前用户名称（用于日志记录）
|
*
|
* @return 用户名称
|
*/
|
public String getCurrentUserName() {
|
try {
|
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
|
if (authentication != null && authentication.isAuthenticated()) {
|
return authentication.getName();
|
}
|
} catch (Exception e) {
|
logger.warn("获取当前用户名称时发生异常: {}", e.getMessage());
|
}
|
return "未知用户";
|
}
|
}
|
