gitlab-org--gitlab-foss/lib/gitlab/graphql/tracers/metrics_tracer.rb

48 lines
1.4 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module Graphql
module Tracers
class MetricsTracer
def self.use(schema)
schema.tracer(self.new)
end
# See https://graphql-ruby.org/api-doc/1.12.16/GraphQL/Tracing for full list of events
def trace(key, data)
result = yield
case key
when "execute_query"
increment_query_sli(data)
end
result
end
private
def increment_query_sli(data)
duration_s = data.fetch(:duration_s, nil)
query = data.fetch(:query, nil)
# We're just being defensive here...
# duration_s comes from TimerTracer and we should be pretty much guaranteed it exists
return unless duration_s && query
operation = ::Gitlab::Graphql::KnownOperations.default.from_query(query)
query_urgency = operation.query_urgency
Gitlab::Metrics::RailsSlis.graphql_query_apdex.increment(
labels: {
endpoint_id: ::Gitlab::ApplicationContext.current_context_attribute(:caller_id),
feature_category: ::Gitlab::ApplicationContext.current_context_attribute(:feature_category),
query_urgency: query_urgency.name
},
success: duration_s <= query_urgency.duration
)
end
end
end
end
end