spring cloud gateway 修改请求体和响应体 亲测有效

本文介绍了如何在Spring Cloud Gateway中使用全局过滤器和局部过滤器修改请求体和响应体。通过自定义MyCachedBodyOutputMessage解决访问限制,并详细讲解了全局和局部过滤器的实现逻辑。
摘要由CSDN通过智能技术生成

全局过滤器修改请求体

ModifyRequestBodyGlobalFilter是参考官方的ModifyRequestBodyGatewayFilterFactory,可以直接复制,只需在modifyBody方法中实现自定义逻辑。在ModifyRequestBodyGatewayFilterFactory是使用CachedBodyOutputMessage,但CachedBodyOutputMessage的isCache方法是default权限,我们自定义的类无法访问,故给isCache加上public权限就有了MyCachedBodyOutputMessage。

/**
 * 参考 {@link org.springframework.cloud.gateway.filter.factory.rewrite.ModifyRequestBodyGatewayFilterFactory}
 * @author mask
 * @since 2022/7/14 17:02
 */
@Component
@Slf4j
public class ModifyRequestBodyGlobalFilter implements GlobalFilter, Ordered {
   

    @Override
    public int getOrder() {
   
        return -2;
    }

    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
   

        ServerRequest serverRequest = ServerRequest.create(exchange, HandlerStrategies.withDefaults().messageReaders());
        Mono<String> modifiedBody = serverRequest.bodyToMono(String.class).flatMap(originalBody -> modifyBody()
                .apply(exchange,originalBody));

        BodyInserter bodyInserter = BodyInserters.fromPublisher(modifiedBody, String.class);

        HttpHeaders headers = new HttpHeaders();
        headers.putAll(exchange.getRequest().getHeaders());
        headers.remove("Content-Length");

        MyCachedBodyOutputMessage outputMessage = new MyCachedBodyOutputMessage(exchange, headers);
        return bodyInserter.insert(outputMessage, new BodyInserterContext()).then(Mono.defer(() -> {
   
            ServerHttpRequest decorator = decorate(exchange, headers, outputMessage);
            return chain.filter(exchange.mutate().request(decorator).build());
        })).onErrorResume((throwable) -> {
   
            return release(exchange, outputMessage,(Throwable) throwable);
        });
    }

    private BiFunction<ServerWebExchange, String, Mono<String>> modifyBody() {
   
        return new BiFunction<ServerWebExchange, String, Mono<String>>() {
   
            @Override
            public Mono<String> apply(ServerWebExchange serverWebExchange, String raw) {
   
                try {
   
             
  • 6
    点赞
  • 27
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值