彻底解决Spring Cloud Gateway中Body读取问题

开发 前端
AdaptCachedBodyGlobalFilter会监听EnableBodyCachingEvent事件,当发布该事件时就将RouteId放入routesToCache中。为了方便使用,我们可以编写一个配置类,在初始化时发布EnableBodyCachingEvent事件,将所有路由都启用缓存功能。

在构建微服务架构时,Spring Cloud Gateway作为一个重要的微服务网关,经常需要在过滤器(Filter)中对POST请求的Body内容进行操作,如日志记录、签名验证和权限验证等。然而,由于Request的Body只能读取一次,如果直接在过滤器中读取而不进行封装,可能导致后续服务无法获取数据。

网上搜这个问题的解决方案,大多数文章都是告诉你写一个Filter将Request的Body缓存起来。这种方法确实可以,只不过需要对代码经过充分压力测试,否则很有可能出现如下所示的堆外内存溢出问题。

reactor.netty.ReactorNetty$InternalNettyException: io.netty.util.internal.OutOfDirectMemoryError:failed to allocate

实际上,Spring Cloud Gateway已经内置了AdaptCachedBodyGlobalFilter过滤器,它在Exchange中巧妙地缓存了Request的Body,避免了直接读取导致的一系列问题。这种方式更为稳妥,避免了潜在的内存溢出风险。

图片图片

在需要获取Body的地方,我们只需要通过以下方法即可:

DataBuffer body = exchange.getAttributeOrDefault("cachedRequestBody", null);
String bodyStr = body.toString(StandardCharsets.UTF_8);

只不过通过源码可以看出,缓存RequestBody需要路由被标记为需要缓存,也就是this.routesToCache.containsKey(rouceId)方法必须返回true。

AdaptCachedBodyGlobalFilter会监听EnableBodyCachingEvent事件,当发布该事件时就将RouteId放入routesToCache中。为了方便使用,我们可以编写一个配置类,在初始化时发布EnableBodyCachingEvent事件,将所有路由都启用缓存功能。

@Configuration(proxyBeanMethods = false)
@Slf4j
public class EnableCachedBodyConfiguration {
    
    @Resource
    private ApplicationEventPublisher publisher;
    
    @Resource
    private GatewayProperties gatewayProperties;
    
    @PostConstruct
    public void init() {
        gatewayProperties.getRoutes().forEach(routeDefinition -> {
            // 对 spring cloud gateway 路由配置中的每个路由都启用 AdaptCachedBodyGlobalFilter
            EnableBodyCachingEvent enableBodyCachingEvent = new EnableBodyCachingEvent(new Object(), routeDefinition.getId());
            publisher.publishEvent(enableBodyCachingEvent);
        });
    }
}

通过这种方式,我们可以更加方便地处理POST请求的Body内容,避免了一些常见的问题。在实际应用中,考虑到稳定性和性能,这种解决方案提供了一种更为可靠的选择。

责任编辑:武晓燕 来源: JAVA日知录
相关推荐

2009-11-27 10:31:02

GPRS路由

2021-12-03 12:15:01

QT中文乱码Windows

2023-02-27 08:08:54

Pulsar源码重复消费

2010-01-04 15:05:53

2010-01-11 18:05:24

VB.NET窗体继承

2010-01-14 10:19:05

2009-12-25 09:39:08

ADSL MODEM

2009-12-03 18:45:41

2009-11-24 19:50:10

2009-12-03 16:33:02

路由交换设备

2022-10-08 23:55:58

iOS苹果开发

2024-10-09 12:05:27

2009-12-21 17:20:19

2022-05-31 09:01:13

GitHub工具安全

2009-12-03 16:54:04

无线宽带路由器

2020-09-28 14:41:24

Event Loop

2024-11-04 10:05:00

AI模型

2009-12-04 16:25:24

2021-03-13 21:00:30

电脑PC电脑弹窗广告

2021-05-18 09:06:19

零信任邮件安全安全威胁
点赞
收藏

51CTO技术栈公众号