module GraphQL::Execution::Instrumentation

Public Class Methods

apply_instrumenters(multiplex) { || ... } click to toggle source

This function implements the instrumentation policy:

  • Instrumenters are a stack; the first `before_query` will have the last `after_query`

  • If a `before_` hook returned without an error, its corresponding `after_` hook will run.

  • If the `before_` hook did not run, the `after_` hook will not be called.

When errors are raised from `after_` hooks:

- Subsequent `after_` hooks _are_ called
- The first raised error is captured; later errors are ignored
- If an error was capture, it's re-raised after all hooks are finished

Partial runs of instrumentation are possible:

  • If a `before_multiplex` hook raises an error, no `before_query` hooks will run

  • If a `before_query` hook raises an error, subsequent `before_query` hooks will not run (on any query)

# File lib/graphql/execution/instrumentation.rb, line 19
def self.apply_instrumenters(multiplex)
  schema = multiplex.schema
  queries = multiplex.queries
  query_instrumenters = schema.instrumenters[:query]
  multiplex_instrumenters = schema.instrumenters[:multiplex]

  # First, run multiplex instrumentation, then query instrumentation for each query
  call_hooks(multiplex_instrumenters, multiplex, :before_multiplex, :after_multiplex) do
    each_query_call_hooks(query_instrumenters, queries) do
      # Let them be executed
      yield
    end
  end
end

Private Class Methods

call_after_hooks(instrumenters, object, after_hook_name, ex) click to toggle source
# File lib/graphql/execution/instrumentation.rb, line 79
def call_after_hooks(instrumenters, object, after_hook_name, ex)
  instrumenters.reverse.each do |instrumenter|
    begin
      instrumenter.public_send(after_hook_name, object)
    rescue => e
      ex = e
    end
  end
  ex
end
call_hooks(instrumenters, object, before_hook_name, after_hook_name) { || ... } click to toggle source

Call each before hook, and if they all succeed, yield. If they don't all succeed, call after_ for each one that succeeded.

# File lib/graphql/execution/instrumentation.rb, line 54
def call_hooks(instrumenters, object, before_hook_name, after_hook_name)
  begin
    successful = []
    instrumenters.each do |instrumenter|
      instrumenter.public_send(before_hook_name, object)
      successful << instrumenter
    end

    # if any before hooks raise an exception, quit calling before hooks,
    # but call the after hooks on anything that succeeded but also
    # raise the exception that came from the before hook.
  rescue GraphQL::ExecutionError => err
    object.context.errors << err
  rescue => e
    raise call_after_hooks(successful, object, after_hook_name, e)
  end

  begin
    yield # Call the user code
  ensure
    ex = call_after_hooks(successful, object, after_hook_name, nil)
    raise ex if ex
  end
end
each_query_call_hooks(instrumenters, queries, i = 0) { || ... } click to toggle source

Call the before_ hooks of each query, Then yield if no errors. `call_hooks` takes care of appropriate cleanup.

# File lib/graphql/execution/instrumentation.rb, line 39
def each_query_call_hooks(instrumenters, queries, i = 0)
  if i >= queries.length
    yield
  else
    query = queries[i]
    call_hooks(instrumenters, query, :before_query, :after_query) {
      each_query_call_hooks(instrumenters, queries, i + 1) {
        yield
      }
    }
  end
end