Sha256: 9e242ac37db9e0c1ad56dac3adb23df8ce5a03acc1c8c0f2e3a97a615e2ef234

Contents?: true

Size: 1.8 KB

Versions: 29

Compression:

Stored size: 1.8 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

      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

29 entries across 29 versions & 1 rubygems

Version Path
graphql-1.9.21 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.20 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.19 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.18 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.10.0.pre3 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.17 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.10.0.pre2 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.16 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.15 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.14 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.10.0.pre1 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.13 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.12 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.11 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.10 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.9 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.8 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.7 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.6 lib/graphql/tracing/prometheus_tracing.rb
graphql-1.9.5 lib/graphql/tracing/prometheus_tracing.rb