Merge pull request #5603 from rmosolgo/next-rename

Exec-next: remove next namespacing

Author: rmosolgo

lib/graphql/execution/field_resolve_step.rb

@@ -0,0 +1,217 @@
+# frozen_string_literal: true
+module GraphQL
+  module Execution
+    class Finalize
+      def initialize(query, data, runner)
+        @query = query
+        @data = data
+        @static_type_at = runner.static_type_at
+        @runner = runner
+        @current_exec_path = query.path.dup
+        @current_result_path = query.path.dup
+        @finalizers = runner.finalizers ? runner.finalizers[query] : {}.compare_by_identity
+        @finalizers_count = 0
+        @finalizers.each do |key, values|
+          values.each do |key2, values2|
+            case values2
+            when Array
+              @finalizers_count += values2.size
+            else
+              @finalizers_count += 1
+            end
+          end
+        end
+
+        query.context.errors.each do |err|
+          err_path = err.path - @current_exec_path
+          key = err_path.pop
+          targets = [data]
+          while (part = err_path.shift)
+            targets.map! { |t| t[part] }
+            targets.flatten!
+          end
+
+          targets.each_with_index do |target, idx|
+            if target.is_a?(Hash)
+              if target[key].equal?(err)
+                tf = @finalizers[target] ||= {}.compare_by_identity
+                tf[key] = err
+                @finalizers_count += 1
+              elsif (arr = target[key]).is_a?(Array)
+                arr.each_with_index do |el, idx|
+                  if el.equal?(err)
+                    tf = @finalizers[arr] ||= {}.compare_by_identity
+                    tf[idx] = err
+                    @finalizers_count += 1
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+
+      def run
+        if (selected_operation = @query.selected_operation) && @data
+          if @data.is_a?(Hash)
+            check_object_result(@data, @query.root_type, selected_operation.selections)
+          elsif @data.is_a?(Array)
+            check_list_result(@data, @query.root_type, selected_operation.selections)
+          elsif @data.is_a?(Finalizer)
+            dummy_data = {}
+            dummy_key = "__dummy"
+            @data.path = @query.path
+            @data.finalize_graphql_result(@query, dummy_data, dummy_key)
+            dummy_data[dummy_key]
+          else
+            raise ArgumentError, "Unexpected @data: #{@data.inspect}"
+          end
+        else
+          @data
+        end
+      end
+
+      private
+
+      def run_finalizers(result_path, finalizer_or_finalizers, result_data, result_key)
+        if finalizer_or_finalizers.is_a?(Array)
+          finalizer_or_finalizers.each { |f|
+            f.path = result_path
+            f.finalize_graphql_result(@query, result_data, result_key)
+          }
+          @finalizers_count -= finalizer_or_finalizers.size
+        else
+          f = finalizer_or_finalizers
+          f.path = result_path
+          f.finalize_graphql_result(@query, result_data, result_key)
+          @finalizers_count -= 1
+        end
+      end
+
+      def finalizers(result_value, key)
+        finalizers_for_value = @finalizers[result_value]
+        finalizers_for_value && finalizers_for_value[key]
+      end
+
+      def check_object_result(result_h, parent_type, ast_selections)
+        if (f = finalizers(result_h, nil))
+          run_finalizers(@current_result_path.dup, f, result_h, nil)
+          return result_h if @finalizers_count == 0
+        end
+
+        if parent_type.kind.abstract?
+          parent_type = @runner.runtime_type_at[result_h]
+        end
+
+        ast_selections.each do |ast_selection|
+          case ast_selection
+          when Language::Nodes::Field
+            key = ast_selection.alias || ast_selection.name
+            if (f = finalizers(result_h, key))
+              result_value = result_h[key]
+              run_finalizers(@current_result_path.dup << key, f, result_h, key)
+              new_result_value = result_h.key?(key) ? result_h[key] : :unassigned
+            end
+            next if !(f || result_h.key?(key))
+            begin
+              @current_exec_path << key
+              @current_result_path << key
+
+              field_defn = @query.context.types.field(parent_type, ast_selection.name) || raise("Invariant: No field found for #{static_type.to_type_signature}.#{ast_selection.name}")
+              result_type = field_defn.type
+              if (result_type_non_null = result_type.non_null?)
+                result_type = result_type.of_type
+              end
+
+              if !f
+                result_value = result_h[key]
+                new_result_value = if result_type.list? && result_value
+                  check_list_result(result_value, result_type.of_type, ast_selection.selections)
+                elsif !result_type.kind.leaf? && result_value
+                  check_object_result(result_value, result_type, ast_selection.selections)
+                else
+                  result_value
+                end
+              end
+
+              if new_result_value.nil? && result_type_non_null
+                return nil
+              elsif :unassigned.equal?(new_result_value)
+                # Do nothing
+                break if @finalizers_count == 0
+              elsif !new_result_value.equal?(result_value)
+                result_h[key] = new_result_value
+                break if @finalizers_count == 0
+              end
+            ensure
+              @current_exec_path.pop
+              @current_result_path.pop
+            end
+          when Language::Nodes::InlineFragment
+            static_type_at_result = @static_type_at[result_h]
+            if static_type_at_result && (
+                (t = ast_selection.type).nil? ||
+                @runner.type_condition_applies?(@query.context, static_type_at_result, t.name)
+              )
+              result_h = check_object_result(result_h, parent_type, ast_selection.selections)
+            end
+          when Language::Nodes::FragmentSpread
+            fragment_defn = @query.document.definitions.find { |defn| defn.is_a?(Language::Nodes::FragmentDefinition) && defn.name == ast_selection.name }
+            static_type_at_result = @static_type_at[result_h]
+            if static_type_at_result && @runner.type_condition_applies?(@query.context, static_type_at_result, fragment_defn.type.name)
+              result_h = check_object_result(result_h, parent_type, fragment_defn.selections)
+            end
+          end
+        end
+
+        result_h
+      end
+
+      def check_list_result(result_arr, inner_type, ast_selections)
+        inner_type_non_null = false
+        if inner_type.non_null?
+          inner_type_non_null = true
+          inner_type = inner_type.of_type
+        end
+
+        new_invalid_null = false
+
+        if (f = finalizers(result_arr, nil))
+          run_finalizers(@current_result_path.dup, f, result_arr, nil)
+          return result_arr if @finalizers_count == 0
+        end
+
+        result_arr.each_with_index do |result_item, idx|
+          @current_result_path << idx
+          new_result = if (f = finalizers(result_arr, idx))
+            run_finalizers(@current_result_path.dup, f, result_arr, idx)
+            result_arr[idx]
+          elsif inner_type.list? && result_item
+            check_list_result(result_item, inner_type.of_type, ast_selections)
+          elsif !inner_type.kind.leaf? && result_item
+            check_object_result(result_item, inner_type, ast_selections)
+          else
+            result_item
+          end
+
+          if new_result.nil? && inner_type_non_null
+            new_invalid_null = true
+            result_arr[idx] = nil
+            break if @finalizers_count == 0
+          elsif !new_result.equal?(result_item)
+            result_arr[idx] = new_result
+            break if @finalizers_count == 0
+          end
+        ensure
+          @current_result_path.pop
+        end
+
+        if new_invalid_null
+          nil
+        else
+          result_arr
+        end
+      end
+    end
+  end
+end

