1 files changed, 49 insertions, 0 deletions
diff --git a/lib/gitlab/instrumentation/zoekt.rb b/lib/gitlab/instrumentation/zoekt.rb
new file mode 100644
index 00000000000..cd9b15bcee8
--- /dev/null
+++ b/lib/gitlab/instrumentation/zoekt.rb
@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Instrumentation
+    class Zoekt
+      ZOEKT_REQUEST_COUNT = :zoekt_request_count
+      ZOEKT_CALL_DURATION = :zoekt_call_duration
+      ZOEKT_CALL_DETAILS = :zoekt_call_details
+
+      class << self
+        def get_request_count
+          ::Gitlab::SafeRequestStore[ZOEKT_REQUEST_COUNT] || 0
+        end
+
+        def increment_request_count
+          ::Gitlab::SafeRequestStore[ZOEKT_REQUEST_COUNT] ||= 0
+          ::Gitlab::SafeRequestStore[ZOEKT_REQUEST_COUNT] += 1
+        end
+
+        def detail_store
+          ::Gitlab::SafeRequestStore[ZOEKT_CALL_DETAILS] ||= []
+        end
+
+        def query_time
+          query_time = ::Gitlab::SafeRequestStore[ZOEKT_CALL_DURATION] || 0
+          query_time.round(::Gitlab::InstrumentationHelper::DURATION_PRECISION)
+        end
+
+        def add_duration(duration)
+          ::Gitlab::SafeRequestStore[ZOEKT_CALL_DURATION] ||= 0
+          ::Gitlab::SafeRequestStore[ZOEKT_CALL_DURATION] += duration
+        end
+
+        def add_call_details(duration:, method:, path:, params: nil, body: nil)
+          return unless Gitlab::PerformanceBar.enabled_for_request?
+
+          detail_store << {
+            method: method,
+            path: path,
+            params: params,
+            body: body,
+            duration: duration,
+            backtrace: ::Gitlab::BacktraceCleaner.clean_backtrace(caller)
+          }
+        end
+      end
+    end
+  end
+end