Allow different Micrometer backends to be configured

smallrye · Jun 27, 2022 · 8622394 · 8622394
1 parent 0f1b6b0
commit 8622394
Show file tree

Hide file tree

Showing 8 changed files with 759 additions and 17 deletions.
diff --git a/implementation/pom.xml b/implementation/pom.xml
@@ -133,6 +133,86 @@
       <artifactId>micrometer-registry-prometheus</artifactId>
       <optional>true</optional>
     </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-appoptics</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-atlas</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-datadog</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-dynatrace</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-elastic</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-ganglia</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-graphite</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-humio</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-influx</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-jmx</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-kairos</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-new-relic</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-stackdriver</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-signalfx</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-statsd</artifactId>
+      <optional>true</optional>
+    </dependency>
+    <dependency>
+      <groupId>io.micrometer</groupId>
+      <artifactId>micrometer-registry-wavefront</artifactId>
+      <optional>true</optional>
+    </dependency>
   </dependencies>
 
   <profiles>

diff --git a/implementation/src/main/java/io/smallrye/metrics/MetricRegistries.java b/implementation/src/main/java/io/smallrye/metrics/MetricRegistries.java
@@ -1,14 +1,15 @@
 package io.smallrye.metrics;
 
-import java.util.Arrays;
-import java.util.Map;
-import java.util.Optional;
+import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
 
 import javax.annotation.PreDestroy;
 import javax.enterprise.context.ApplicationScoped;
 import javax.enterprise.inject.Default;
 import javax.enterprise.inject.Produces;
+import javax.enterprise.inject.spi.Bean;
+import javax.enterprise.inject.spi.BeanManager;
+import javax.enterprise.inject.spi.CDI;
 
 import org.eclipse.microprofile.config.ConfigProvider;
 import org.eclipse.microprofile.metrics.MetricRegistry;
@@ -21,7 +22,10 @@
 import io.micrometer.prometheus.PrometheusConfig;
 import io.smallrye.metrics.base.LegacyBaseMetrics;
 import io.smallrye.metrics.legacyapi.LegacyMetricRegistryAdapter;
+import io.smallrye.metrics.micrometer.Backend;
+import io.smallrye.metrics.micrometer.MicrometerBackends;
 import io.smallrye.metrics.setup.ApplicationNameResolver;
