Track new Redis connections

Increment the counter `new_redis_connections` on each call to
`Redis::Client#connect`, if we're in a transaction.

1 files changed, 16 insertions, 0 deletions

diff --git a/config/initializers/metrics.rb b/config/initializers/metrics.rb
index f6509ee43f1..4bc6acdedb9 100644
--- a/config/initializers/metrics.rb
+++ b/config/initializers/metrics.rb
@@ -138,4 +138,20 @@ if Gitlab::Metrics.enabled?
   GC::Profiler.enable
 
   Gitlab::Metrics::Sampler.new.start
+
+  module TrackNewRedisConnections
+    def connect(*args)
+      val = super
+
+      if current_transaction = Gitlab::Metrics::Transaction.current
+        current_transaction.increment(:new_redis_connections, 1)
+      end
+
+      val
+    end
+  end
+
+  class ::Redis::Client
+    prepend TrackNewRedisConnections
+  end
 end