1 files changed, 38 insertions, 0 deletions
diff --git a/lib/gitlab/sidekiq_middleware/metrics_helper.rb b/lib/gitlab/sidekiq_middleware/metrics_helper.rb
new file mode 100644
index 00000000000..5c1ce2b98e8
--- /dev/null
+++ b/lib/gitlab/sidekiq_middleware/metrics_helper.rb
@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module SidekiqMiddleware
+    module MetricsHelper
+      TRUE_LABEL = "yes"
+      FALSE_LABEL = "no"
+
+      private
+
+      def create_labels(worker_class, queue)
+        labels = { queue: queue.to_s,
+                   worker: worker_class.to_s,
+                   urgency: "",
+                   external_dependencies: FALSE_LABEL,
+                   feature_category: "",
+                   boundary: "" }
+
+        return labels unless worker_class && worker_class.include?(WorkerAttributes)
+
+        labels[:urgency] = worker_class.get_urgency.to_s
+        labels[:external_dependencies] = bool_as_label(worker_class.worker_has_external_dependencies?)
+
+        feature_category = worker_class.get_feature_category
+        labels[:feature_category] = feature_category.to_s
+
+        resource_boundary = worker_class.get_worker_resource_boundary
+        labels[:boundary] = resource_boundary == :unknown ? "" : resource_boundary.to_s
+
+        labels
+      end
+
+      def bool_as_label(value)
+        value ? TRUE_LABEL : FALSE_LABEL
+      end
+    end
+  end
+end