java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > SpringBoot Service获取request.getHeader()

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. 推荐方案1:通过方法参数传递,保持Service层纯净
  2. 异步场景:如果需要跨线程,需要手动传递Header信息
  3. 日志追踪:可以使用MDC存储TraceID等跟踪信息
  4. 安全性:注意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()的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文