diff options
| -rw-r--r-- | src/main/java/net/geedge/confagent/config/MyRestTemplateExchangeTagsProvider.java | 30 |
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)); + } + +} |
