Compare commits

..

No commits in common. "071b19e42e5a11dbff0e4be3c19f277eb02f381d" and "840cf849e63453d78c0c277a278d752c0eaf9824" have entirely different histories.

4 changed files with 63 additions and 66 deletions

View File

@ -1,59 +0,0 @@
package com.lesingle.edu.common.aspect;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.slf4j.MDC;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
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.UUID;
/**
* TraceId 切面
* Controller 执行前生成 traceId 并放入 MDC便于日志链路追踪
*/
@Slf4j
@Aspect
@Component
@Order(Ordered.HIGHEST_PRECEDENCE)
public class TraceIdAspect {
/**
* MDC traceId key
*/
private static final String TRACE_ID_KEY = "traceId";
/**
* 切点Controller 层所有方法
*/
@Around("execution(* com.lesingle.edu.controller..*.*(..))")
public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
// 生成 traceId使用 UUID 8 位大写
String traceId = generateTraceId();
// 放入 MDC 上下文
MDC.put(TRACE_ID_KEY, traceId);
try {
// 执行目标方法
return joinPoint.proceed();
} finally {
// 清理 MDC防止内存泄漏
MDC.clear();
}
}
/**
* 生成 traceId
* 使用 UUID 的前 8 保证唯一性的同时保持简洁
*/
private String generateTraceId() {
return UUID.randomUUID().toString().replace("-", "").substring(0, 8).toUpperCase();
}
}

View File

@ -1,5 +1,6 @@
package com.lesingle.edu.common.config; package com.lesingle.edu.common.config;
import com.lesingle.edu.common.filter.TraceIdFilter;
import com.lesingle.edu.common.security.JwtAuthenticationFilter; import com.lesingle.edu.common.security.JwtAuthenticationFilter;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
@ -30,7 +31,7 @@ import java.util.List;
@RequiredArgsConstructor @RequiredArgsConstructor
public class SecurityConfig { public class SecurityConfig {
private final JwtAuthenticationFilter jwtAuthenticationFilter; private final TraceIdFilter traceIdFilter;
@Bean @Bean
public SecurityFilterChain securityFilterChain(HttpSecurity http) throws Exception { public SecurityFilterChain securityFilterChain(HttpSecurity http) throws Exception {
@ -52,8 +53,7 @@ public class SecurityConfig {
// All other requests require authentication // All other requests require authentication
.anyRequest().authenticated() .anyRequest().authenticated()
) )
// 添加 JWT 过滤器到 UsernamePasswordAuthenticationFilter 之前 .addFilterBefore(traceIdFilter, UsernamePasswordAuthenticationFilter.class);
.addFilterBefore(jwtAuthenticationFilter, UsernamePasswordAuthenticationFilter.class);
return http.build(); return http.build();
} }

View File

@ -0,0 +1,57 @@
package com.lesingle.edu.common.filter;
import jakarta.servlet.FilterChain;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.MDC;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
import java.io.IOException;
import java.util.UUID;
/**
* 链路追踪过滤器
* 为每个请求生成唯一的 traceId放入 MDC 上下文便于日志链路追踪
*/
@Slf4j
@Component
@Order(Ordered.HIGHEST_PRECEDENCE + 1) // 最优先执行确保 traceId 在所有过滤器之前生成
public class TraceIdFilter extends OncePerRequestFilter {
/**
* MDC traceId key
*/
private static final String TRACE_ID_KEY = "traceId";
@Override
protected void doFilterInternal(HttpServletRequest request,
HttpServletResponse response,
FilterChain filterChain) throws ServletException, IOException {
// 生成 traceId使用 UUID 8
String traceId = generateTraceId();
// 放入 MDC 上下文
MDC.put(TRACE_ID_KEY, traceId);
try {
// 执行过滤器链
filterChain.doFilter(request, response);
} finally {
// 清理 MDC防止内存泄漏
MDC.clear();
}
}
/**
* 生成 traceId
* 使用 UUID 的前 8 保证唯一性的同时保持简洁
*/
private String generateTraceId() {
return UUID.randomUUID().toString().replace("-", "").substring(0, 8).toUpperCase();
}
}

View File

@ -18,6 +18,8 @@ import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.http.HttpStatus; import org.springframework.http.HttpStatus;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.authority.SimpleGrantedAuthority; import org.springframework.security.core.authority.SimpleGrantedAuthority;
@ -37,6 +39,7 @@ import java.util.Map;
@Slf4j @Slf4j
@Component @Component
@RequiredArgsConstructor @RequiredArgsConstructor
@Order(Ordered.HIGHEST_PRECEDENCE + 10) // TraceIdFilter 之后执行
public class JwtAuthenticationFilter extends OncePerRequestFilter { public class JwtAuthenticationFilter extends OncePerRequestFilter {
private final JwtTokenProvider jwtTokenProvider; private final JwtTokenProvider jwtTokenProvider;
@ -49,17 +52,13 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
@Override @Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response,
FilterChain filterChain) throws ServletException, IOException { FilterChain filterChain) throws ServletException, IOException {
log.debug("JwtAuthenticationFilter doFilterInternal called for: {}", request.getRequestURI());
try { try {
String token = resolveToken(request); String token = resolveToken(request);
log.debug("Token extracted: {}", token != null ? "present" : "null");
if (StringUtils.hasText(token)) { if (StringUtils.hasText(token)) {
log.debug("Token validation starting...");
// 验证 token 并获取错误原因 // 验证 token 并获取错误原因
String tokenErrorReason = jwtTokenProvider.validateTokenWithReason(token); String tokenErrorReason = jwtTokenProvider.validateTokenWithReason(token);
if (tokenErrorReason != null) { if (tokenErrorReason != null) {
// token 无效返回 401 错误 // token 无效返回 401 错误
log.debug("Token validation failed: {}", tokenErrorReason);
sendError(response, HttpStatus.UNAUTHORIZED, getErrorMessage(tokenErrorReason)); sendError(response, HttpStatus.UNAUTHORIZED, getErrorMessage(tokenErrorReason));
return; return;
} }