Sha256: 648f7a2ae47da1575254a266546f25feaad1ee7987e4d33bfc89d01645e6ea8a

Contents?: true

Size: 1.4 KB

Versions: 11

Compression:

Stored size: 1.4 KB

Contents

require 'ddtrace/contrib/patcher'

module Datadog
  module Contrib
    module GraphQL
      # Provides instrumentation for `graphql` through the GraphQL tracing framework
      module Patcher
        include Contrib::Patcher

        module_function

        def patched?
          done?(:graphql)
        end

        def patch
          return if get_option(:schemas).nil?

          do_once(:graphql) do
            begin
              require 'ddtrace/ext/app_types'
              require 'ddtrace/ext/http'
              get_option(:schemas).each { |s| patch_schema!(s) }
            rescue StandardError => e
              Datadog::Tracer.log.error("Unable to apply GraphQL integration: #{e}")
            end
          end
        end

        def patch_schema!(schema)
          tracer = get_option(:tracer)
          service_name = get_option(:service_name)

          if schema.respond_to?(:use)
            schema.use(
              ::GraphQL::Tracing::DataDogTracing,
              tracer: tracer,
              service: service_name
            )
          else
            schema.define do
              use(
                ::GraphQL::Tracing::DataDogTracing,
                tracer: tracer,
                service: service_name
              )
            end
          end
        end

        def get_option(option)
          Datadog.configuration[:graphql].get_option(option)
        end
      end
    end
  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
ddtrace-0.20.0 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.19.1 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.19.0 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.18.3 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.18.2 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.18.1 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.18.0 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.17.3 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.17.2 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.17.1 lib/ddtrace/contrib/graphql/patcher.rb
ddtrace-0.17.0 lib/ddtrace/contrib/graphql/patcher.rb