api监控指标

miaoguoxin
1 parent 45276d37
Showing 2 changed files with 26 additions and 2 deletions
gateway-core/src/main/java/com/diligrp/xtrade/core/filters/global/CacheRequestBodyGlobalFilter.java
gateway-core/src/main/java/com/diligrp/xtrade/core/support/ApiMetricsInfo.java
 package com.diligrp.xtrade.core.filters.global;
 import com.diligrp.xtrade.core.common.constant.GatewayConst;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.cloud.gateway.filter.GatewayFilterChain;
 import org.springframework.cloud.gateway.filter.GlobalFilter;
 import org.springframework.cloud.gateway.filter.factory.rewrite.CachedBodyOutputMessage;
@@ -28,7 +29,7 @@ import java.util.List;
 /**
  * @Auther: miaoguoxin
  * @Date: 2020/4/13 11:49
- * @Description: 缓存请求body的过滤器,参考
+ * @Description: 缓存请求body的过滤器, 参考
  * {@link ModifyRequestBodyGatewayFilterFactory}
  */
 @Component
@@ -38,6 +39,10 @@ public class CacheRequestBodyGlobalFilter implements GlobalFilter, Ordered {
     @Override
     public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
+        String contentType = exchange.getRequest().getHeaders().getFirst(HttpHeaders.CONTENT_TYPE);
+        if (StringUtils.isNotBlank(contentType) && !contentType.startsWith("multipart")) {
+            return chain.filter(exchange);
+        }
         ServerRequest serverRequest = ServerRequest.create(exchange,
                 messageReaders);
         Mono<String> modifiedBody = serverRequest.bodyToMono(String.class)
@@ -63,7 +68,7 @@ public class CacheRequestBodyGlobalFilter implements GlobalFilter, Ordered {
     }
     private ServerHttpRequestDecorator decorate(ServerWebExchange exchange, HttpHeaders headers,
-                                        CachedBodyOutputMessage outputMessage) {
+                                                CachedBodyOutputMessage outputMessage) {
         return new ServerHttpRequestDecorator(exchange.getRequest()) {
             @Override
             public HttpHeaders getHeaders() {
@@ -10,10 +10,14 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.cloud.gateway.route.Route;
 import org.springframework.cloud.gateway.support.ServerWebExchangeUtils;
+import org.springframework.http.HttpHeaders;
 import org.springframework.web.server.ServerWebExchange;
 import java.net.URI;
 import java.time.LocalDateTime;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
 /**
  * @Auther: miaoguoxin
@@ -30,6 +34,8 @@ public class ApiMetricsInfo {
     private String serviceId;
     /**目标地址*/
     private String url;
+    /**请求头*/
+    private String requestHeaders;
     /**请求体*/
     private String requestBody;
     /**异常信息*/
@@ -44,6 +50,7 @@ public class ApiMetricsInfo {
                           String message,
                           String serviceId,
                           String url,
+                          String requestHeaders,
                           String requestBody,
                           Throwable throwable,
                           Long executeTime) {
@@ -51,6 +58,7 @@ public class ApiMetricsInfo {
         this.message = message;
         this.serviceId = serviceId;
         this.url = url;
+        this.requestHeaders = requestHeaders;
         this.requestBody = requestBody;
         this.throwable = throwable;
         this.executeTime = executeTime;
@@ -72,6 +80,7 @@ public class ApiMetricsInfo {
                         message.getMessage(),
                         uri != null ? uri.toString() : "unknown",
                         route != null ? route.getId() : "unknown",
+                        getRequestHeaderJson(exchange.getRequest().getHeaders()),
                         requestBody,
                         throwable,
                         time);
@@ -83,6 +92,12 @@ public class ApiMetricsInfo {
         }
     }
+    private static String getRequestHeaderJson(HttpHeaders headers){
+        Map<String, List<String>> headerMap = new LinkedHashMap<>();
+        headerMap.put(HttpHeaders.CONTENT_TYPE,headers.get(HttpHeaders.CONTENT_TYPE));
+        return JsonUtils.toJsonString(headerMap);
+    }
+
     public Integer getCode() {
         return code;
     }
@@ -103,6 +118,10 @@ public class ApiMetricsInfo {
         return requestBody;
     }
+    public String getRequestHeaders() {
+        return requestHeaders;
+    }
+
     public Throwable getThrowable() {
         return throwable;
     }