NEZ-1563 fix: 解决因 /actuator/prometheus 接口 http_client_requests 指标 uri 带有时间戳参数导致记录内容过多问题dev-3.0-tls

1 files changed, 30 insertions, 0 deletions

diff --git a/src/main/java/net/geedge/confagent/config/MyRestTemplateExchangeTagsProvider.java b/src/main/java/net/geedge/confagent/config/MyRestTemplateExchangeTagsProvider.java
new file mode 100644
index 0000000..dd59ff3
--- /dev/null
+++ b/src/main/java/net/geedge/confagent/config/MyRestTemplateExchangeTagsProvider.java
@@ -0,0 +1,30 @@
+package net.geedge.confagent.config;
+
+import io.micrometer.core.instrument.Tag;
+import org.springframework.boot.actuate.metrics.web.client.RestTemplateExchangeTags;
+import org.springframework.boot.actuate.metrics.web.client.RestTemplateExchangeTagsProvider;
+import org.springframework.http.HttpRequest;
+import org.springframework.http.client.ClientHttpResponse;
+import org.springframework.stereotype.Component;
+import org.springframework.util.StringUtils;
+
+import java.util.Arrays;
+
+@Component
+public class MyRestTemplateExchangeTagsProvider implements RestTemplateExchangeTagsProvider {
+
+    @Override
+    public Iterable<Tag> getTags(String urlTemplate, HttpRequest request, ClientHttpResponse response) {
+        Tag uriTag;
+        // 取 request.getURI().getPath() 作为 uri 的 value
+        if (StringUtils.hasText(request.getURI().getPath())) {
+            uriTag = Tag.of("uri", request.getURI().getPath());
+        } else {
+            uriTag = (StringUtils.hasText(urlTemplate) ? RestTemplateExchangeTags.uri(urlTemplate)
+                    : RestTemplateExchangeTags.uri(request));
+        }
+        return Arrays.asList(RestTemplateExchangeTags.method(request), uriTag,
+                RestTemplateExchangeTags.status(response), RestTemplateExchangeTags.clientName(request));
+    }
+
+}