Sha256: 5bb0cbdbfc7a2e95e1ac4b5892a6513378a8dbe3f3e8d646a683cd0f663a5b2c

Contents?: true

Size: 1.99 KB

Versions: 96

Compression:

Stored size: 1.99 KB

Contents

# frozen_string_literal: true

module GraphQL
  module Tracing
    class PrometheusTracing < PlatformTracing
      DEFAULT_WHITELIST = ['execute_field', 'execute_field_lazy'].freeze
      DEFAULT_COLLECTOR_TYPE = 'graphql'.freeze

      self.platform_keys = {
        'lex' => "graphql.lex",
        'parse' => "graphql.parse",
        'validate' => "graphql.validate",
        'analyze_query' => "graphql.analyze",
        'analyze_multiplex' => "graphql.analyze",
        'execute_multiplex' => "graphql.execute",
        'execute_query' => "graphql.execute",
        'execute_query_lazy' => "graphql.execute",
        'execute_field' => "graphql.execute",
        'execute_field_lazy' => "graphql.execute"
      }

      def initialize(opts = {})
        @client = opts[:client] || PrometheusExporter::Client.default
        @keys_whitelist = opts[:keys_whitelist] || DEFAULT_WHITELIST
        @collector_type = opts[:collector_type] || DEFAULT_COLLECTOR_TYPE

        super opts
      end

      def platform_trace(platform_key, key, data, &block)
        return yield unless @keys_whitelist.include?(key)
        instrument_execution(platform_key, key, data, &block)
      end

      def platform_field_key(type, field)
        "#{type.graphql_name}.#{field.graphql_name}"
      end

      def platform_authorized_key(type)
        "#{type.graphql_name}.authorized"
      end

      def platform_resolve_type_key(type)
        "#{type.graphql_name}.resolve_type"
      end

      private

      def instrument_execution(platform_key, key, data, &block)
        start = ::Process.clock_gettime ::Process::CLOCK_MONOTONIC
        result = block.call
        duration = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC) - start
        observe platform_key, key, duration
        result
      end

      def observe(platform_key, key, duration)
        @client.send_json(
          type: @collector_type,
          duration: duration,
          platform_key: platform_key,
          key: key
        )
      end
    end
  end
end

Version data entries

96 entries across 96 versions & 2 rubygems

Version Path
graphql-1.13.23 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.22 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.21 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.20 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.17.2 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.17.1 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.17 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.19 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.18 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.16 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.17 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.15 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.14 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.16 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.13 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.12 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.15 lib/graphql/tracing/prometheus_tracing.rb
graphql-2.0.11 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.14 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.13.13 lib/graphql/tracing/prometheus_tracing.rb