gateway.version: 2.1.5.RELEASE
获取Request Body:
@Component
public class CacheBodyGlobalFilterimplements Ordered, GlobalFilter {
private static final LoggerLOGGER = LoggerFactory.getLogger(CacheBodyGlobalFilter.class);
@Autowired
private IdmInvocationSecurityMetadataSourcemetadataSource;
@Override
public Monofilter(ServerWebExchange exchange, GatewayFilterChain chain) {
if (exchange.getRequest().getHeaders().getContentType() ==null) {
return chain.filter(exchange);
}else {
return DataBufferUtils.join(exchange.getRequest().getBody()).flatMap(dataBuffer -> {
if (dataBuffer.readableByteCount() >0) {
exchange.getAttributes().put(ServerWebExchangeUtils.CACHED_REQUEST_BODY_ATTR, dataBuffer);
}
ServerHttpRequest decorator =new ServerHttpRequestDecorator(
exchange.getRequest()) {
@Override
public FluxgetBody() {
return Mono.fromSupplier(() -> {
// gateway 向服务转发请求时还会调用一次getBody第二次直接返回dataBuffer
if (exchange.getAttributeOrDefault(ServerWebExchangeUtils.CACHED_REQUEST_BODY_ATTR,
null) ==null) {
return dataBuffer;
}
// TODO: deal with Netty NettyDataBuffer pdb = (NettyDataBuffer)dataBuffer;
return pdb.factory().wrap(pdb.getNativeBuffer().retainedSlice());
}).flux();
}
};
return chain.filter(exchange.mutate().request(decorator).build());
});
}
}
@Override
public int getOrder() {
return -4;
}
}
//body
Flux dataBufferFlux = exchange.getRequest().getBody();
AtomicReference bodyRef =new AtomicReference<>();
// 缓存读取的request body信息
dataBufferFlux .subscribe(dataBuffer -> {
CharBuffer charBuffer = StandardCharsets.UTF_8.decode(dataBuffer.asByteBuffer());
DataBufferUtils.release(dataBuffer);
bodyRef.set(charBuffer.toString());
});
String body = bodyRef.get();
exchange.getAttributes().remove(ServerWebExchangeUtils.CACHED_REQUEST_BODY_ATTR);
获取Response Body:
@Component
public class GatewayResponseFilterimplements GlobalFilter, Ordered {
private static final LoggerLOGGER = LoggerFactory.getLogger(GatewayResponseFilter.class);
@Override
public int getOrder() {
return -2;
}
@Override
public Monofilter(ServerWebExchange exchange, GatewayFilterChain chain) {
//获取response的 返回数据
ServerHttpResponse originalResponse = exchange.getResponse();
DataBufferFactory bufferFactory = originalResponse.bufferFactory();
ServerHttpResponseDecorator response =new ServerHttpResponseDecorator(originalResponse) {
@Override
public MonowriteWith(Publisher body) {
if (getStatusCode().equals(HttpStatus.OK) && bodyinstanceof Flux) {
// 获取ContentType，判断是否返回JSON格式数据
String originalResponseContentType =exchange.getAttribute(ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR);
if (org.apache.commons.lang.StringUtils.isNotBlank(originalResponseContentType) && originalResponseContentType.contains("application/json")) {
Flux fluxBody = Flux.from(body);
return super.writeWith(fluxBody.buffer().map(dataBuffers -> {//解决返回体分段传输
DataBufferFactory dataBufferFactory =new DefaultDataBufferFactory();
String responseData =null;
try {
DataBuffer dataBuffer = dataBufferFactory.join(dataBuffers);
byte[] content =new byte[dataBuffer.readableByteCount()];
dataBuffer.read(content);
DataBufferUtils.release(dataBuffer);
responseData =new String(content, "UTF-8");
}catch (UnsupportedEncodingException e) {
LOGGER.info("获取response body，失败原因：{}", Throwables.getStackTraceAsString(e));
}
// 二次处理（加密/过滤等）如果不需要做二次处理可直接跳过下行
// 。。。。body处理
byte[] uppedContent =new String(responseData.getBytes(), Charset.forName("UTF-8")).getBytes();
originalResponse.getHeaders().setContentLength(uppedContent.length);
return bufferFactory.wrap(uppedContent);
}));
}
}
return super.writeWith(body);
}
@Override
public MonowriteAndFlushWith(Publisher> body) {
return writeWith(Flux.from(body).flatMapSequential(p -> p));
}
};
return chain.filter(exchange.mutate().response(response).build());
}