Contents

# frozen_string_literal: true

module GraphQL::Stitching
  class Executor
    class RootSource < GraphQL::Dataloader::Source
      def initialize(executor, location)
        @executor = executor
        @location = location
      end

      def fetch(ops)
        op = ops.first # There should only ever be one per location at a time

        query_document = build_document(
          op,
          @executor.request.operation_name,
          @executor.request.operation_directives,
        )
        query_variables = @executor.request.variables.slice(*op.variables.keys)
        result = @executor.request.supergraph.execute_at_location(op.location, query_document, query_variables, @executor.request)
        @executor.query_count += 1

        @executor.data.merge!(result["data"]) if result["data"]
        if result["errors"]&.any?
          result["errors"].each { _1.delete("locations") }
          @executor.errors.concat(result["errors"])
        end

        ops.map(&:step)
      end

      # Builds root source documents
      # "query MyOperation_1($var:VarType) { rootSelections ... }"
      def build_document(op, operation_name = nil, operation_directives = nil)
        doc = String.new
        doc << op.operation_type

        if operation_name
          doc << " #{operation_name}_#{op.step}"
        end

        if op.variables.any?
          variable_defs = op.variables.map { |k, v| "$#{k}:#{v}" }.join(",")
          doc << "(#{variable_defs})"
        end

        if operation_directives
          doc << " #{operation_directives} "
        end

        doc << op.selections
        doc
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
graphql-stitching-1.2.1	lib/graphql/stitching/executor/root_source.rb