Spring Cloud Gateway 响应值获取,网关日志打印

*/

@Component

public class RequestFilter implements GlobalFilter, Ordered {

@Override

public Mono filter(ServerWebExchange exchange, GatewayFilterChain chain) {

return DataBufferUtils.join(exchange.getRequest().getBody()).map(dataBuffer -> {

byte[] bytes = new byte[dataBuffer.readableByteCount()];

dataBuffer.read(bytes);

DataBufferUtils.release(dataBuffer);

return bytes;

}).defaultIfEmpty(new byte[0])

.doOnNext(bytes -> {

// 将请求体存入Attributes

exchange.getAttributes().put(Constant.REQUEST_BODY_CACHE, bytes);

})

.then(chain.filter(exchange));

}

@Override

public int getOrder() {

return -3;

}

}

2.响应过滤器,解析响应消息

import com.bzcst.bop.gateway.common.Constant;

import com.google.common.base.Charsets;

import lombok.extern.slf4j.Slf4j;

import org.reactivestreams.Publisher;

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.DataBufferFactory;

import org.springframework.core.io.buffer.DataBufferUtils;

import org.springframework.core.io.buffer.DefaultDataBufferFactory;

import org.springframework.http.HttpStatus;

import org.springframework.http.server.reactive.ServerHttpResponse;

import org.springframework.http.server.reactive.ServerHttpResponseDecorator;

import org.springframework.stereotype.Component;

import org.springframework.web.server.ServerWebExchange;

import reactor.core.publisher.Flux;

import reactor.core.publisher.Mono;

import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR;

/**

  • 响应过滤器

  • @author 向振华

  • @date 2021/01/30 10:01

*/

@Slf4j

@Component

public class ResponseFilter implements GlobalFilter, Ordered {

@Override

public Mono filter(ServerWebExchange exchange, GatewayFilterChain chain) {

try {

ServerHttpResponse originalResponse = exchange.getResponse();

DataBufferFactory bufferFactory = originalResponse.bufferFactory();

ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) {

@Override

public Mono writeWith(Publisher body) {

Object contentType = exchange.getAttribute(ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR);

if (HttpStatus.OK.equals(getStatusCode()) && body instanceof Flux

&& Constant.CONTENT_TYPES.contains(contentType)) {

Flux fluxBody = Flux.from(body);

return super.writeWith(fluxBody.buffer().map(dataBuffers -> {

DataBufferFactory dataBufferFactory = new DefaultDataBufferFactory();

DataBuffer join = dataBufferFactory.join(dataBuffers);

byte[] content = new byte[join.readableByteCount()];

join.read(content);

DataBufferUtils.release(join);

String responseMessage = new String(content, Charsets.UTF_8);

// responseMessage就是响应消息,这里可以打印出来或做其他业务处理

byte[] uppedContent = responseMessage.getBytes(Charsets.UTF_8);

return bufferFactory.wrap(uppedContent);

}));

} else {

return chain.filter(exchange);

}

}

};

return chain.filter(exchange.mutate().response(decoratedResponse).build());

} catch (Exception e) {

log.error(“网关响应解析异常:”, e);

return chain.filter(exchange);

}

}

@Override

public int getOrder() {

return -2;

}

}

3.接受过滤器,解决P 《一线大厂Java面试题解析+后端开发学习笔记+最新架构讲解视频+实战项目源码讲义》无偿开源 威信搜索公众号【编程进阶路】 OST请求报错:Only one connection receive subscriber allowed问题

import com.bzcst.bop.gateway.common.Constant;

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;

你可能感兴趣的:(Java,经验分享,架构,java)