32 lines
1 KiB
Ruby
32 lines
1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module SidekiqMiddleware
|
|
class Metrics
|
|
TRUE_LABEL = "yes"
|
|
FALSE_LABEL = "no"
|
|
|
|
private
|
|
|
|
def create_labels(worker_class, queue)
|
|
labels = { queue: queue.to_s, latency_sensitive: FALSE_LABEL, external_dependencies: FALSE_LABEL, feature_category: "", boundary: "" }
|
|
return labels unless worker_class && worker_class.include?(WorkerAttributes)
|
|
|
|
labels[:latency_sensitive] = bool_as_label(worker_class.latency_sensitive_worker?)
|
|
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
|