SpringBoot实现RequestBodyAdvice和ResponseBodyAdvice接口
Spring Boot 提供了一种机制,允许开发者在请求体(RequestBody)和响应体(ResponseBody)被处理之前和之后执行自定义逻辑。这通过 RequestBodyAdvice 和 ResponseBodyAdvice 接口实现。
RequestBodyAdvice:此类用于在请求体被处理之前执行自定义逻辑。
例如,你可以使用它来解析请求体,或者在请求体被转换为 Java 对象之前对其进行修改。
ResponseBodyAdvice:此类用于在响应体被处理之后执行自定义逻辑。
例如,你可以使用它来修改响应体,或者在响应体被写入 HTTP 响应之前对其进行修改。
要使用这些 Advice,你需要实现相应的接口,并使用 Spring 的 AOP(面向切面编程)功能将其注册为一个切面(Aspect)。
以下是一个简单的示例,展示如何创建一个 RequestBodyAdvice:
import org.springframework.http.MediaType;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ServerWebExchange;
import org.springframework.web.server.WebFilter;
import org.springframework.web.server.WebFilterChain;
import org.springframework.web.server.adapter.WebHttpHandlerBuilder;
import org.springframework.web.server.handler.WebHandler;
import org.springframework.web.server.ServerWebExchange;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.http.server.reactive.ServerHttpResponseDecorator;
import org.springframework.http.server.reactive.ServerHttpResponseOutput;
import org.springframework.http.server.reactive.ServerResponse;
import org.springframework.stereotype.Component;
import org.springframework.webflux.core.*;
import org.springframework.webflux.http.*;
import org.springframework.webflux.*;
import org.springframework.*;
import org.springframework.*;
import org.*;
@Component
public class CustomRequestBodyAdvice implements RequestBodyAdvice<MyDto> {
@Override
public Mono<MyDto> beforeBodyRead(ServerHttpRequest request, MediaType mediaType,
Class<? extends MyDto> type, ServerWebExchange exchange) {
// 在请求体被读取之前执行的逻辑
return Mono.just(new MyDto()); // 示例逻辑:返回一个新的 MyDto 对象
}
@Override
public BodySpec<MyDto> afterBodyRead(BodySpec<MyDto> bodySpec,
ServerWebExchange exchange) {
// 在请求体被读取之后执行的逻辑
return bodySpec; // 示例逻辑:返回原始的 BodySpec 对象,不进行任何修改
}
@Override
public Mono<Void> beforeBodyWritten(Mono<MyDto> body, MediaType mediaType,
ServerWebExchange exchange, WebRequest缀> headers) {
// 在响应体被写入之前执行的逻辑
return Mono.empty(); // 示例逻辑:返回一个空的 Mono 对象,不进行任何修改
}
}