Only one connection receive subscriber allowed.

安迪范德

已于 2022-08-12 16:24:13 修改

阅读量1.4k

点赞数

分类专栏： springboot 文章标签： java log4j apache

于 2022-08-12 16:18:42 首次发布

本文链接：https://blog.csdn.net/qq_37280992/article/details/126306507

版权

springboot 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这个filter 中报Only one connection receive subscriber allowed.

主要是body体为null的情况

@Component
public class DecryptionFilter implements GlobalFilter, Ordered {

@SneakyThrows
@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {

解决办法：
在这里插入图片描述
参考：https://blog.csdn.net/weixin_37644979/article/details/115264027

完全代码：

package com.riocento.filter;


import com.alibaba.fastjson.JSONObject;
import io.netty.buffer.ByteBufAllocator;

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.logging.log4j.util.Strings;
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.NettyDataBufferFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
import org.springframework.stereotype.Component;
import org.springframework.util.MultiValueMap;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.CoreSubscriber;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.Date;
import java.util.Iterator;

/**
 * 解密过滤器
 * 对请求的request进行解密
 */
@Slf4j
@Component
public class DecryptionFilter implements GlobalFilter, Ordered {

    @SneakyThrows
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        log.info("request解密处理");
        //源request对象
        ServerHttpRequest request = exchange.getRequest();
        //获取请求方式
        HttpMethod method = request.getMethod();
        URI uri = request.getURI();
        log.info("请求方式:"+method);
        log.info("请求地址:"+uri);
        MultiValueMap<String, String> queryParams = request.getQueryParams();
        log.info("所有参数:"+queryParams);
        //是否为加密参数
        Boolean isEncrypt = false;
        // GET请求解密
        if(HttpMethod.GET == method){
            if(queryParams.containsKey("isEncrypt")){
                isEncrypt = Boolean.valueOf(queryParams.get("isEncrypt").get(0));
            }
            if(isEncrypt){
                //加密参数
                String enParams = queryParams.get("enParams").get(0);
                //将参数解密
                String paramsStr = AESUtil.decrypt(enParams);
                //格式化参数
                String urlParams = formatUrlParams(paramsStr);
                //重新创建request
                URI plaintUrl = new URI(uri.getScheme(), uri.getAuthority(), uri.getPath(), urlParams, uri.getFragment());
                ServerHttpRequest newRequest = request.mutate().uri(plaintUrl).build();
                return chain.filter(exchange.mutate().request(newRequest).build());
            }
        }
        //POST和PUT请求解密
        else if(HttpMethod.POST == method
                || HttpMethod.PUT == method) {
            if(queryParams.containsKey("isEncrypt")){
                isEncrypt = Boolean.valueOf(queryParams.get("isEncrypt").get(0));
            }
            if(isEncrypt){
                URI plaintUrl = null;
                //处理url参数
                if(queryParams.size() > 0 && queryParams.containsKey("enParams")){
                    //加密参数
                    String urlEnParams = queryParams.get("enParams").get(0);
                    //将参数解密
                    String urlParamsStr = AESUtil.decrypt(urlEnParams);
                    //格式化参数
                    String urlParams = formatUrlParams(urlParamsStr);
                    //重新创建uri
                    plaintUrl = new URI(uri.getScheme(), uri.getAuthority(), uri.getPath(), urlParams, uri.getFragment());
                }else{
                    plaintUrl = uri;
                }

                //处理表单参数
                //获取表单信息
                Flux<DataBuffer> dataBufferFlux = request.getBody();
                //获取body中的数据
                String strBody = FilterRequestResponseUtil.resolveBodyFromRequest(dataBufferFlux);
                //获取加密参数
                String enParams = getEnParams(strBody);
                //将参数解密
                String paramsStr = AESUtil.decrypt(enParams);

                ServerHttpRequest newRequest = request.mutate().uri(plaintUrl).build();
                DataBuffer bodyDataBuffer = stringBuffer(paramsStr);
                if(paramsStr == null) {
                    newRequest = new ServerHttpRequestDecorator(newRequest) {
                        @Override
                        public Flux<DataBuffer> getBody() {
                            return Flux.just(bodyDataBuffer);
                        }
                    };
                    // 构建新的请求头
                    HttpHeaders headers = new HttpHeaders();
                    headers.putAll(exchange.getRequest().getHeaders());
                    // 重新设置CONTENT_LENGTH
                    int length = 0;
                    headers.remove(HttpHeaders.CONTENT_LENGTH);
                    headers.setContentLength(length);
                    newRequest = new ServerHttpRequestDecorator(newRequest) {
                        @Override
                        public HttpHeaders getHeaders() {
                            return headers;
                        }
                    };
                    return chain.filter(exchange.mutate().request(newRequest).build());
                }
                Flux<DataBuffer> bodyFlux = Flux.just(bodyDataBuffer);
                newRequest = new ServerHttpRequestDecorator(newRequest) {
                    @Override
                    public Flux<DataBuffer> getBody() {
                        return bodyFlux;
                    }
                };
                // 构建新的请求头
                HttpHeaders headers = new HttpHeaders();
                headers.putAll(exchange.getRequest().getHeaders());
                // 重新设置CONTENT_LENGTH
                int length = paramsStr.getBytes().length;
                headers.remove(HttpHeaders.CONTENT_LENGTH);
                headers.setContentLength(length);
                newRequest = new ServerHttpRequestDecorator(newRequest) {
                    @Override
                    public HttpHeaders getHeaders() {
                        return headers;
                    }
                };
                return chain.filter(exchange.mutate().request(newRequest).build());
            }
        }
        //DELETE等请求方式不做处理
        return chain.filter(exchange);
    }

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

    /**
     * 格式化url参数
     * @param paramsStr 字符串参数
     */
    private String formatUrlParams(String paramsStr){
        StringBuilder stbResult = new StringBuilder();
        if(Strings.isNotEmpty(paramsStr)){
            if(paramsStr.startsWith("{")){
                JSONObject paramsJSON = JSONObject.parseObject(paramsStr);
                Iterator<String> iterator = paramsJSON.keySet().iterator();
                while (iterator.hasNext()) {
                    String key = iterator.next();
                    String value = paramsJSON.getString(key);
                    if(stbResult.length() > 0){
                        stbResult.append("&");
                    }
                    stbResult.append(key).append("=").append(value);
                }
            }else{
                stbResult.append(paramsStr);
            }
        }

        return stbResult.toString();
    }

    /**
     * 获取加密字符串参数
     * @return
     */
    private String getEnParams(String paramsStr){
        String result = null;
        if(Strings.isNotEmpty(paramsStr)){
            if(paramsStr.startsWith("{")){
                //将字符串转换为JSON格式
                JSONObject jsonBody = JSONObject.parseObject(paramsStr);
                result = jsonBody.getString("enParams");
            }else if(paramsStr.indexOf("=") > 0){
                String[] paramsArr = paramsStr.split("=");
                result = paramsArr[1];
            }
        }
        return  result;
    }

    private DataBuffer stringBuffer(String value) {
        if(value == null) {
            NettyDataBufferFactory nettyDataBufferFactory = new NettyDataBufferFactory(ByteBufAllocator.DEFAULT);
            DataBuffer buffer = nettyDataBufferFactory.allocateBuffer(0);
            return buffer;
        }
        byte[] bytes = value.getBytes(StandardCharsets.UTF_8);
        NettyDataBufferFactory nettyDataBufferFactory = new NettyDataBufferFactory(ByteBufAllocator.DEFAULT);
        DataBuffer buffer = nettyDataBufferFactory.allocateBuffer(bytes.length);
        buffer.write(bytes);
        return buffer;
    }
}