Change how collectors are registered

mkuratczyk · mkuratczyk · commit ee035100e557 · 2025-05-21T17:42:19.000+02:00
diff --git a/deps/rabbitmq_prometheus/src/rabbit_prometheus_dispatcher.erl b/deps/rabbitmq_prometheus/src/rabbit_prometheus_dispatcher.erl
@@ -13,14 +13,14 @@
 
 build_dispatcher() ->
     {ok, _} = application:ensure_all_started(prometheus),
-    prometheus_registry:register_collectors([
+    AggregatedCollectors = [
         prometheus_rabbitmq_core_metrics_collector,
         prometheus_rabbitmq_global_metrics_collector,
         prometheus_rabbitmq_message_size_metrics_collector,
         prometheus_rabbitmq_alarm_metrics_collector,
         prometheus_rabbitmq_dynamic_collector,
-        prometheus_process_collector]),
-    prometheus_registry:register_collectors('per-object', [
+        prometheus_process_collector],
+    PerObjectCollectors = [
         prometheus_vm_system_info_collector,
         prometheus_vm_dist_collector,
         prometheus_vm_memory_collector,
@@ -31,7 +31,14 @@ build_dispatcher() ->
         prometheus_rabbitmq_global_metrics_collector,
         prometheus_rabbitmq_raft_metrics_collector,
         prometheus_rabbitmq_message_size_metrics_collector
-        ]),
+        ],
+    prometheus_registry:register_collectors(
+        case application:get_env(rabbitmq_prometheus, return_per_object_metrics, false) of
+            false -> AggregatedCollectors;
+            true -> PerObjectCollectors
+        end),
+    prometheus_registry:register_collectors('per-object',
+        PerObjectCollectors),
     prometheus_registry:register_collectors('detailed', [
         prometheus_rabbitmq_core_metrics_collector
         ]),
