Merge pull request #316 from alphagov/add-custom-collectors

Add ability to specify custom collectors
alphagov · Sep 8, 2023 · 6a68034 · 6a68034
2 parents 447f9c7 + 74759c6
commit 6a68034
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,7 @@
 
 * GovukAppConfig silences OpenTelemetry log output when running a rake task ([#311](https://github.com/alphagov/govuk_app_config/pull/311))
 * Update warning message for Prometheus metric server address already in use.
+* Add ability to support custom collectors in the Prometheus exporter.
 
 # 9.0.4
 

diff --git a/lib/govuk_app_config/govuk_prometheus_exporter.rb b/lib/govuk_app_config/govuk_prometheus_exporter.rb
@@ -11,7 +11,7 @@ def self.should_configure
     end
   end
 
-  def self.configure
+  def self.configure(collectors: [])
     return unless should_configure
 
     require "prometheus_exporter"
@@ -37,6 +37,9 @@ def self.configure
 
     begin
       server = PrometheusExporter::Server::WebServer.new bind: "0.0.0.0", port: 9394
+
+      collectors.each { |collector| server.collector.register_collector(collector.new) }
+
       server.start
 
       if defined?(Rails)