Gateway网关自定义拦截器的不可重复读取数据

       最近在开发gateway网关时，通过自定义拦截器对某些接口的数据进行处理，发现，无法读取到数据。经过查询，发现在Spring5的webflux编程或者普通web编程中，只能从request中获取body一次，后面无法再获取。参考网上的方法先通过全局过滤器把body先缓存起来。

这个网上有很多例子：

/**
 * @author zhong
 * @date 2022/1/6 - 15:34
 * 
 */
@Configuration
public class CacheBodyGlobalFilter implements Ordered, GlobalFilter {

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        //判断是否时认证接口
        if (!AuthUtils.isAuthorPath(exchange.getRequest())) {
            //不是认证接口
            return chain.filter(exchange);
        } else {
            //指定接口可建数据从新读出构建
            return DataBufferUtils.join(exchange.getRequest().getBody())
                    .flatMap(dataBuffer -> {
                        DataBufferUtils.retain(dataBuffer);
                        Flux<DataBuffer> cachedFlux = Flux
                                .defer(() -> Flux.just(dataBuffer.slice(0, dataBuffer.readableByteCount())));
                        ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {
                            @Override
                            public Flux<DataBuffer> getBody() {
                                return cachedFlux;
                            }
                        };
                        //exchange.getAttributes().put(CACHE_REQUEST_BODY_OBJECT_KEY, cachedFlux);
                        return chain.filter(exchange.mutate().request(mutatedRequest).build());
                    });
        }
    }

    @Override
    public int getOrder() {
        return Ordered.HIGHEST_PRECEDENCE;
    }
}

       CacheBodyGlobalFilter这个全局过滤器的目的就是把原有的request请求中的body内容读出来，并且使用ServerHttpRequestDecorator这个请求装饰器对request进行包装，重写getBody方法，并把包装后的请求放到过滤器链中传递下去。这样后面的过滤器中再使用exchange.getRequest().getBody()来获取body时，实际上就是调用的重载后的getBody方法，获取的最先已经缓存了的body数据。这样就能够实现body的多次读取了。
       这个过滤器的order设置的是Ordered.HIGHEST_PRECEDENCE，即最高优先级的过滤器。优先级设置这么高的原因是某些系统内置的过滤器可能也会去读body。

      但是在测试中我发现，如果我在请求接口中如果没有body内容，就会导致程序只能执行CacheBodyGlobalFilter这一个拦截器，而无法执行其他拦截器(自定义的和默认的)。而且接口返回200，这和我的预期时不一致的。

     通过测试发现，原因是，按照以上代码执行，如果是body为null，Conten-Type也为空，所以没有执行代码中的flatMap()方法.所以也就没有执行后面的调用链。

chain.filter(exchange.mutate().request(mutatedRequest).build());

解决办法，在获取到数据流时，如果数据流为null，我们可以构建一个空的数据流，这也也就能执行我们后面的拦截器。

利用Flux.defaultIfEmpty(defaultDataBuffer);的方法可以实现这个功能。

public final Flux<T> defaultIfEmpty(T defaultV) {
   return onAssembly(new FluxDefaultIfEmpty<>(this, defaultV));
} 

代码如下： 

@Configuration
public class CacheBodyGlobalFilter implements Ordered, GlobalFilter {

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        //判断是否时认证接口
        if (!AuthUtils.isAuthorPath(exchange.getRequest())) {
            //不是认证接口
            return chain.filter(exchange);
        } else {
            //指定接口可建数据从新读出构建

            //当body中没有缓存时，只会执行这一个拦截器， 原因是fileMap中的代码没有执行，所以需要在body为空时构建一个空的缓存
            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()) {
                            @Override
                            public Flux<DataBuffer> getBody() {
                                return cachedFlux;
                            }
                        };
                        //exchange.getAttributes().put(CACHE_REQUEST_BODY_OBJECT_KEY, cachedFlux);
                        return chain.filter(exchange.mutate().request(mutatedRequest).build());
                    });
        }
    }

    @Override
    public int getOrder() {
        return Ordered.HIGHEST_PRECEDENCE;
    }
}

当然这个是在我们的拦截的接口存在没有body，但是也要执行后面的拦截器的情况下。如果说接口必须要有body。那我们可以在CacheBodyGlobalFilter 这拦截器中直接返回指定的内容。这个我还没有测试过。