lib/graphql/execution/finalize.rb

@@ -5,7 +5,7 @@ module Execution
     class Interpreter
       # Wrapper for raw values
       class RawValue
-        include GraphQL::Execution::Next::Finalizer
+        include GraphQL::Execution::Finalizer
 
         def finalize_graphql_result(query, result_data, result_key)
           result_data[result_key] = @object

lib/graphql/execution/interpreter/handles_raw_value.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+module GraphQL
+  module Execution
+    class LoadArgumentStep
+      def initialize(field_resolve_step:, arguments:, load_receiver:, argument_value:, argument_definition:, argument_key:)
+        @field_resolve_step = field_resolve_step
+        @load_receiver = load_receiver
+        @arguments = arguments
+        @argument_value = argument_value
+        @argument_definition = argument_definition
+        @argument_key = argument_key
+        @loaded_value = nil
+      end
+
+      def value
+        @loaded_value = @field_resolve_step.sync(@loaded_value)
+        assign_value
+      end
+
+      def call
+        context = @field_resolve_step.selections_step.query.context
+        @loaded_value = begin
+          @load_receiver.load_and_authorize_application_object(@argument_definition, @argument_value, context)
+        rescue GraphQL::UnauthorizedError => auth_err
+          context.schema.unauthorized_object(auth_err)
+        end
+        if (runner = @field_resolve_step.runner).resolves_lazies && runner.lazy?(@loaded_value)
+          runner.dataloader.lazy_at_depth(@field_resolve_step.path.size, self)
+        else
+          assign_value
+        end
+      rescue GraphQL::RuntimeError => err
+        @loaded_value = err
+        assign_value
+      rescue StandardError => stderr
+        @loaded_value = begin
+          context.query.handle_or_reraise(stderr)
+        rescue GraphQL::ExecutionError => ex_err
+          ex_err
+        end
+        assign_value
+      end
+
+      private
+
+      def assign_value
+        if @loaded_value.is_a?(GraphQL::RuntimeError)
+          @loaded_value.path = @field_resolve_step.path
+          @field_resolve_step.arguments = @loaded_value
+        else
+          query = @field_resolve_step.selections_step.query
+          query.current_trace.object_loaded(@argument_definition, @loaded_value, query.context)
+          @arguments[@argument_key] = @loaded_value
+        end
+
+        field_pending_steps = @field_resolve_step.pending_steps
+        field_pending_steps.delete(self)
+        if @field_resolve_step.arguments && field_pending_steps.size == 0 # rubocop:disable Development/ContextIsPassedCop
+          @field_resolve_step.runner.add_step(@field_resolve_step)
+        end
+      end
+    end
+  end
+end

