Decode

package com.netty.demo.filter;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import io.netty.buffer.ByteBufAllocator;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
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.HttpCookie;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.RequestPath;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.http.server.reactive.SslInfo;
import org.springframework.stereotype.Component;
import org.springframework.util.MultiValueMap;
import org.springframework.util.ObjectUtils;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.List;

@Slf4j
@Component
public class Decode implements GlobalFilter, Ordered {

@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
    ServerHttpRequest request = exchange.getRequest();
    HttpHeaders headers = request.getHeaders();
    List<String> json = headers.get("Content-Type");
    if (!ObjectUtils.isEmpty(json) && json.contains("application/json") ){
        ServerHttpRequestDecorator serverHttpRequestDecorator = requestDecorator(exchange);
        return chain.filter(exchange.mutate().request(serverHttpRequestDecorator).build());
    }

    return chain.filter(exchange);
}

private ServerHttpRequestDecorator requestDecorator(ServerWebExchange exchange) {
    ServerHttpRequestDecorator requestDecorator = new ServerHttpRequestDecorator(exchange.getRequest()) {
        @Override
        public Flux<DataBuffer> getBody() {
            Flux<DataBuffer> body = super.getBody();
            return body.map(dataBuffer -> {
                byte[] content = new byte[dataBuffer.readableByteCount()];
                dataBuffer.read(content);
                //释放掉内存
                DataBufferUtils.release(dataBuffer);
                //request body的json格式数据
                String bodyJson = new String(content, Charset.forName("UTF-8"));
                //转化成json对象
                JSONObject jsonObject = JSON.parseObject(bodyJson);
                String data = jsonObject.getString("data");

                String requestBody = "{}";
                String decrKey = null;
                try {
                    decrKey = new String(Base64.getDecoder().decode(data));
                } catch (Exception e) {
                    log.error("key异常");
                    return getDataBuffer();
                }

                requestBody = decrKey;
                log.info(">>>> requestBody:{}", requestBody);

                //转成字节
                byte[] bytes = requestBody.getBytes();
                NettyDataBufferFactory nettyDataBufferFactory = new NettyDataBufferFactory(ByteBufAllocator.DEFAULT);
                DataBuffer buffer = nettyDataBufferFactory.allocateBuffer(bytes.length);
                buffer.write(bytes);
                return buffer;
            });
        }

        //复写getHeaders方法
        @Override
        public HttpHeaders getHeaders() {
            HttpHeaders httpHeaders = new HttpHeaders();
            httpHeaders.putAll(super.getHeaders());
            //由于修改了请求体的body，导致content-length长度不确定，因此需要删除原先的content-length
            httpHeaders.remove(HttpHeaders.CONTENT_LENGTH);
            httpHeaders.set(HttpHeaders.TRANSFER_ENCODING, "chunked");
            return httpHeaders;
        }

    };
    return requestDecorator;
}

private DataBuffer getDataBuffer() {
    String requestBody;
    JSONObject errJson = new JSONObject();
    errJson.put("code", 500);
    errJson.put("msg", "key错误");
    requestBody = errJson.toJSONString();
    byte[] bytes = requestBody.getBytes();
    NettyDataBufferFactory nettyDataBufferFactory = new NettyDataBufferFactory(ByteBufAllocator.DEFAULT);
    DataBuffer buffer = nettyDataBufferFactory.allocateBuffer(bytes.length);
    buffer.write(bytes);
    return buffer;
}

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

}