Add latest changes from gitlab-org/gitlab@13-10-stable-eev13.10.0-rc40

1 files changed, 49 insertions, 0 deletions

diff --git a/lib/gitlab/metrics/background_transaction.rb b/lib/gitlab/metrics/background_transaction.rb
new file mode 100644
index 00000000000..3dda68bf93f
--- /dev/null
+++ b/lib/gitlab/metrics/background_transaction.rb
@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Metrics
+    class BackgroundTransaction < Transaction
+      # Separate web transaction instance and background transaction instance
+      BACKGROUND_THREAD_KEY = :_gitlab_metrics_background_transaction
+      BACKGROUND_BASE_LABEL_KEYS = %i(endpoint_id feature_category).freeze
+
+      class << self
+        def current
+          Thread.current[BACKGROUND_THREAD_KEY]
+        end
+
+        def prometheus_metric(name, type, &block)
+          fetch_metric(type, name) do
+            # set default metric options
+            docstring "#{name.to_s.humanize} #{type}"
+
+            evaluate(&block)
+            # always filter sensitive labels and merge with base ones
+            label_keys BACKGROUND_BASE_LABEL_KEYS | (label_keys - ::Gitlab::Metrics::Transaction::FILTERED_LABEL_KEYS)
+          end
+        end
+      end
+
+      def run
+        Thread.current[BACKGROUND_THREAD_KEY] = self
+
+        yield
+      ensure
+        Thread.current[BACKGROUND_THREAD_KEY] = nil
+      end
+
+      def labels
+        @labels ||= {
+          endpoint_id: current_context&.get_attribute(:caller_id),
+          feature_category: current_context&.get_attribute(:feature_category)
+        }
+      end
+
+      private
+
+      def current_context
+        Labkit::Context.current
+      end
+    end
+  end
+end