class GraphQL::Tracing::PrometheusTrace::GraphQLCollector

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 initialize
  @graphql_gauge = PrometheusExporter::Metric::Base.default_aggregation.new(
    'graphql_duration_seconds',
    'Time spent in GraphQL operations, in seconds'
  )
end

def metrics
  [@graphql_gauge]
end

def type
  'graphql'
end