From 55a575f87a7b15271b5cc734cccc9e1e85567410 Mon Sep 17 00:00:00 2001
From: Eugene Burmakin <iamfrey@gmail.com>
Date: Wed, 20 Nov 2024 20:50:21 +0100
Subject: [PATCH] Update sidekiq configuration to include Prometheus
 instrumentation

---
 config/initializers/sidekiq.rb | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/config/initializers/sidekiq.rb b/config/initializers/sidekiq.rb
index d3c9bbea..fdbca0dd 100644
--- a/config/initializers/sidekiq.rb
+++ b/config/initializers/sidekiq.rb
@@ -3,10 +3,22 @@
 Sidekiq.configure_server do |config|
   config.redis = { url: ENV['REDIS_URL'] }
 
-  config.server_middleware do |chain|
+  if ENV.fetch('PROMETHEUS_EXPORTER_ENABLED', false)
     require 'prometheus_exporter/instrumentation'
-    chain.add PrometheusExporter::Instrumentation::Sidekiq
- end
+
+    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
 
 Sidekiq.configure_client do |config|