graphql 2.6.13 → 2.6.14

data/lib/graphql/schema/build_from_definition.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
-# typed: false
+# typed: true
 require "graphql/schema/build_from_definition/resolve_map"
 
 module GraphQL
@@ -7,18 +7,18 @@ module GraphQL
     module BuildFromDefinition
       class << self
         # See [Schema.from_definition](rdoc-ref:Schema.from_definition)
-        def from_definition(schema_superclass, definition_string, parser: GraphQL.default_parser, **kwargs)
+        def from_definition(schema_superclass, definition_string, default_resolve:, parser: GraphQL.default_parser, using: {}, base_types: {}, relay: false)
           if defined?(parser::SchemaParser)
-            parser = parser::SchemaParser
+            parser = parser.const_get(:SchemaParser)
           end
-          from_document(schema_superclass, parser.parse(definition_string), **kwargs)
+          from_document(schema_superclass, parser.parse(definition_string), default_resolve: default_resolve, using: using, base_types: base_types, relay: relay)
         end
 
-        def from_definition_path(schema_superclass, definition_path, parser: GraphQL.default_parser, **kwargs)
+        def from_definition_path(schema_superclass, definition_path, default_resolve:, parser: GraphQL.default_parser, using: {}, base_types: {}, relay: false)
           if defined?(parser::SchemaParser)
-            parser = parser::SchemaParser
+            parser = parser.const_get(:SchemaParser)
           end
-          from_document(schema_superclass, parser.parse_file(definition_path), **kwargs)
+          from_document(schema_superclass, parser.parse_file(definition_path), default_resolve: default_resolve, using: using, base_types: base_types, relay: relay)
         end
 
         def from_document(schema_superclass, document, default_resolve:, using: {}, base_types: {}, relay: false)
@@ -48,24 +48,61 @@ module GraphQL
             default_resolve = ResolveMap.new(default_resolve)
           end
 
-          schema_defns = document.definitions.select { |d| d.is_a?(GraphQL::Language::Nodes::SchemaDefinition) }
+          definitions = document.definitions #: Array[untyped]
+          schema_defns = definitions.select { |d| d.is_a?(GraphQL::Language::Nodes::SchemaDefinition) }
           if schema_defns.size > 1
             raise InvalidDocumentError.new('Must provide only one schema definition.')
           end
-          schema_definition = schema_defns.first
+          schema_definition = schema_defns.first #: GraphQL::Language::Nodes::SchemaDefinition?
           types = {}
           directives = schema_superclass.directives.dup
           # Directive applications are queued while types are built, then applied once every type is complete,
           # because applying a directive coerces its arguments, which requires fully-built input types.
           queued_directives = []
-          type_resolver = TypeResolver.new(types, directives, queued_directives) { |type_name| types[type_name] ||= Schema::LateBoundType.new(type_name) }
+          definitions_by_name = {}
+          # Classes (or modules, for interfaces) made before any type is built, so that references to them don't need late-bound types
+          type_classes = {}
+          interface_definitions = {}
+          definitions.each do |d|
+            if d.respond_to?(:name)
+              definitions_by_name[d.name] ||= d
+              if !type_classes.key?(d.name)
+                base_type = case d
+                when GraphQL::Language::Nodes::ObjectTypeDefinition then base_types[:object]
+                when GraphQL::Language::Nodes::EnumTypeDefinition then base_types[:enum]
+                when GraphQL::Language::Nodes::InputObjectTypeDefinition then base_types[:input_object]
+                when GraphQL::Language::Nodes::UnionTypeDefinition then base_types[:union]
+                when GraphQL::Language::Nodes::ScalarTypeDefinition then base_types[:scalar]
+                when GraphQL::Language::Nodes::InterfaceTypeDefinition
+                  interface_definitions[d.name] = d
+                  type_classes[d.name] = new_interface_module(d.name, base_types[:interface])
+                  nil
+                end
+                if base_type
+                  type_classes[d.name] = Class.new(base_type) {
+                    #: self as singleton(GraphQL::Schema::Member)
+                    graphql_name(d.name)
+                  }
+                end
+              end
+            end
+          end
+          type_resolver = TypeResolver.new(types, directives, queued_directives, type_classes) { |type_name|
+            types[type_name] ||= if (type_class = type_classes[type_name])
+              type_class
+            elsif !definitions_by_name.key?(type_name) && (built_in_type = GraphQL::Schema::BUILT_IN_TYPES[type_name])
+              built_in_type
+            else
+              Schema::LateBoundType.new(type_name)
+            end
+          }
           # Make a different type resolver because directive argument types must be real classes (not late-bound)
           # when queued directives are applied. It builds types eagerly. (Each type registers itself in `types`
           # before building its members, so cycles between input objects terminate.)
           directive_type_resolver = nil
