SpringCloud-Gateway自定义过滤器，解决body流不能重复读的问题

网上的代码千奇百怪，这个是我测试后整理的。

当前端请求不含body时，也能兼容。

1.读取body流进行缓存

/*** 获取body请求数据（解决流不能重复读取问题）* 这个过滤器的order设置的是Ordered.HIGHEST_PRECEDENCE，即最高优先级的过滤器。* 优先级设置这么高的原因是某些系统内置的过滤器可能也会去读body，* 这样就会导致我们自定义过滤器中获取body的时候报body只能读取一次这样的错误* @author yuchen*/
@Slf4j
@Component
public class CacheRequestFilter implements GlobalFilter, Ordered {@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {ServerHttpRequest request = exchange.getRequest();if (request.getHeaders().getContentType() == null) {return chain.filter(exchange);}HttpMethod method = request.getMethod();if (method == null || method.matches("GET") || method.matches("DELETE")) {return chain.filter(exchange);}//当body中没有缓存时，只会执行这一个拦截器， 原因是fileMap中的代码没有执行，所以需要在波多野为空时构建一个空的缓存DefaultDataBufferFactory defaultDataBufferFactory = new DefaultDataBufferFactory();DefaultDataBuffer defaultDataBuffer = defaultDataBufferFactory.allocateBuffer(0);//构建新数据流， 当body为空时，构建空流Flux<DataBuffer> bodyDataBuffer = exchange.getRequest().getBody().defaultIfEmpty(defaultDataBuffer);return DataBufferUtils.join(bodyDataBuffer).flatMap(dataBuffer -> {DataBufferUtils.retain(dataBuffer);Flux<DataBuffer> cachedFlux = Flux.defer(() -> Flux.just(dataBuffer.slice(0, dataBuffer.readableByteCount())));ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {@Overridepublic Flux<DataBuffer> getBody() {return cachedFlux;}};//exchange.getAttributes().put(CACHE_REQUEST_BODY_OBJECT_KEY, cachedFlux);return chain.filter(exchange.mutate().request(mutatedRequest).build());});}@Overridepublic int getOrder() {return Ordered.HIGHEST_PRECEDENCE;}}

2.从缓存读取body，就可以重复读：

    private static String resolveBodyFromRequest(ServerHttpRequest serverHttpRequest) {// 获取请求体Flux<DataBuffer> body = serverHttpRequest.getBody();AtomicReference<String> bodyRef = new AtomicReference<>();body.subscribe(buffer -> {CharBuffer charBuffer = StandardCharsets.UTF_8.decode(buffer.asByteBuffer());DataBufferUtils.release(buffer);bodyRef.set(charBuffer.toString());});return bodyRef.get();}

SpringCloud-Gateway自定义过滤器，解决body流不能重复读的问题相关推荐

数据库隔离级别解决脏读、不可重复读、幻读
一.脏读.不可重复读.幻读脏读读取了未提交的事务. 事务A读取了事务B中尚未提交的数据.如果事务B回滚,则A读取使用了错误的数据. 解决:如果一个事务在读的时候,禁止读取未提交的事务. 不可重复读 ...
mysql脏读解决方案_MySQL为什么可以解决脏读和不可重复读？
一般是通过锁机制,解决掉不可重复读和幻读的问题.是不是可以通过乐观锁的问题去解决不可重复读和幻读的问题,MySQL 采用的是 MVCC 机制来解决脏读.不可重复读的问题. MVCC 英文全称是 Mui ...
可重复读隔离级别如何解决脏读、不可重复读
我们都知道可重复读隔离级别可以解决脏读.不可重复读.那么具体是如何解决的呢? 下面先通过实验来演示可重复读能够解决脏读.不可重复读问题,然后解释具体的原因. 环境搭建 1. 建立两个session连接 ...
mysql脏读解决方案_mysql是如何解决脏读、不可重复读、幻读？
刚开始写博客.. 写的太low. 1.数据库的两种读,每种读读的数据版本不一样,所以也称为MVCC,即多版本并发控制 a) 快照读 select * from where xxx 这种形式的都是快照 ...
springcloud gateway 自定义 accesslog elk
大家好,我是烤鸭: 最近用 springcloud gateway 时,想使用类似 logback-access的功能,用来做数据统计和图表绘制等等,发现没有类似的功能,只能自己开发了. 环境: ...
SpringCloud Gateway 服务网关，限流
SpringCloud Gateway 提供了基于Redis 和lua脚本实现的令牌桶算法进行限流,即 RequestRateLimiterGatewayFilterFactory类,通过设置过滤器实 ...
drools动态配置规则_微服务实战系列（八）-网关springcloud gateway自定义规则
1. 场景描述先说明下项目中使用的网关是:springcloud gateway, 因需要给各个网关服务系统提供自定义配置路由规则,实时生效,不用重启网关(重启风险大),目前已实现:动态加载自定义路 ...
SpringCloud Gateway 通过redis实现限流
前言在高并发的系统中,往往需要在系统中做限流,一方面是为了防止大量的请求使服务器过载,导致服务不可用,另一方面是为了防止网络攻击. 常见的限流方式,比如Hystrix适用线程池隔离,超过线程池的负载 ...
SpringCloud Gateway 通过redis实现限流【SpringCloud系列8】
SpringCloud 大型系列课程正在制作中,欢迎大家关注与提意见. 程序员每天的CV 与板砖,也要知其所以然,本系列课程可以帮助初学者学习 SpringBooot 项目开发与 SpringCl ...

SpringCloud-Gateway自定义过滤器，解决body流不能重复读的问题

SpringCloud-Gateway自定义过滤器，解决body流不能重复读的问题相关推荐

最新文章

热门文章