从零搭建Spring Cloud Gateway网关(二)—— 打印请求响应日志 (3)

工具类也贴下:

package com.lifengdi.gateway.log; import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.ObjectMapper; import com.lifengdi.gateway.constant.HeaderConstant; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang.StringUtils; import org.springframework.core.io.buffer.DataBufferUtils; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.server.reactive.ServerHttpRequest; import org.springframework.lang.NonNull; import org.springframework.lang.Nullable; import reactor.core.publisher.Mono; import java.nio.charset.Charset; import java.nio.charset.StandardCharsets; import java.util.Objects; import java.util.concurrent.atomic.AtomicReference; @Slf4j public class LogHelper { private final static ObjectMapper objectMapper = new ObjectMapper(); /** * Log转JSON * @param dto Log * @return JSON字符串 */ public static String toJsonString(@NonNull Log dto) { try { return objectMapper.writeValueAsString(dto); } catch (JsonProcessingException e) { log.error("Log转换JSON异常", e); return null; } } /** * 根据MediaType获取字符集,如果获取不到,则默认返回<tt>UTF_8</tt> * @param mediaType MediaType * @return Charset */ public static Charset getMediaTypeCharset(@Nullable MediaType mediaType) { if (Objects.nonNull(mediaType) && mediaType.getCharset() != null) { return mediaType.getCharset(); } else { return StandardCharsets.UTF_8; } } /** * 记录日志(后期可扩展为通过MQ将日志发送到ELK系统) * @param dto Log * @return Mono.empty() */ public static Mono<Void> doRecord(Log dto) { log.info(toJsonString(dto)); return Mono.empty(); } /** * 从HttpHeaders获取请求开始时间 * <p> * 要求请求头中必须要有参数{@link HeaderConstant#START_TIME_KEY},否则将返回当前时间戳 * </p> * @param headers HttpHeaders请求头 * @return 开始时间时间戳(Mills) */ public static long getStartTime(HttpHeaders headers) { String startTimeStr = headers.getFirst(HeaderConstant.START_TIME_KEY); return StringUtils.isNotBlank(startTimeStr) ? Long.parseLong(startTimeStr) : System.currentTimeMillis(); } /** * 根据HttpHeaders请求头获取请求执行时间 * <p> * 要求请求头中必须要有参数{@link HeaderConstant#START_TIME_KEY} * </p> * @param headers HttpHeaders请求头 * @return 请求执行时间 */ public static long getHandleTime(HttpHeaders headers) { String startTimeStr = headers.getFirst(HeaderConstant.START_TIME_KEY); long startTime = StringUtils.isNotBlank(startTimeStr) ? Long.parseLong(startTimeStr) : System.currentTimeMillis(); return System.currentTimeMillis() - startTime; } /** * 读取请求体内容 * @param request ServerHttpRequest * @return 请求体 */ public static String readRequestBody(ServerHttpRequest request) { HttpHeaders headers = request.getHeaders(); MediaType mediaType = headers.getContentType(); String method = request.getMethodValue().toUpperCase(); if (Objects.nonNull(mediaType) && mediaType.equals(MediaType.MULTIPART_FORM_DATA)) { return "上传文件"; } else { if (method.equals("GET")) { if (!request.getQueryParams().isEmpty()) { return request.getQueryParams().toString(); } return null; } else { AtomicReference<String> bodyString = new AtomicReference<>(); request.getBody().subscribe(buffer -> { byte[] bytes = new byte[buffer.readableByteCount()]; buffer.read(bytes); DataBufferUtils.release(buffer); bodyString.set(new String(bytes, getMediaTypeCharset(mediaType))); }); return bodyString.get(); } } } /** * 判断是否是上传文件 * @param mediaType MediaType * @return Boolean */ public static boolean isUploadFile(@Nullable MediaType mediaType) { if (Objects.isNull(mediaType)) { return false; } return mediaType.equals(MediaType.MULTIPART_FORM_DATA) || mediaType.equals(MediaType.IMAGE_GIF) || mediaType.equals(MediaType.IMAGE_JPEG) || mediaType.equals(MediaType.IMAGE_PNG) || mediaType.equals(MediaType.MULTIPART_MIXED); } } 打印响应报文

响应报文需要在Spring重写了响应体之后才能获取到,所以对filter的执行顺序有要求,需要在
NettyWriteResponseFilter.WRITE_RESPONSE_FILTER_ORDER之前执行。代码如下:

