Merge pull request #1 from unidev-platform/metrics-labels

Add default labels to metrics
unidev-platform · Mar 18, 2023 · 149188c · 149188c
2 parents 263c3ba + 5992bae
commit 149188c
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/unidev-prometheus/src/main/java/com/unidev/platform/prometheus/MetricService.java b/unidev-prometheus/src/main/java/com/unidev/platform/prometheus/MetricService.java
@@ -13,6 +13,8 @@
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ConcurrentMap;
 
+import static java.util.Collections.emptyMap;
+
 /**
  * Service used to track metrics
  */
@@ -21,6 +23,8 @@ public class MetricService {
 
     private final String prefix;
 
+    private final Map<String, String> defaultLabels;
+
     private final ConcurrentMap<String, Counter> counters = new ConcurrentHashMap<>();
 
     private final ConcurrentMap<String, Gauge> gauges = new ConcurrentHashMap<>();
@@ -29,13 +33,17 @@ public class MetricService {
 
     private final ConcurrentMap<String, Histogram> histograms = new ConcurrentHashMap<>();
 
-
     public MetricService(String prefix, int prometheusPort) throws IOException {
+        this(prefix, prometheusPort, emptyMap());
+    }
+
+    public MetricService(String prefix, int prometheusPort, Map<String, String> defaultLabels) throws IOException {
         this.prefix = prefix;
         DefaultExports.initialize();
-        server = new HTTPServer.Builder()
+        this.server = new HTTPServer.Builder()
                 .withPort(prometheusPort)
                 .build();
+        this.defaultLabels = defaultLabels;
     }
 
     /**
@@ -44,6 +52,7 @@ public MetricService(String prefix, int prometheusPort) throws IOException {
     public Counter.Child counter(String name, Map<String, String> labels) {
         try {
             TreeMap<String, String> labelMap = new TreeMap<>(labels);
+            labelMap.putAll(defaultLabels);
             String prometheusName = buildName(name);
             counters.computeIfAbsent(prometheusName, s -> {
                 Counter value = Counter.build()
@@ -71,7 +80,7 @@ public void inc(String name, Map<String, String> labels) {
      */
     public Gauge.Child gauge(String name, Map<String, String> labels) {
         TreeMap<String, String> labelMap = new TreeMap<>(labels);
-
+        labelMap.putAll(defaultLabels);
         try {
             String prometheusName = buildName(name);
             gauges.computeIfAbsent(prometheusName, s -> {
@@ -94,6 +103,7 @@ public Gauge.Child gauge(String name, Map<String, String> labels) {
      */
     public Summary.Child summary(String name, Map<String, String> labels) {
         TreeMap<String, String> labelMap = new TreeMap<>(labels);
+        labelMap.putAll(defaultLabels);
         try {
             String prometheusName = buildName(name);
             summaries.computeIfAbsent(prometheusName, s -> {
@@ -116,6 +126,7 @@ public Summary.Child summary(String name, Map<String, String> labels) {
      */
     public Histogram.Child histogram(String name, Map<String, String> labels) {
         TreeMap<String, String> labelMap = new TreeMap<>(labels);
+        labelMap.putAll(defaultLabels);
         try {
             String prometheusName = buildName(name);
             histograms.computeIfAbsent(prometheusName, s -> {
@@ -150,6 +161,9 @@ String[] mapValues(Map<String, String> map) {
     }
 
      String buildName(String name) {
+        if (prefix == null || prefix.isEmpty()) {
+            return name.toLowerCase().replaceAll("[^A-Za-z0-9]", "_");
+        }
         return (prefix + "_" + name.toLowerCase()).replaceAll("[^A-Za-z0-9]", "_");
     }