1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
# frozen_string_literal: true
module GraphQL
module Tracing
module PrometheusTrace
class GraphQLCollector < ::PrometheusExporter::Server::TypeCollector
def initialize
@graphql_gauge = PrometheusExporter::Metric::Base.default_aggregation.new(
'graphql_duration_seconds',
'Time spent in GraphQL operations, in seconds'
)
end
def type
'graphql'
end
def collect(object)
default_labels = { key: object['key'], platform_key: object['platform_key'] }
custom = object['custom_labels']
labels = custom.nil? ? default_labels : default_labels.merge(custom)
@graphql_gauge.observe object['duration'], labels
end
def metrics
[@graphql_gauge]
end
end
end
# Backwards-compat:
PrometheusTracing::GraphQLCollector = PrometheusTrace::GraphQLCollector
end
end
|
