class GraphQL::Schema::Timeout


end
use MyTimeout, max_seconds: 2
class MySchema < GraphQL::Schema
end
end
Bugsnag.notify(error, {query_string: query.query_string})
Rails.logger.warn(“GraphQL Timeout: #{error.message}: #{query.query_string}”)
def handle_timeout(error, query)
class MyTimeout < GraphQL::Schema::Timeout
@example Notifying Bugsnag and logging a timeout
end
use GraphQL::Schema::Timeout, max_seconds: 2
class MySchema < GraphQL::Schema
@example Stop resolving fields after 2 seconds
/
timeout options for external connections. For more info, see
it doesn’t interrupt long-running ‘resolve` functions. Be sure to use
Note that this will stop a query _in between_ field resolutions, but
to provide custom logic when a timeout error occurs.
You can subclass `GraphQL::Schema::Timeout` and override `max_seconds` and/or `handle_timeout`
you’ll get a partial response.
to the ‘errors` key. Any already-resolved fields will be in the `data` key, so
After the time has passed, any remaining fields will be `nil`, with errors added
This plugin will stop resolving new fields after `max_seconds` have elapsed.

def self.use(schema, **options)

def self.use(schema, **options)
  tracer = new(**options)
  schema.tracer(tracer)
end

def handle_timeout(error, query)

Parameters:
  • query (GraphQL::Error) --
  • error (GraphQL::Schema::Timeout::TimeoutError) --
def handle_timeout(error, query)
  # override to do something interesting
end

def initialize(max_seconds:)

Parameters:
  • max_seconds (Numeric) -- how many seconds the query should be allowed to resolve new fields
def initialize(max_seconds:)
  @max_seconds = max_seconds
end

def max_seconds(query)

Returns:
  • (Integer, false) - The number of seconds after which to interrupt query execution and call {#handle_error}, or `false` to bypass the timeout.

Parameters:
  • query (GraphQL::Query) -- The query that's about to run
def max_seconds(query)
  @max_seconds
end

def trace(key, data)

def trace(key, data)
  case key
  when 'execute_multiplex'
    data.fetch(:multiplex).queries.each do |query|
      timeout_duration_s = max_seconds(query)
      timeout_state = if timeout_duration_s == false
        # if the method returns `false`, don't apply a timeout
        false
      else
        now = Process.clock_gettime(Process::CLOCK_MONOTONIC, :millisecond)
        timeout_at = now + (max_seconds(query) * 1000)
        {
          timeout_at: timeout_at,
          timed_out: false
        }
      end
      query.context.namespace(self.class)[:state] = timeout_state
    end
    yield
  when 'execute_field', 'execute_field_lazy'
    query_context = data[:context] || data[:query].context
    timeout_state = query_context.namespace(self.class).fetch(:state)
    # If the `:state` is `false`, then `max_seconds(query)` opted out of timeout for this query.
    if timeout_state != false && Process.clock_gettime(Process::CLOCK_MONOTONIC, :millisecond) > timeout_state.fetch(:timeout_at)
      error = if data[:context]
        GraphQL::Schema::Timeout::TimeoutError.new(query_context.parent_type, query_context.field)
      else
        field = data.fetch(:field)
        GraphQL::Schema::Timeout::TimeoutError.new(field.owner, field)
      end
      # Only invoke the timeout callback for the first timeout
      if !timeout_state[:timed_out]
        timeout_state[:timed_out] = true
        handle_timeout(error, query_context.query)
      end
      error
    else
      yield
    end
  else
    yield
  end
end