package com.lifengdi.gateway.filter; import com.lifengdi.gateway.constant.HeaderConstant; import com.lifengdi.gateway.constant.OrderedConstant; import com.lifengdi.gateway.log.Log; import com.lifengdi.gateway.log.LogHelper; import com.lifengdi.gateway.utils.IpUtils; import io.netty.buffer.UnpooledByteBufAllocator; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang.StringUtils; 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.NettyDataBufferFactory; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpStatus; import org.springframework.http.MediaType; 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.ServerHttpResponseDecorator; import org.springframework.stereotype.Component; import org.springframework.web.reactive.function.server.HandlerStrategies; import org.springframework.web.reactive.function.server.ServerRequest; import org.springframework.web.server.ServerWebExchange; import reactor.core.publisher.Flux; import reactor.core.publisher.Mono; import java.net.URI; import java.nio.charset.Charset; import java.util.Objects; import java.util.concurrent.atomic.AtomicReference; /** * 请求响应日志打印 */ @Component @Slf4j public class ResponseLogFilter implements GlobalFilter, Ordered { @Override public int getOrder() { return OrderedConstant.LOGGING_FILTER; } @Override public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { try { ServerHttpRequest request = exchange.getRequest(); ServerRequest serverRequest = ServerRequest.create(exchange, HandlerStrategies.withDefaults().messageReaders()); URI requestUri = request.getURI(); String uriQuery = requestUri.getQuery(); HttpHeaders headers = request.getHeaders(); MediaType mediaType = headers.getContentType(); String schema = requestUri.getScheme(); String method = request.getMethodValue().toUpperCase(); // 只记录http、https请求 if ((!"http".equals(schema) && !"https".equals(schema))) { return chain.filter(exchange); } final AtomicReference<String> requestBody = new AtomicReference<>();// 原始请求体 // 排除流文件类型,比如上传的文件contentType.contains("multipart/form-data") if (Objects.nonNull(mediaType) && LogHelper.isUploadFile(mediaType)) { requestBody.set("上传文件"); return chain.filter(exchange); } else { if (method.equals("GET")) { if (StringUtils.isNotBlank(uriQuery)) { requestBody.set(uriQuery); } } else if (headers.getContentLength() > 0){ return serverRequest.bodyToMono(String.class).flatMap(reqBody -> { requestBody.set(reqBody); // 重写原始请求 ServerHttpRequestDecorator requestDecorator = new ServerHttpRequestDecorator(exchange.getRequest()) { @Override public HttpHeaders getHeaders() { HttpHeaders httpHeaders = new HttpHeaders(); httpHeaders.putAll(super.getHeaders()); return httpHeaders; } @Override public Flux<DataBuffer> getBody() { NettyDataBufferFactory nettyDataBufferFactory = new NettyDataBufferFactory(new UnpooledByteBufAllocator(false)); DataBuffer bodyDataBuffer = nettyDataBufferFactory.wrap(reqBody.getBytes()); return Flux.just(bodyDataBuffer); // return Flux.just(reqBody).map(bx -> exchange.getRequest().bufferFactory().wrap(bx.getBytes())); } }; ServerHttpResponseDecorator responseDecorator = getServerHttpResponseDecorator(exchange, requestBody); return chain.filter(exchange.mutate() .request(requestDecorator) .response(responseDecorator) .build()); }); } ServerHttpResponseDecorator decoratedResponse = getServerHttpResponseDecorator(exchange, requestBody); return chain.filter(exchange.mutate() .response(decoratedResponse) .build()); } } catch (Exception e) { log.error("请求响应日志打印出现异常", e); return chain.filter(exchange); } } private ServerHttpResponseDecorator getServerHttpResponseDecorator(ServerWebExchange exchange, AtomicReference<String> requestBody) { // 获取response的返回数据 ServerHttpResponse originalResponse = exchange.getResponse(); DataBufferFactory bufferFactory = originalResponse.bufferFactory(); HttpStatus httpStatus = originalResponse.getStatusCode(); ServerHttpRequest request = exchange.getRequest(); URI requestUri = request.getURI(); String uriQuery = requestUri.getQuery(); String url = requestUri.getPath() + (StringUtils.isNotBlank(uriQuery) ? "?" + uriQuery : ""); HttpHeaders headers = request.getHeaders(); String method = request.getMethodValue().toUpperCase(); String requestId = headers.getFirst(HeaderConstant.REQUEST_ID); // 封装返回体 return new ServerHttpResponseDecorator(originalResponse) { @Override public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) { if (body instanceof Flux) { Flux<? extends DataBuffer> fluxBody = Flux.from(body); return super.writeWith(fluxBody.buffer().map(dataBuffers -> { DataBuffer join = bufferFactory.join(dataBuffers); byte[] content = new byte[join.readableByteCount()]; join.read(content); DataBufferUtils.release(join); Charset charset = LogHelper.getMediaTypeCharset(originalResponse.getHeaders().getContentType()); String responseBody = new String(content, charset); long handleTime = LogHelper.getHandleTime(headers); Log logDTO = new Log(Log.TYPE.RESPONSE); logDTO.setLevel(Log.LEVEL.INFO); logDTO.setRequestUrl(url); logDTO.setRequestBody(requestBody.get()); logDTO.setResponseBody(responseBody); logDTO.setRequestMethod(method); if (Objects.nonNull(httpStatus)) { logDTO.setStatus(httpStatus.value()); } logDTO.setHandleTime(handleTime); logDTO.setRequestId(requestId); logDTO.setIp(IpUtils.getClientIp(request)); exchange.getSession().subscribe(webSession -> { logDTO.setSessionId(webSession.getId()); }); log.info("url:{},method:{},请求内容:{},响应内容:{},status:{},handleTime:{},requestId:{}", url, method, requestBody.get(), responseBody, httpStatus, handleTime, requestId); log.info(LogHelper.toJsonString(logDTO)); return bufferFactory.wrap(content); })); } return super.writeWith(body); } }; } }

代码已上传到git上,需要的可以去看看。

git代码地址:https://github.com/lifengdi/spring-cloud-gateway-demo

原文地址:https://www.lifengdi.com/archives/article/1778

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zwwfgg.html