SpringBoot从Service层获取request.getHeader()的几种方式
作者:悟能不能悟
在SpringBoot中,可以通过多种方式从Service层获取request.getHeader(),包括通过方法参数传递、使用RequestContextHolder、ThreadLocal、拦截器、注入HttpServletRequest等,推荐使用方法参数传递,适用于大多数场景,根据具体需求,可以选择适合的方案
在Spring Boot中,有几种方式可以从Service层获取request.getHeader():
1.通过方法参数传递(推荐)
Service接口:
@Service
public class UserService {
public String getUserInfo(String headerValue) {
// 使用headerValue
return "处理Header: " + headerValue;
}
}Controller层:
@RestController
public class UserController {
@Autowired
private UserService userService;
@GetMapping("/user")
public String getUser(@RequestHeader("Authorization") String token) {
return userService.getUserInfo(token);
}
}2.使用RequestContextHolder(常用)
@Service
public class UserService {
public String getUserInfo() {
// 获取HttpServletRequest对象
HttpServletRequest request = ((ServletRequestAttributes)
RequestContextHolder.getRequestAttributes()).getRequest();
// 获取特定Header
String token = request.getHeader("Authorization");
String userAgent = request.getHeader("User-Agent");
// 获取所有Header名称
Enumeration<String> headerNames = request.getHeaderNames();
return "Token: " + token;
}
}3.通过ThreadLocal持有Request
创建RequestHolder工具类:
@Component
public class RequestHolder {
private static final ThreadLocal<HttpServletRequest> requestHolder = new ThreadLocal<>();
public static void setRequest(HttpServletRequest request) {
requestHolder.set(request);
}
public static HttpServletRequest getRequest() {
return requestHolder.get();
}
public static String getHeader(String headerName) {
HttpServletRequest request = getRequest();
return request != null ? request.getHeader(headerName) : null;
}
public static void remove() {
requestHolder.remove();
}
}创建拦截器:
@Component
public class RequestInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request,
HttpServletResponse response,
Object handler) {
RequestHolder.setRequest(request);
return true;
}
@Override
public void afterCompletion(HttpServletRequest request,
HttpServletResponse response,
Object handler,
Exception ex) {
RequestHolder.remove(); // 防止内存泄漏
}
}注册拦截器:
@Configuration
public class WebConfig implements WebMvcConfigurer {
@Autowired
private RequestInterceptor requestInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(requestInterceptor);
}
}Service中使用:
@Service
public class UserService {
public String getUserInfo() {
String token = RequestHolder.getHeader("Authorization");
return "Token: " + token;
}
}4.注入HttpServletRequest
@Service
public class UserService {
@Autowired
private HttpServletRequest request; // 需要@RequestScope
public String getUserInfo() {
String token = request.getHeader("Authorization");
return "Token: " + token;
}
}注意:这种方式需要在Spring Boot 2.x及以上版本,并且Service必须是RequestScope。
@Service
@RequestScope
public class UserService {
// ...
}5.封装Header信息到DTO
HeaderDTO:
@Data
public class RequestContext {
private String token;
private String userAgent;
private String clientIp;
// 其他需要的Header信息
}Controller:
@RestController
public class UserController {
@Autowired
private UserService userService;
@GetMapping("/user")
public String getUser(HttpServletRequest request) {
RequestContext context = new RequestContext();
context.setToken(request.getHeader("Authorization"));
context.setUserAgent(request.getHeader("User-Agent"));
return userService.getUserInfo(context);
}
}6.使用AOP获取
@Aspect
@Component
public class RequestAspect {
@Pointcut("execution(* com.example.service..*.*(..))")
public void servicePointcut() {}
@Before("servicePointcut()")
public void before(JoinPoint joinPoint) {
HttpServletRequest request = ((ServletRequestAttributes)
RequestContextHolder.getRequestAttributes()).getRequest();
// 可以将Header信息放入MDC或ThreadLocal
MDC.put("token", request.getHeader("Authorization"));
}
}最佳实践建议:
- 推荐方案1:通过方法参数传递,保持Service层纯净
- 异步场景:如果需要跨线程,需要手动传递Header信息
- 日志追踪:可以使用MDC存储TraceID等跟踪信息
- 安全性:注意ThreadLocal的清理,避免内存泄漏
// 异步线程中传递Header的示例
@Async
public CompletableFuture<String> asyncProcess() {
// 在主线程获取Header
String token = ((ServletRequestAttributes)
RequestContextHolder.getRequestAttributes())
.getRequest()
.getHeader("Authorization");
return CompletableFuture.supplyAsync(() -> {
// 异步线程中无法直接获取RequestContextHolder
// 需要手动传递
return processWithToken(token);
});
}选择哪种方式取决于具体业务场景,一般来说,优先考虑方案1或方案2。
以上就是SpringBoot从Service层获取request.getHeader()的几种方式的详细内容,更多关于SpringBoot Service获取request.getHeader()的资料请关注脚本之家其它相关文章!
