Store Sherlock::Query in Peek adapter

Signed-off-by: Rémy Coutable <remy@rymai.me>

1 files changed, 32 insertions, 0 deletions

diff --git a/config/initializers/peek.rb b/config/initializers/peek.rb
index 73da7be7889..6cb0e711e48 100644
--- a/config/initializers/peek.rb
+++ b/config/initializers/peek.rb
@@ -7,3 +7,35 @@ Peek.into Peek::Views::Redis
 Peek.into Peek::Views::Sidekiq
 Peek.into Peek::Views::Rblineprof
 Peek.into Peek::Views::GC
+
+if Gitlab::Database.mysql?
+  class Mysql2::Client
+    class << self
+      attr_accessor :query_details
+    end
+    self.query_details = Concurrent::Array.new
+  end
+
+  module Peek
+    module Views
+      class Mysql2 < View
+        prepend ::Gitlab::PerformanceBar::PeekMysqlWithQueries
+      end
+    end
+  end
+else
+  class PG::Connection
+    class << self
+      attr_accessor :query_details
+    end
+    self.query_details = Concurrent::Array.new
+  end
+
+  module Peek
+    module Views
+      class PG < View
+        prepend ::Gitlab::PerformanceBar::PeekPgWithQueries
+      end
+    end
+  end
+end