add sinatra

alphagov · Aug 4, 2022 · 4a03ddb · 4a03ddb
1 parent 41ff5ec
commit 4a03ddb
Showing 1 changed file with 15 additions and 13 deletions.
diff --git a/lib/govuk_app_config/govuk_prometheus_exporter.rb b/lib/govuk_app_config/govuk_prometheus_exporter.rb
@@ -7,27 +7,29 @@ def self.configure
       require "prometheus_exporter/server"
       require "prometheus_exporter/middleware"
 
-      #if defined?(Sidekiq)
-      Sidekiq.configure_server do |config|
-        require "prometheus_exporter/instrumentation"
-        config.server_middleware do |chain|
-          chain.add PrometheusExporter::Instrumentation::Sidekiq
-        end
-        config.death_handlers << PrometheusExporter::Instrumentation::Sidekiq.death_handler
-        config.on :startup do
-          PrometheusExporter::Instrumentation::Process.start type: "sidekiq"
-          PrometheusExporter::Instrumentation::SidekiqProcess.start
-          PrometheusExporter::Instrumentation::SidekiqQueue.start
-          PrometheusExporter::Instrumentation::SidekiqStats.start
+      if defined?(Sidekiq)
+        Sidekiq.configure_server do |config|
+          require "prometheus_exporter/instrumentation"
+          config.server_middleware do |chain|
+            chain.add PrometheusExporter::Instrumentation::Sidekiq
+          end
+          config.death_handlers << PrometheusExporter::Instrumentation::Sidekiq.death_handler
+          config.on :startup do
+            PrometheusExporter::Instrumentation::Process.start type: "sidekiq"
+            PrometheusExporter::Instrumentation::SidekiqProcess.start
+            PrometheusExporter::Instrumentation::SidekiqQueue.start
+            PrometheusExporter::Instrumentation::SidekiqStats.start
+          end
         end
       end
-      #end
 
       server = PrometheusExporter::Server::WebServer.new bind: "0.0.0.0", port: 9394
       server.start
 
       if defined?(Rails)
         Rails.application.middleware.unshift PrometheusExporter::Middleware
+      else defined?(Sinatra)
+        Sinatra.application.middleware.unshift PrometheusExporter::Middleware
       end
     end
   end