+import io.smallrye.metrics.setup.MPCompositeMeterRegistry;
 import io.smallrye.metrics.setup.MPPrometheusMeterRegistry;
 
 /**
@@ -116,21 +120,51 @@ public static MetricRegistry getOrCreate(MetricRegistry.Type type, ApplicationNa
     private static MeterRegistry resolveMeterRegistry(MetricRegistry.Type type) {
         final MeterRegistry meterRegistry;
 
-        if (type == MetricRegistry.Type.BASE) {
-            meterRegistry = new MPPrometheusMeterRegistry(PrometheusConfig.DEFAULT, MetricRegistry.Type.BASE);
-            meterRegistry.config().commonTags("scope", MetricRegistry.Type.BASE.getName());
-            meterRegistry.config().meterFilter(mpMeterBaseRegistryAccessFilter);
-
-        } else if (type == MetricRegistry.Type.APPLICATION) {
-            meterRegistry = new MPPrometheusMeterRegistry(PrometheusConfig.DEFAULT, MetricRegistry.Type.APPLICATION);
-            meterRegistry.config().commonTags("scope", MetricRegistry.Type.APPLICATION.getName());
-            meterRegistry.config().meterFilter(mpMeterAppRegistryAccessFilter);
-        } else if (type == MetricRegistry.Type.VENDOR) {
-            meterRegistry = new MPPrometheusMeterRegistry(PrometheusConfig.DEFAULT, MetricRegistry.Type.VENDOR);
-            meterRegistry.config().commonTags("scope", MetricRegistry.Type.VENDOR.getName());
-            meterRegistry.config().meterFilter(mpMeterVendorRegistryAccessFilter);
+        if (isCDIAvailable()) {
+            final BeanManager bm = CDI.current().getBeanManager();
+            final List<MeterRegistry> meterRegistries = new ArrayList<>();
+
+            final Set<Bean<?>> beans = bm.getBeans(MeterRegistry.class, MicrometerBackends.class.getAnnotation(Backend.class));
+            for (Bean<?> bean : beans) {
+                final Object reference = bm.getReference(bean, MeterRegistry.class, bm.createCreationalContext(bean));
+                if (MeterRegistry.class.isInstance(reference)) {
+                    meterRegistries.add(MeterRegistry.class.cast(reference));
+                }
+            }
+
+            if (type == MetricRegistry.Type.BASE) {
+                meterRegistry = new MPCompositeMeterRegistry(meterRegistries, MetricRegistry.Type.BASE);
+                meterRegistry.config().commonTags("scope", MetricRegistry.Type.BASE.getName());
+                meterRegistry.config().meterFilter(mpMeterBaseRegistryAccessFilter);
+
+            } else if (type == MetricRegistry.Type.APPLICATION) {
+                meterRegistry = new MPCompositeMeterRegistry(meterRegistries, MetricRegistry.Type.APPLICATION);
+                meterRegistry.config().commonTags("scope", MetricRegistry.Type.APPLICATION.getName());
+                meterRegistry.config().meterFilter(mpMeterAppRegistryAccessFilter);
+            } else if (type == MetricRegistry.Type.VENDOR) {
+                meterRegistry = new MPCompositeMeterRegistry(meterRegistries, MetricRegistry.Type.VENDOR);
+                meterRegistry.config().commonTags("scope", MetricRegistry.Type.VENDOR.getName());
+                meterRegistry.config().meterFilter(mpMeterVendorRegistryAccessFilter);
+            } else {
+                meterRegistry = Metrics.globalRegistry;
+            }
         } else {
-            meterRegistry = Metrics.globalRegistry;
+            if (type == MetricRegistry.Type.BASE) {
+                meterRegistry = new MPPrometheusMeterRegistry(PrometheusConfig.DEFAULT, MetricRegistry.Type.BASE);
+                meterRegistry.config().commonTags("scope", MetricRegistry.Type.BASE.getName());
+                meterRegistry.config().meterFilter(mpMeterBaseRegistryAccessFilter);
+
+            } else if (type == MetricRegistry.Type.APPLICATION) {
+                meterRegistry = new MPPrometheusMeterRegistry(PrometheusConfig.DEFAULT, MetricRegistry.Type.APPLICATION);
+                meterRegistry.config().commonTags("scope", MetricRegistry.Type.APPLICATION.getName());
+                meterRegistry.config().meterFilter(mpMeterAppRegistryAccessFilter);
+            } else if (type == MetricRegistry.Type.VENDOR) {
+                meterRegistry = new MPPrometheusMeterRegistry(PrometheusConfig.DEFAULT, MetricRegistry.Type.VENDOR);
+                meterRegistry.config().commonTags("scope", MetricRegistry.Type.VENDOR.getName());
+                meterRegistry.config().meterFilter(mpMeterVendorRegistryAccessFilter);
+            } else {
+                meterRegistry = Metrics.globalRegistry;
+            }
         }
 
         meterRegistry.config().meterFilter(MeterFilter.deny());
@@ -155,6 +189,15 @@ private static MeterRegistry resolveMeterRegistry(MetricRegistry.Type type) {
         return meterRegistry;
     }
 
+    private static boolean isCDIAvailable() {
+        try {
+            CDI.current();
+            return true;
+        } catch (Exception e) {
+            return false;
+        }
+    }
+
     @PreDestroy
     public void cleanUp() {
         registries.remove(MetricRegistry.Type.APPLICATION);

diff --git a/implementation/src/main/java/io/smallrye/metrics/legacyapi/LegacyMetricsExtension.java b/implementation/src/main/java/io/smallrye/metrics/legacyapi/LegacyMetricsExtension.java
@@ -49,6 +49,8 @@
 import io.smallrye.metrics.legacyapi.interceptors.MetricResolver;
 import io.smallrye.metrics.legacyapi.interceptors.MetricsBinding;
 import io.smallrye.metrics.legacyapi.interceptors.TimedInterceptor;
+import io.smallrye.metrics.micrometer.MicrometerBackends;
+import io.smallrye.metrics.micrometer.RequiresClass;
 import io.smallrye.metrics.setup.MetricsMetadata;
 
 /**
@@ -109,6 +111,15 @@ void registerAnnotatedTypes(@Observes BeforeBeanDiscovery bbd, BeanManager manag
         }) {
             bbd.addAnnotatedType(manager.createAnnotatedType(clazz), extensionName + "_" + clazz.getName());
         }
+
+        for (Class clazz : MicrometerBackends.classes()) {
+            try {
+                final RequiresClass requiresClass = (RequiresClass) clazz.getAnnotation(RequiresClass.class);
+                bbd.addAnnotatedType(manager.createAnnotatedType(clazz), extensionName + "_" + clazz.getName());
+            } catch (Exception e) {
+                // ignore and don't add
+            }
+        }
     }
 
     /*

diff --git a/implementation/src/main/java/io/smallrye/metrics/micrometer/Backend.java b/implementation/src/main/java/io/smallrye/metrics/micrometer/Backend.java
@@ -0,0 +1,30 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.smallrye.metrics.micrometer;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+import javax.inject.Qualifier;
+
+@Qualifier
+@Target({ ElementType.TYPE, ElementType.METHOD })
+@Retention(RetentionPolicy.RUNTIME)
+public @interface Backend {
+}