feat(forest): 增加接口访问指标

2024-01-31 15:50:05 +08:00
parent ee22986d6f
commit 7a1f28b82b
2 changed files with 110 additions and 1 deletions
--- a/service-forest/src/main/java/com/lanyuanxiaoyao/service/forest/configuration/MetricsInterceptor.java
+++ b/service-forest/src/main/java/com/lanyuanxiaoyao/service/forest/configuration/MetricsInterceptor.java
@@ -0,0 +1,107 @@
+package com.lanyuanxiaoyao.service.forest.configuration;
+
+import cn.hutool.core.util.ObjectUtil;
+import com.dtflys.forest.exceptions.ForestRuntimeException;
+import com.dtflys.forest.http.ForestQueryMap;
+import com.dtflys.forest.http.ForestRequest;
+import com.dtflys.forest.http.ForestResponse;
+import com.dtflys.forest.interceptor.Interceptor;
+import io.micrometer.core.instrument.Counter;
+import io.micrometer.core.instrument.MeterRegistry;
+import io.micrometer.core.instrument.Tag;
+import io.micrometer.core.instrument.Timer;
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+import org.eclipse.collections.api.factory.Lists;
+import org.eclipse.collections.api.factory.Maps;
+import org.eclipse.collections.api.list.MutableList;
+import org.eclipse.collections.api.map.MutableMap;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+/**
+ * 指标
+ *
+ * @author lanyuanxiaoyao
+ * @date 2024-01-31
+ */
+@SuppressWarnings("unchecked")
+@Component
+public class MetricsInterceptor implements Interceptor<Object> {
+    public static final String FOREST_SEND_REQUEST = "forest_send_request";
+    public static final String FOREST_SEND_REQUEST_SECONDS = "forest_send_request_seconds";
+    private static final Logger logger = LoggerFactory.getLogger(MetricsInterceptor.class);
+    private static final String PARSED_TAGS = "parsed_tags";
+    private final MeterRegistry meterRegistry;
+    private final MutableMap<String, Counter> counterCache =
+            Maps.mutable.<String, Counter>empty().asSynchronized();
+    private final MutableMap<String, Timer> timerCache =
+            Maps.mutable.<String, Timer>empty().asSynchronized();
+
+    public MetricsInterceptor(MeterRegistry meterRegistry) {
+        this.meterRegistry = meterRegistry;
+    }
+
+    private Iterable<Tag> parseTags(ForestRequest<?> request) {
+        MutableList<Tag> tags = Lists.mutable.of(
+                Tag.of("host", request.getHost()),
+                Tag.of("port", String.valueOf(request.getPort())),
+                Tag.of("path", request.getPath()),
+                Tag.of("schema", request.getScheme())
+        );
+        ForestQueryMap query = request.getQuery();
+        if (ObjectUtil.isNotEmpty(query)) {
+            for (Map.Entry<String, Object> entry : query.entrySet()) {
+                tags.add(Tag.of("query_" + entry.getKey(), String.valueOf(entry.getValue())));
+            }
+        }
+        return tags;
+    }
+
+    private void increaseCounter(ForestRequest<?> request, ForestResponse<?> response, MutableList<Tag> tags) {
+        Counter counter = counterCache.getIfAbsentPut(
+                request.getUrl(),
+                meterRegistry.counter(
+                        MetricsInterceptor.FOREST_SEND_REQUEST,
+                        tags.with(Tag.of("code", String.valueOf(response.getStatusCode())))
+                )
+        );
+        counter.increment();
+    }
+
+    @Override
+    public boolean beforeExecute(ForestRequest request) {
+        addAttribute(request, PARSED_TAGS, parseTags(request));
+        return Interceptor.super.beforeExecute(request);
+    }
+
+    @Override
+    public void afterExecute(ForestRequest request, ForestResponse response) {
+        Interceptor.super.afterExecute(request, response);
+
+        MutableList<Tag> tags = (MutableList<Tag>) getAttribute(request, PARSED_TAGS);
+        Timer timer = timerCache.getIfAbsentPut(request.getUrl(), meterRegistry.timer(FOREST_SEND_REQUEST_SECONDS, tags));
+        timer.record(response.getTimeAsMillisecond(), TimeUnit.MILLISECONDS);
+    }
+
+    @Override
+    public void onSuccess(Object data, ForestRequest request, ForestResponse response) {
+        Interceptor.super.onSuccess(data, request, response);
+
+        MutableList<Tag> tags = (MutableList<Tag>) getAttribute(request, PARSED_TAGS);
+        if (ObjectUtil.isNotNull(tags)) {
+            increaseCounter(request, response, tags);
+        }
+    }
+
+    @Override
+    public void onError(ForestRuntimeException ex, ForestRequest request, ForestResponse response) {
+        Interceptor.super.onError(ex, request, response);
+
+        MutableList<Tag> tags = (MutableList<Tag>) getAttribute(request, PARSED_TAGS);
+        if (ObjectUtil.isNotNull(tags)) {
+            increaseCounter(request, response, tags);
+        }
+    }
+}
--- a/service-forest/src/main/resources/application-forest.yml
+++ b/service-forest/src/main/resources/application-forest.yml
@@ -2,4 +2,6 @@ forest:
  backend: httpclient
  timeout: 60000
  log-enabled: false
-  interceptors: com.lanyuanxiaoyao.service.forest.configuration.SpringCloudDiscoveryInterceptor
+  interceptors:
+    - com.lanyuanxiaoyao.service.forest.configuration.SpringCloudDiscoveryInterceptor
+    - com.lanyuanxiaoyao.service.forest.configuration.MetricsInterceptor