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 对象,不进行任何修改
    }
}
posted @ 2024-01-23 17:07  盗梦笔记  阅读(396)  评论(0编辑  收藏  举报