Contents

module GraphQL::Batch
  class ExecutionStrategy < GraphQL::Query::SerialExecution
    def execute(_, _, _)
      as_promise(super).sync
    ensure
      GraphQL::Batch::Executor.current.clear
    end

    private

    def as_promise(result)
      GraphQL::Batch::Promise.resolve(as_promise_unless_resolved(result))
    end

    def as_promise_unless_resolved(result)
      all_promises = []
      each_promise(result) do |obj, key, promise|
        obj[key] = nil
        all_promises << promise.then do |value|
          obj[key] = value
          as_promise_unless_resolved(value)
        end
      end
      return result if all_promises.empty?
      Promise.all(all_promises).then { result }
    end

    def each_promise(obj, &block)
      case obj
      when Array
        obj.each_with_index do |value, idx|
          each_promise_in_entry(obj, idx, value, &block)
        end
      when Hash
        obj.each do |key, value|
          each_promise_in_entry(obj, key, value, &block)
        end
      end
    end

    def each_promise_in_entry(obj, key, value, &block)
      if value.is_a?(::Promise)
        yield obj, key, value
      else
        each_promise(value, &block)
      end
    end

    class FieldResolution < GraphQL::Query::SerialExecution::FieldResolution
      def get_finished_value(raw_value)
        if raw_value.is_a?(::Promise)
          raw_value.then(->(result) { super(result) }, lambda do |error|
            error.is_a?(GraphQL::ExecutionError) ? super(error) : raise(error)
          end)
        else
          super
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
graphql-batch-0.2.4	lib/graphql/batch/execution_strategy.rb
graphql-batch-0.2.3	lib/graphql/batch/execution_strategy.rb
graphql-batch-0.2.2	lib/graphql/batch/execution_strategy.rb
graphql-batch-0.2.1	lib/graphql/batch/execution_strategy.rb
graphql-batch-0.2.0	lib/graphql/batch/execution_strategy.rb