-          directive_type_resolver = TypeResolver.new(types, directives, queued_directives) { |type_name|
+          directive_type_resolver = TypeResolver.new(types, directives, queued_directives, type_classes) { |type_name|
             types[type_name] ||= begin
-              defn = document.definitions.find { |d| d.respond_to?(:name) && d.name == type_name }
+              defn = definitions_by_name[type_name]
               if defn
                 build_definition_from_node(defn, directive_type_resolver, default_resolve, base_types)
               elsif (built_in_defn = GraphQL::Schema::BUILT_IN_TYPES[type_name])
@@ -76,8 +113,20 @@ module GraphQL
             end
           }
 
+          build_interface = ->(interface_name, resolver) {
+            if (interface_definition = interface_definitions.delete(interface_name)) && type_classes[interface_name]
+              check_transitive_interfaces(interface_definition, definitions_by_name)
+              build_interface_type(interface_definition, resolver, base_types[:interface])
+            end
+          }
+          type_resolver.build_interface = build_interface
+          directive_type_resolver.build_interface = build_interface
+          type_links = TypeLinks.new
+          type_resolver.links = type_links
+          directive_type_resolver.links = type_links
+
           directives.merge!(GraphQL::Schema.default_directives)
-          document.definitions.each do |definition|
+          definitions.each do |definition|
             if definition.is_a?(GraphQL::Language::Nodes::DirectiveDefinition)
               directives[definition.name] = build_directive(definition, directive_type_resolver)
             end
@@ -86,9 +135,8 @@ module GraphQL
           # In case any directives referenced built-in types for their arguments:
           replace_late_bound_types_with_built_in(types)
 
-          schema_extensions = nil
-          definitions_by_name = nil
-          document.definitions.each do |definition|
+          schema_extensions = nil #: Array[untyped]?
+          definitions.each do |definition|
             case definition
             when GraphQL::Language::Nodes::SchemaDefinition, GraphQL::Language::Nodes::DirectiveDefinition
               nil # already handled
@@ -104,24 +152,10 @@ module GraphQL
             else
               # It's possible that this was already loaded by the directives
               prev_type = types[definition.name]
-              if prev_type.nil? || prev_type.is_a?(Schema::LateBoundType)
+              if prev_type.nil? || prev_type.is_a?(Schema::LateBoundType) || type_classes[definition.name]
                 if definition.is_a?(GraphQL::Language::Nodes::ObjectTypeDefinition) || definition.is_a?(Language::Nodes::InterfaceTypeDefinition)
-                  interface_names = definition.interfaces.map(&:name)
-                  if !interface_names.empty?
-                    definitions_by_name ||= document.definitions.each_with_object({}) do |d, by_name|
-                      by_name[d.name] ||= d if d.respond_to?(:name)
-                    end
-                    transitive_names = interface_names.map { |n| definitions_by_name[n]&.interfaces&.map(&:name) }
-                    transitive_names.flatten!
-                    transitive_names.compact!
-                  else
-                    transitive_names = interface_names
-                  end
-                  if !(missing_transitive_interfaces = transitive_names - interface_names).empty?
-                    raise GraphQL::Schema::InvalidDocumentError, "type #{definition.name} is missing one or more transitive interface names: #{missing_transitive_interfaces.join(", ")}. Add them to the type's `implements` list and try again."
-                  end
+                  check_transitive_interfaces(definition, definitions_by_name)
                 end
-
                 build_definition_from_node(definition, type_resolver, default_resolve, base_types)
               end
             end
@@ -196,13 +230,19 @@ module GraphQL
           builder = self
 
           found_types = types.values
-          object_types = found_types.select { |t| t.respond_to?(:kind) && t.kind.object? }
+          object_types = found_types.select { |t| t.is_a?(Class) && t < GraphQL::Schema::Object }
           schema_class = Class.new(schema_superclass) do
             #: self as singleton(GraphQL::Schema)
             begin
               # Add these first so that there's some chance of resolving late-bound types
-              add_type_and_traverse(found_types, root: false)
-              orphan_types(object_types)
+              if !builder.add_linked_types(self, found_types, type_links, directives, base_types)
+                add_type_and_traverse(found_types, root: false)
+              end
+              # Without visibility, this would only re-traverse types that were just added
+              # (and `own_orphan_types` is cleared below), so it can be skipped.
+              if use_visibility_profile? || visibility
+                orphan_types(object_types)
+              end
               query query_root_type
               mutation mutation_root_type
               subscription subscription_root_type
@@ -255,7 +295,7 @@ module GraphQL
               attr_accessor :definition_default_resolve
             end
 
-            self.definition_default_resolve = default_resolve
+            @definition_default_resolve = default_resolve
 
             def definition_default_resolve
               self.class.definition_default_resolve
@@ -274,9 +314,131 @@ module GraphQL
           end
           apply_queued_directives(queued_directives)
 
+          directives.each_value do |dir_class|
+            if dir_class.respond_to?(:coerced_arguments_cache=) && dir_class.coerced_arguments_cache
+              dir_class.coerced_arguments_cache = nil
+            end
+          end
+
           schema_class
         end
 
+        # References between types in this document, recorded while they're built.
+        class TypeLinks
+          def initialize
+            @references = {}.compare_by_identity
+            @arguments = []
+            @complete = true
+          end
+
+          # Hash of type => Array of the fields and arguments that use it (no duplicates, since each one is added once)
+          attr_reader :references
+
+          attr_reader :arguments
+
+          # True if everything was recorded, so the generic traversal isn't needed
+          def complete?
+            @complete
+          end
+
+          def incomplete!
+            @complete = false
+          end
+
+          def add_argument(argument, argument_type)
+            @arguments << argument
+            add_reference(argument, argument_type)
+          end
+
+          def add_reference(member, member_type)
+            member_type = member_type.unwrap
+            if member_type.is_a?(Schema::LateBoundType)
+              # The generic traversal will raise an error for this
+              incomplete!
+            else
+              (@references[member_type] ||= []) << member
+            end
+          end
+        end
+
+        # Like `schema_class.add_type_and_traverse(found_types, root: false)`, but it uses the links recorded while types were built
+        # instead of a traversal. (Directive classes are added later, with `directives(...)`.)
+        # Returns false if the traversal is required, for example, when base types have their own fields or interfaces.
+        def add_linked_types(schema_class, found_types, type_links, directives, base_types)
+          if !type_links.complete? ||
+              schema_class.use_visibility_profile? ||
+              # A redefined `@deprecated` isn't the one used by `deprecation_reason:`, so the traversal would find another directive.
+              !directives["deprecated"].equal?(GraphQL::Schema::Directive::Deprecated) ||
+              !base_types_without_members?(base_types)
+            return false
+          end
+
+          own_types = schema_class.send(:own_types)
+          found_types.each do |type|
+            if !(Module === type)
+              return false
+            end
+            type = type #: as GraphQL::Schema::Member::type_definition
+            if (Class === type && GraphQL::Schema::Directive > type) ||
+                ((prev_type = own_types[type.graphql_name]) && !prev_type.equal?(type))
+              return false
+            end
+          end
+
+          # Like the traversal, check each argument's type now (it raises an error if the type is invalid)
+          type_links.arguments.each(&:type)
+
+          possible_types = {}.compare_by_identity
+          possible_types_sets = {}.compare_by_identity
+          union_memberships = {}
+          found_types.each do |type|
+            own_types[type.graphql_name] = type
+            if Class === type
+              if GraphQL::Schema::Object > type
+                (possible_types[type] ||= []) << type
+                object_type = type #: as singleton(GraphQL::Schema::Object)
+                object_type.interface_type_memberships.each do |type_membership|
+                  interface_type = type_membership.abstract_type
+                  if (possible_types_sets[interface_type] ||= Set.new).add?(type)
+                    (possible_types[interface_type] ||= []) << type
+                  end
+                end
+              elsif GraphQL::Schema::Union > type
+                union_type = type #: as singleton(GraphQL::Schema::Union)
+                member_types = possible_types[type] = union_type.all_possible_types
+                member_types.each do |member_type|
+                  (union_memberships[member_type.graphql_name] ||= []) << type
+                end
+              end
+            end
+          end
+
+          schema_class.send(:own_possible_types).merge!(possible_types) { |key, old_val, new_val| old_val + new_val }
+          schema_class.send(:own_union_memberships).merge!(union_memberships)
+          own_references_to = schema_class.send(:own_references_to)
+          type_links.references.each do |thing, new_refs|
+            prev_refs = own_references_to[thing]
+            own_references_to[thing] = prev_refs ? (prev_refs + (new_refs - prev_refs)) : new_refs
+          end
+          type_links.arguments.each do |argument|
+            if argument.default_value?
+              argument.validate_default_value
+            end
+          end
+          true
+        end
+
+        def base_types_without_members?(base_types)
+          object = base_types[:object]
+          interface = base_types[:interface]
+          object.all_field_definitions.empty? && object.interface_type_memberships.empty? && object.directives.empty? &&
+            interface.all_field_definitions.empty? && interface.interface_type_memberships.empty? && interface.directives.empty? && interface.orphan_types.empty? &&
+            base_types[:input_object].all_argument_definitions.empty? && base_types[:input_object].directives.empty? &&
+            base_types[:union].all_possible_types.empty? && base_types[:union].directives.empty? &&
+            base_types[:enum].all_enum_value_definitions.empty? && base_types[:enum].directives.empty? &&
+            base_types[:scalar].directives.empty?
+        end
+
         NullResolveType = ->(type, obj, ctx) {
           raise(GraphQL::RequiredImplementationMissingError, "Generated Schema cannot use Interface or Union types for execution. Implement resolve_type on your resolver.")
         }
@@ -326,7 +488,7 @@ module GraphQL
         def build_directives(definition, ast_node, type_resolver)
           return if ast_node.directives.empty?
 
-          if ast_node.directives.all? { |dir_node| dir_node.name == "deprecated" || type_resolver.directive_class(dir_node.name)&.all_argument_definitions&.empty? }
+          if ast_node.directives.all? { |dir_node| dir_node.name == "deprecated" || type_resolver.directive_without_arguments?(dir_node.name) }
             apply_directives(definition, ast_node, type_resolver)
           else
             type_resolver.queued_directives << [definition, ast_node, type_resolver]
@@ -340,45 +502,34 @@ module GraphQL
         end
 
         def apply_directives(definition, ast_node, type_resolver)
-          dirs = prepare_directives(ast_node, type_resolver)
-          dirs.each do |(dir_class, options)|
-            if definition.respond_to?(:schema_directive)
-              # it's a schema
-              definition.schema_directive(dir_class, **options)
-            else
-              definition.directive(dir_class, **options)
-            end
-          end
-        end
-
-        def prepare_directives(ast_node, type_resolver)
-          dirs = []
+          is_schema = Class === definition && GraphQL::Schema >= definition
           ast_node.directives.each do |dir_node|
-            if dir_node.name == "deprecated"
-              # This is handled using `deprecation_reason`
-              next
+            # This is handled using `deprecation_reason`
+            next if dir_node.name == "deprecated"
+            dir_class = type_resolver.call(dir_node.name)
+            if dir_class.nil?
+              raise ArgumentError, "No definition for @#{dir_node.name} #{ast_node.respond_to?(:name) ? "on #{ast_node.name} " : ""}at #{ast_node.line}:#{ast_node.col}"
+            end
+            options = args_to_kwargs(dir_class, dir_node, type_resolver)
+            if is_schema
+              schema_class = definition #: as singleton(GraphQL::Schema)
+              schema_class.schema_directive(dir_class, **options)
             else
-              dir_class = type_resolver.call(dir_node.name)
-              if dir_class.nil?
-                raise ArgumentError, "No definition for @#{dir_node.name} #{ast_node.respond_to?(:name) ? "on #{ast_node.name} " : ""}at #{ast_node.line}:#{ast_node.col}"
-              end
-              options = args_to_kwargs(dir_class, dir_node)
-              dirs << [dir_class, options]
+              definition.directive(dir_class, **options)
             end
           end
-          dirs
         end
 
-        def args_to_kwargs(arg_owner, node)
+        def args_to_kwargs(arg_owner, node, type_resolver)
           if node.respond_to?(:arguments)
             kwargs = {}
             node.arguments.each do |arg_node|
-              arg_defn = arg_owner.get_argument(arg_node.name)
-              kwargs[arg_defn.keyword] = args_to_kwargs(arg_defn.type.unwrap, arg_node.value)
+              arg_defn = type_resolver.get_argument(arg_owner, arg_node.name)
+              kwargs[arg_defn.keyword] = args_to_kwargs(arg_defn.type.unwrap, arg_node.value, type_resolver)
             end
             kwargs
           elsif node.is_a?(Array)
-            node.map { |n| args_to_kwargs(arg_owner, n) }
+            node.map { |n| args_to_kwargs(arg_owner, n, type_resolver) }
           elsif node.is_a?(Language::Nodes::Enum)
             node.name
           else
@@ -388,16 +539,13 @@ module GraphQL
         end
 
         def build_enum_type(enum_type_definition, type_resolver, base_type)
-          builder = self
-          Class.new(base_type) do
-            #: self as singleton(GraphQL::Schema::Enum)
-            graphql_name(enum_type_definition.name)
-            type_resolver.register(self)
-            builder.build_directives(self, enum_type_definition, type_resolver)
-            description(enum_type_definition.description)
-            ast_node(enum_type_definition)
-            builder.build_values(self, enum_type_definition.values, type_resolver)
-          end
+          type_class = type_resolver.type_class(enum_type_definition.name, base_type) #: singleton(GraphQL::Schema::Enum)
+          type_resolver.register(type_class)
+          build_directives(type_class, enum_type_definition, type_resolver)
+          type_class.description(enum_type_definition.description)
+          type_class.ast_node(enum_type_definition)
+          build_values(type_class, enum_type_definition.values, type_resolver)
+          type_class
         end
 
         def build_values(type_class, enum_value_definitions, type_resolver)
@@ -413,32 +561,34 @@ module GraphQL
         end
 
         def build_deprecation_reason(directives)
-          deprecated_directive = directives.find{ |d| d.name == 'deprecated' }
-          return unless deprecated_directive
-
-          reason = deprecated_directive.arguments.find{ |a| a.name == 'reason' }
-          return GraphQL::Schema::Directive::DEFAULT_DEPRECATION_REASON unless reason
-
-          reason.value
+          return if directives.empty?
+          directives.each do |d|
+            if d.name == "deprecated"
+              d.arguments.each do |a|
+                if a.name == "reason"
+                  return a.value
+                end
+              end
+              return GraphQL::Schema::Directive::DEFAULT_DEPRECATION_REASON
+            end
+          end
+          nil
         end
 
         def build_scalar_type(scalar_type_definition, type_resolver, base_type, default_resolve:)
-          builder = self
-          Class.new(base_type) do
-            #: self as singleton(GraphQL::Schema::Scalar)
-            graphql_name(scalar_type_definition.name)
-            type_resolver.register(self)
-            description(scalar_type_definition.description)
-            ast_node(scalar_type_definition)
-            builder.build_directives(self, scalar_type_definition, type_resolver)
-
-            if default_resolve.respond_to?(:coerce_input)
-              # Put these method definitions in another method to avoid retaining `type_resolve`
-              # from this method's bindiing
-              builder.build_scalar_type_coerce_method(self, :coerce_input, default_resolve)
-              builder.build_scalar_type_coerce_method(self, :coerce_result, default_resolve)
-            end
-          end
+          type_class = type_resolver.type_class(scalar_type_definition.name, base_type) #: singleton(GraphQL::Schema::Scalar)
+          type_resolver.register(type_class)
+          type_class.description(scalar_type_definition.description)
+          type_class.ast_node(scalar_type_definition)
+          build_directives(type_class, scalar_type_definition, type_resolver)
+
+          if default_resolve.respond_to?(:coerce_input)
+            # Put these method definitions in another method to avoid retaining `type_resolve`
+            # from this method's bindiing
+            build_scalar_type_coerce_method(type_class, :coerce_input, default_resolve)
+            build_scalar_type_coerce_method(type_class, :coerce_result, default_resolve)
+          end
+          type_class
         end
 
         def build_scalar_type_coerce_method(scalar_class, method_name, default_definition_resolve)
@@ -448,52 +598,70 @@ module GraphQL
         end
 
         def build_union_type(union_type_definition, type_resolver, base_type)
-          builder = self
-          Class.new(base_type) do
-            #: self as singleton(GraphQL::Schema::Union)
-            graphql_name(union_type_definition.name)
-            type_resolver.register(self)
-            description(union_type_definition.description)
-            possible_types(*union_type_definition.types.map { |type_name| type_resolver.call(type_name) })
-            ast_node(union_type_definition)
-            builder.build_directives(self, union_type_definition, type_resolver)
-          end
+          type_class = type_resolver.type_class(union_type_definition.name, base_type) #: singleton(GraphQL::Schema::Union)
+          type_resolver.register(type_class)
+          type_class.description(union_type_definition.description)
+          type_class.possible_types(*union_type_definition.types.map { |type_name| type_resolver.call(type_name) })
+          type_class.ast_node(union_type_definition)
+          build_directives(type_class, union_type_definition, type_resolver)
+          type_class
         end
 
         def build_object_type(object_type_definition, type_resolver, base_type)
-          builder = self
-
-          Class.new(base_type) do
-            #: self as singleton(GraphQL::Schema::Object)
-            graphql_name(object_type_definition.name)
-            type_resolver.register(self)
-            description(object_type_definition.description)
-            ast_node(object_type_definition)
-            builder.build_directives(self, object_type_definition, type_resolver)
-            builder.build_interfaces(self, object_type_definition.interfaces, type_resolver)
-            builder.build_fields(self, object_type_definition.fields, type_resolver, default_resolve: true)
-          end
+          type_class = type_resolver.type_class(object_type_definition.name, base_type) #: singleton(GraphQL::Schema::Object)
+          type_resolver.register(type_class)
+          type_class.description(object_type_definition.description)
+          type_class.ast_node(object_type_definition)
+          build_directives(type_class, object_type_definition, type_resolver)
+          build_interfaces(type_class, object_type_definition.interfaces, type_resolver)
+          build_fields(type_class, object_type_definition.fields, type_resolver, default_resolve: true)
+          type_class
         end
 
         def build_interfaces(type_class, interface_names, type_resolver)
           interface_names.each do |interface_name|
+            # Implementing an interface copies some of its configuration, so it must be built first
+            type_resolver.build_interface_now(interface_name.name)
             type_class.implements(type_resolver.call(interface_name))
           end
         end
 
-        def build_input_object_type(input_object_type_definition, type_resolver, base_type)
-          builder = self
-          Class.new(base_type) do
-            #: self as singleton(GraphQL::Schema::InputObject)
-            graphql_name(input_object_type_definition.name)
-            type_resolver.register(self)
-            description(input_object_type_definition.description)
-            ast_node(input_object_type_definition)
-            builder.build_directives(self, input_object_type_definition, type_resolver)
-            builder.build_arguments(self, input_object_type_definition.fields, type_resolver)
+        def check_transitive_interfaces(definition, definitions_by_name)
+          interface_names = definition.interfaces.map(&:name)
+          if !interface_names.empty?
+            transitive_names = interface_names.map { |n|
+              defn = definitions_by_name[n]
+              # Other kinds of types don't have `interfaces`; `implements` raises a better error for them later.
+              defn.is_a?(Language::Nodes::InterfaceTypeDefinition) ? defn.interfaces.map(&:name) : nil
+            }
+            transitive_names.flatten!
+            transitive_names.compact!
+          else
+            transitive_names = interface_names
+          end
+          if !(missing_transitive_interfaces = transitive_names - interface_names).empty?
+            raise GraphQL::Schema::InvalidDocumentError, "type #{definition.name} is missing one or more transitive interface names: #{missing_transitive_interfaces.join(", ")}. Add them to the type's `implements` list and try again."
           end
         end
 
+        def new_interface_module(name, base_type)
+          Module.new {
+            #: self as GraphQL::Schema::Member::interface_definition
+            include base_type
+            graphql_name(name)
+          }
+        end
+
+        def build_input_object_type(input_object_type_definition, type_resolver, base_type)
+          type_class = type_resolver.type_class(input_object_type_definition.name, base_type) #: singleton(GraphQL::Schema::InputObject)
+          type_resolver.register(type_class)
+          type_class.description(input_object_type_definition.description)
+          type_class.ast_node(input_object_type_definition)
+          build_directives(type_class, input_object_type_definition, type_resolver)
+          build_arguments(type_class, input_object_type_definition.fields, type_resolver)
+          type_class
+        end
+
         def build_default_value(default_value)
           case default_value
           when GraphQL::Language::Nodes::Enum
@@ -511,7 +679,8 @@ module GraphQL
 
         def build_arguments(type_class, arguments, type_resolver)
           builder = self
-
+          # Directive arguments are linked when directives are added to the schema
+          links = (Class === type_class && GraphQL::Schema::Directive > type_class) ? nil : type_resolver.links
           arguments.each do |argument_defn|
             default_value_kwargs = if !argument_defn.default_value.nil?
               { default_value: builder.build_default_value(argument_defn.default_value) }
@@ -519,9 +688,10 @@ module GraphQL
               EMPTY_HASH
             end
 
+            argument_type = type_resolver.call(argument_defn.type)
             argument = type_class.argument(
               argument_defn.name,
-              type: type_resolver.call(argument_defn.type),
+              type: argument_type,
               required: false,
               description: argument_defn.description,
               deprecation_reason: builder.build_deprecation_reason(argument_defn.directives),
@@ -529,6 +699,7 @@ module GraphQL
               camelize: false,
               **default_value_kwargs
             )
+            links&.add_argument(argument, argument_type)
             builder.build_directives(argument, argument_defn, type_resolver)
           end
         end
@@ -542,79 +713,159 @@ module GraphQL
             locations(*directive_definition.locations.map { |location| location.name.to_sym })
             ast_node(directive_definition)
             builder.build_arguments(self, directive_definition.arguments, type_resolver)
+            self.coerced_arguments_cache = {}
           end
         end
 
         def build_interface_type(interface_type_definition, type_resolver, base_type)
-          builder = self
-          Module.new do
-            #: self as GraphQL::Schema::Member::interface_definition
-            include base_type
-            graphql_name(interface_type_definition.name)
-            type_resolver.register(self)
-            description(interface_type_definition.description)
-            builder.build_interfaces(self, interface_type_definition.interfaces, type_resolver)
-            ast_node(interface_type_definition)
-            builder.build_directives(self, interface_type_definition, type_resolver)
-
-            builder.build_fields(self, interface_type_definition.fields, type_resolver, default_resolve: nil)
-          end
+          type_module = type_resolver.interface_module(interface_type_definition.name, base_type) #: GraphQL::Schema::Member::interface_definition
+          type_resolver.register(type_module)
+          type_module.description(interface_type_definition.description)
+          build_interfaces(type_module, interface_type_definition.interfaces, type_resolver)
+          type_module.ast_node(interface_type_definition)
+          build_directives(type_module, interface_type_definition, type_resolver)
+          build_fields(type_module, interface_type_definition.fields, type_resolver, default_resolve: nil)
+          type_module
         end
 
         def build_fields(owner, field_definitions, type_resolver, default_resolve:)
           builder = self
-
+          # If `owner.field` isn't customized, make fields directly. It's the same, but faster.
+          field_class = owner.method(:field).owner.equal?(GraphQL::Schema::Member::HasFields) ? owner.field_class : nil
+          resolve_method_names = type_resolver.field_resolve_methods.method_names
+          links = type_resolver.links
+          # Custom field classes might add arguments (for example, with extensions), which the links wouldn't include
+          if !field_class.equal?(GraphQL::Schema::Field)
+            links.incomplete!
+          end
           field_definitions.each do |field_definition|
-            resolve_method_name = -"resolve_field_#{field_definition.name}"
-            schema_field_defn = owner.field(
-              field_definition.name,
-              description: field_definition.description,
-              type: type_resolver.call(field_definition.type),
-              null: true,
-              connection_extension: nil,
-              deprecation_reason: build_deprecation_reason(field_definition.directives),
-              ast_node: field_definition,
-              method_conflict_warning: false,
-              camelize: false,
-              resolver_method: resolve_method_name,
-              resolve_batch: resolve_method_name,
-            )
+            resolve_method_name = resolve_method_names[field_definition.name]
+            if field_class
+              schema_field_defn = field_class.new(
+                name: field_definition.name,
+                owner: owner,
+                description: field_definition.description,
+                type: (field_type = type_resolver.call(field_definition.type)),
+                null: true,
+                connection_extension: nil,
+                deprecation_reason: build_deprecation_reason(field_definition.directives),
+                ast_node: field_definition,
+                method_conflict_warning: false,
+                camelize: false,
+                resolver_method: resolve_method_name,
+                resolve_batch: resolve_method_name,
+              )
+              owner.add_field(schema_field_defn)
+            else
+              schema_field_defn = owner.field(
+                field_definition.name,
+                description: field_definition.description,
+                type: (field_type = type_resolver.call(field_definition.type)),
+                null: true,
+                connection_extension: nil,
+                deprecation_reason: build_deprecation_reason(field_definition.directives),
+                ast_node: field_definition,
+                method_conflict_warning: false,
+                camelize: false,
+                resolver_method: resolve_method_name,
+                resolve_batch: resolve_method_name,
+              )
+            end
 
+            links.add_reference(schema_field_defn, field_type)
             builder.build_directives(schema_field_defn, field_definition, type_resolver)
             builder.build_arguments(schema_field_defn, field_definition.arguments, type_resolver)
 
             # Don't do this for interfaces
             if default_resolve
-              define_field_resolve_method(owner, resolve_method_name, field_definition.name)
+              define_field_resolve_method(owner, resolve_method_name, field_definition.name, type_resolver)
             end
           end
         end
 
-        def define_field_resolve_method(owner, method_name, field_name)
-          owner.define_method(method_name) { |**args|
-            #: self as GraphQL::Schema::Object
-            field_instance = context.types.field(owner, field_name)
-            context.schema.definition_default_resolve.call(self.class, field_instance, object, args, context)
-          }
-          owner.define_singleton_method(method_name) { |objects, context, **args|
-            field_instance = context.types.field(owner, field_name)
-            objects.map do |object|
-              context.schema.definition_default_resolve.call(self, field_instance, object, args, context)
-            end
-          }
+        def define_field_resolve_method(owner, method_name, field_name, type_resolver)
+          resolve_methods = type_resolver.field_resolve_methods
+          resolve_methods.field_names[method_name.to_sym] = field_name
+          owner.define_method(method_name, &resolve_methods.instance_method_proc)
+          owner.define_singleton_method(method_name, &resolve_methods.singleton_method_proc)
+        end
+
+        # Every field's resolve methods share these procs. They use the method's name to find the field.
+        class FieldResolveMethods
+          def initialize
+            @method_names = Hash.new { |h, field_name| h[field_name] = -"resolve_field_#{field_name}" }
+            field_names = @field_names = {}
+            @instance_method_proc = proc { |**args|
+              #: self as GraphQL::Schema::Object
+              owner = self.class
+              field_instance = context.types.field(owner, field_names[__method__])
+              context.schema.definition_default_resolve.call(owner, field_instance, object, args, context)
+            }
+            @singleton_method_proc = proc { |objects, context, **args|
+              field_instance = context.types.field(self, field_names[__method__])
+              objects.map do |object|
+                context.schema.definition_default_resolve.call(self, field_instance, object, args, context)
+              end
+            }
+          end
+
+          # Field names by resolve method name
+          attr_reader :field_names
+
+          # Resolve method names by field name
+          attr_reader :method_names
+
+          attr_reader :instance_method_proc, :singleton_method_proc
         end
 
         class TypeResolver
-          def initialize(lookup_hash, directives, queued_directives, &missing_type_handler)
+          def initialize(lookup_hash, directives, queued_directives, type_classes, &missing_type_handler)
             @lookup_hash = lookup_hash
             @directives = directives
             @queued_directives = queued_directives
+            @type_classes = type_classes
             @missing_type_handler = missing_type_handler
           end
 
+          # The class made ahead of time for `type_name` (returned once, to be built), or a new subclass of `base_type`
+          def type_class(type_name, base_type)
+            @type_classes.delete(type_name) || Class.new(base_type) {
+              #: self as singleton(GraphQL::Schema::Member)
+              graphql_name(type_name)
+            }
+          end
+
+          def interface_module(type_name, base_type)
+            @type_classes.delete(type_name) || Builder.new_interface_module(type_name, base_type)
+          end
+
+          attr_writer :build_interface
+
+          attr_accessor :links
+
+          def build_interface_now(interface_name)
+            @build_interface&.call(interface_name, self)
+          end
+
           # Directive applications waiting for {Builder#apply_queued_directives}
           attr_reader :queued_directives
 
+          # Like `owner.get_argument(argument_name)`, but cached. Directives are applied after types are built,
+          # so their arguments don't change while this cache is used.
+          def get_argument(owner, argument_name)
+            @arguments_by_owner ||= {}.compare_by_identity
+            owner_arguments = @arguments_by_owner[owner] ||= {}
+            if owner_arguments.key?(argument_name)
+              owner_arguments[argument_name]
+            else
+              owner_arguments[argument_name] = owner.get_argument(argument_name)
+            end
+          end
+
+          def field_resolve_methods
+            @field_resolve_methods ||= FieldResolveMethods.new
+          end
+
           # Add `type` to the lookup hash. Types call this before building their members, so that references back to them (including cycles) find this type.
           def register(type)
             @lookup_hash[type.graphql_name] = type
@@ -625,6 +876,17 @@ module GraphQL
             @directives[name]
           end
 
+          def directive_without_arguments?(name)
+            @directives_without_arguments ||= {}
+            if @directives_without_arguments.key?(name)
+              @directives_without_arguments[name]
+            else
+              dir_class = @directives[name]
+              # Don't cache unknown directives, they might be defined later
+              dir_class ? (@directives_without_arguments[name] = dir_class.all_argument_definitions.empty?) : false
+            end
+          end
+
           def call(ast_node)
             case ast_node
             when GraphQL::Language::Nodes::TypeName