1. 概述
在本快速教程中，我们将探讨使用Spring 5的WebFlux框架为非响应性端点或API设置服务响应标头的不同方法。
我们可以在以前的帖子中找到有关此框架的更多信息。
2. 非响应性组件的标题
如果我们想在单个响应上设置标头，我们可以使用HttpServletResponse或ResponseEntity对象。
相反，如果我们的目标是为所有或多个响应添加一个过滤器，我们将需要配置一个Filter。
2.1 使用HttpServletResponse
我们只需将HttpServletResponse对象作为参数添加到我们的REST端点，然后使用addHeader()方法：
@GetMapping("/http-servlet-response")
public String usingHttpServletResponse(HttpServletResponse response) {
response.addHeader("Tuyucheng-Example-Header", "Value-HttpServletResponse");
return "Response with header using HttpServletResponse";
}
如上例所示，我们不必返回响应对象。
2.2 使用ResponseEntity
在这种情况下，我们将使用ResponseEntity类提供的BodyBuilder：
@GetMapping("/response-entity-builder-with-http-headers")
public ResponseEntity<String> usingResponseEntityBuilderAndHttpHeaders() {
HttpHeaders responseHeaders = new HttpHeaders();
responseHeaders.set("Tuyucheng-Example-Header","Value-ResponseEntityBuilderWithHttpHeaders");
return ResponseEntity.ok()
.headers(responseHeaders)
.body("Response with header using ResponseEntity");
}
HttpHeaders类提供了大量方便的方法来设置最常见的标头。
我们可以在我们的Github仓库中看到更多说明这些要点的示例。
2.3 为所有响应添加标题
现在假设我们要为我们的许多端点设置一个特定的标头。
当然，如果我们必须在每个映射方法上以前的代码，那将是令人沮丧的。
实现此目的的更好方法是在我们的服务中配置过滤器：
@WebFilter("/filter-response-header/*")
public class AddResponseHeaderFilter implements Filter {
@Override
public void doFilter(ServletRequest request, ServletResponse response,
FilterChain chain) throws IOException, ServletException {
HttpServletResponse httpServletResponse = (HttpServletResponse) response;
httpServletResponse.setHeader(
"Tuyucheng-Example-Filter-Header", "Value-Filter");
chain.doFilter(request, response);
}
@Override
public void init(FilterConfig filterConfig) throws ServletException {
// ...
}
@Override
public void destroy() {
// ...
}
}
@WebFilter注解允许我们指示此过滤器将对其生效的urlPatterns。
正如我们在本文中指出的，为了让我们的Filter可以被Spring发现，我们需要在我们的Spring Application类中添加@ServletComponentScan注解：
@ServletComponentScan
@SpringBootApplication
public class ResponseHeadersApplication {
public static void main(String[] args) {
SpringApplication.run(ResponseHeadersApplication.class, args);
}
}
如果我们不需要@WebFilter提供的任何功能，我们可以通过在Filter类中使用@Component注解来避免这最后一步。
3. 响应端点的标头
在本节中，我们将学习如何使用ServerHttpResponse、ResponseEntity或ServerResponse(对于函数式端点)类和接口在单个端点响应上设置标头。
我们还将讨论如何实现Spring 5 WebFilter以在我们所有的响应中添加标头。
3.1 使用ServerHttpResponse
这种方法与HttpServletResponse对应物非常相似 ：
@GetMapping("/server-http-response")
public Mono<String> usingServerHttpResponse(ServerHttpResponse response) {
response.getHeaders().add("Tuyucheng-Example-Header", "Value-ServerHttpResponse");
return Mono.just("Response with header using ServerHttpResponse");
}
3.2 使用ResponseEntity
我们可以像对非响应性端点一样使用ResponseEntity类：
@GetMapping("/response-entity")
public Mono<ResponseEntity<String>> usingResponseEntityBuilder() {
String responseHeaderKey = "Tuyucheng-Example-Header";
String responseHeaderValue = "Value-ResponseEntityBuilder";
String responseBody = "Response with header using ResponseEntity (builder)";
return Mono.just(ResponseEntity.ok()
.header(responseHeaderKey, responseHeaderValue)
.body(responseBody));
}
3.3 使用ServerResponse
在最后两个小节中介绍的类和接口可以在@Controller注解类中使用，但不适合新的Spring 5函数式Web框架。
如果我们想在HandlerFunction上设置标头，那么我们需要接触ServerResponse接口：
public Mono<ServerResponse> useHandler(final ServerRequest request) {
return ServerResponse.ok()
.header("Tuyucheng-Example-Header", "Value-Handler")
.body(Mono.just("Response with header using Handler"),String.class);
}
3.4 为所有响应添加标题
最后，Spring 5提供了一个WebFilter接口来为服务检索的所有响应设置标头：
@Component
public class AddResponseHeaderWebFilter implements WebFilter {
@Override
public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
exchange.getResponse()
.getHeaders()
.add("Tuyucheng-Example-Filter-Header", "Value-Filter");
return chain.filter(exchange);
}
}
4. 总结
在本文中，我们了解了多种在响应中设置标头的不同方法。现在，无论我们是想在单个端点上设置它，配置我们所有的Rest API，还是迁移到响应式堆栈，我们都拥有必要的知识。
与往常一样，本教程的完整源代码可在GitHub上获得。
Post Directory
