Spring Cloud Gateway -- ResponseBody修改（解决ResponseBody不完整的问题）

电玩女神 2022-04-25 10:34 259阅读 0赞

### Spring Cloud Gateway -- ResponseBody修改（解决ResponseBody不完整的问题） ###

*  前言
 *  Response修改

# 前言 #

由于对方接口传输数据可能是xml，可能是json的，但是我希望前端获取到的响应是json格式的，所以需要在在响应中修改一下。

# Response修改 #

由于接口返回数据较多，采用 fluxBody.map(dataBuffer -> 这个方式获取数据是不完整的。其实是多次调用了这个方法。

return super.writeWith(fluxBody.map(dataBuffer -> { 
                            // probably should reuse buffers
                            byte[] content = new byte[dataBuffer.readableByteCount()];
                            dataBuffer.read(content);
                            //释放掉内存
                            DataBufferUtils.release(dataBuffer);
                            String s = new String(content, Charset.forName("UTF-8"));
                            //TODO，s就是response的值，想修改、查看就随意而为了
                            byte[] uppedContent = new String(content, Charset.forName("UTF-8")).getBytes();
                            return bufferFactory.wrap(uppedContent);
                        }));

采用循环调用之后 再将内容拼接好就能获取到完整的ResponseBody。  
这里有一个小问题由于不知道修改完的Body长度，只能采用了分块编码这种代替了CONTENT\_LENGTH。  
尝试了一下好像是触发getHeaders()这个方法，再执行writeWith()这个方法。导致了计算不出来CONTENT\_LENGTH。  
（不知道有没有人能解决这个问题）

@Override
        public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { 
            ServerHttpResponse originalResponse = exchange.getResponse();
            // 定义新的消息头
            HttpHeaders headers = new HttpHeaders();
            headers.putAll(exchange.getResponse().getHeaders());
    
            ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) { 
                @Override
                public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) { 
                    if (getStatusCode().equals(HttpStatus.OK) && body instanceof Flux) { 
                        Flux<? extends DataBuffer> fluxBody = Flux.from(body);
                        return super.writeWith(fluxBody.buffer().map(dataBuffers -> { 
                            List<String> list = new ArrayList<String>();
                            dataBuffers.forEach(dataBuffer -> { 
                                byte[] content = new byte[dataBuffer.readableByteCount()];
                                dataBuffer.read(content);
                                DataBufferUtils.release(dataBuffer);
                                try { 
                                    list.add(new String(content, "utf-8"));
                                } catch (IOException e) { 
                                    e.printStackTrace();
                                }
                            });
                            String s = listToString(list);
                            if(isXML(s)){ 
                                Map m = json2map(xml2jsonString(s,"").get("data").toString(),"");
                                s = map2Json(m);
                            }
                            int length = s.getBytes().length;
                            headers.setContentLength(length);
                            return bufferFactory().wrap(s.getBytes());
                        }));
                    }
                    return super.writeWith(body);
                };
                @Override
                public HttpHeaders getHeaders() { 
                    HttpHeaders httpHeaders = new HttpHeaders();
                    httpHeaders.putAll(super.getHeaders());
                    //由于修改了请求体的body，导致content-length长度不确定，因此使用分块编码
                    httpHeaders.remove(HttpHeaders.CONTENT_LENGTH);
                    httpHeaders.set(HttpHeaders.TRANSFER_ENCODING, "chunked");
                    return httpHeaders;
                }
            };
    
            // replace response with decorator
            return chain.filter(exchange.mutate().response(decoratedResponse).build());
        }