lib/graphql/execution/load_argument_step.rb

@@ -1,12 +1,28 @@
 # frozen_string_literal: true
-require "graphql/execution/next/prepare_object_step"
-require "graphql/execution/next/field_resolve_step"
-require "graphql/execution/next/finalize"
-require "graphql/execution/next/load_argument_step"
-require "graphql/execution/next/runner"
-require "graphql/execution/next/selections_step"
+require "graphql/execution/prepare_object_step"
+require "graphql/execution/field_resolve_step"
+require "graphql/execution/finalize"
+require "graphql/execution/load_argument_step"
+require "graphql/execution/runner"
+require "graphql/execution/selections_step"
 module GraphQL
   module Execution
+    module Finalizer
+      attr_accessor :path
+      def finalize_graphql_result(query, result_data, result_key)
+        raise RequiredImplementationMissingError
+      end
+    end
+
+    module HaltExecution
+    end
+
+    module PostProcessor
+      def after_resolve(field_results)
+        raise RequiredImplementationMissingError, "#{self.class}#after_resolve should handle `field_results` and return a new value to use"
+      end
+    end
+
     module Next
       module SchemaExtension
         def execute_next(query_str = nil, context: nil, document: nil, operation_name: nil, variables: nil, root_value: nil, validate: true, visibility_profile: nil)
@@ -68,22 +84,6 @@ def self.run_all(schema, query_options, context: {}, max_complexity: schema.max_
         runner = Runner.new(multiplex, **schema.execution_next_options)
         runner.execute
       end
-
-      module Finalizer
-        attr_accessor :path
-        def finalize_graphql_result(query, result_data, result_key)
-          raise RequiredImplementationMissingError
-        end
-      end
-
-      module HaltExecution
-      end
-
-      module PostProcessor
-        def after_resolve(field_results)
-          raise RequiredImplementationMissingError, "#{self.class}#after_resolve should handle `field_results` and return a new value to use"
-        end
-      end
     end
   end
 end