springcloud gateway读取请求body

xiawenquan

已于 2022-03-10 13:33:14 修改

阅读量907

点赞数 1

文章标签： spring cloud gateway java

于 2022-03-10 13:31:14 首次发布

本文链接：https://blog.csdn.net/xiawenquan/article/details/123399388

版权

在SpringCloud Gateway中，由于请求Body只能读取一次，因此读取Body内容需特殊处理。本文档详细阐述了如何在该框架下正确读取和处理HTTP请求的Body数据。

摘要由CSDN通过智能技术生成

request body中的内容只允许读取一次，若是多次读取会报错，本章中简单介绍了如何在springcloud gateway中读取body中的内容。

package com.shouwei.gateway.filter;

import io.netty.buffer.ByteBufAllocator;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.ResolvableType;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.core.io.buffer.NettyDataBufferFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.InvalidMediaTypeException;
import org.springframework.http.MediaType;
import org.springframework.http.codec.HttpMessageReader;
import org.springframework.http.codec.ServerCodecConfigurer;
import org.springframework.http.codec.multipart.Part;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.reactive.function.server.HandlerStrategies;
import org.springframework.web.reactive.function.server.ServerRequest;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * 全局拦截器，作用所有的微服务
 * 1. 对请求的API调用过滤，记录接口的请求时间，方便日志审计、告警、分析等运维操作
 * 2. 后期可以扩展对接其他日志系统
 */
@Component
@Slf4j
public class ApiRequestFilter implements GlobalFilter, Ordered {

    private static final String START_TIME = "startTime";

    private static final String GATEWAY_ORIGINAL_REQUEST_URL = "org.springframework.cloud.gateway.support.ServerWebExchangeUtils.gatewayOriginalRequestUrl";


    /**
     * default HttpMessageReader
     */
    private static final List<HttpMessageReader<?>> messageReaders = HandlerStrategies.withDefaults().messageReaders();

    private static final ResolvableType MULTIPART_DATA_TYPE = ResolvableType.forClassWithGenerics(MultiValueMap.class, String.class, Part.class);

    private static final Mono<MultiValueMap<String, Part>> EMPTY_MULTIPART_DATA = Mono.just(CollectionUtils.unmodifiableMultiValueMap(new LinkedMultiValueMap<String, Part>(0))).cache();

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();

        if (request.getMethod() == HttpMethod.GET) {
            // get请求 处理参数
            return handleGetMethod(exchange, chain, request);
        }

        if (request.getMethod() == HttpMethod.POST) {
            // post请求 处理参数
            return handlePostMethod(exchange, chain, request);
        }

        return c