batch.rb in graphql-batch-0.4.0

- old
+ new

@@ -1,9 +1,6 @@
 require "graphql"
-if Gem::Version.new(GraphQL::VERSION) < Gem::Version.new("1.3")
-  warn "graphql gem versions less than 1.3 are deprecated for use with graphql-batch, upgrade so lazy_resolve can be used"
-end
 require "promise.rb"
 
 module GraphQL
   module Batch
     BrokenPromiseError = ::Promise::BrokenError
@@ -18,28 +15,34 @@
       end
     end
 
     def self.use(schema_defn, executor_class: GraphQL::Batch::Executor)
       schema = schema_defn.target
-      if GraphQL::VERSION >= "1.6.0"
+      if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('1.9.0.pre3')
+        require_relative "batch/mutation_field_extension"
+        if schema.mutation
+          schema.mutation.fields.each do |name, f|
+            field = f.metadata[:type_class]
+            field.extension(GraphQL::Batch::MutationFieldExtension)
+          end
+        end
         instrumentation = GraphQL::Batch::SetupMultiplex.new(schema, executor_class: executor_class)
         schema_defn.instrument(:multiplex, instrumentation)
+      elsif GraphQL::VERSION >= "1.6.0"
+        instrumentation = GraphQL::Batch::SetupMultiplex.new(schema, executor_class: executor_class)
+        schema_defn.instrument(:multiplex, instrumentation)
         schema_defn.instrument(:field, instrumentation)
       else
         instrumentation = GraphQL::Batch::Setup.new(schema, executor_class: executor_class)
         schema_defn.instrument(:query, instrumentation)
         schema_defn.instrument(:field, instrumentation)
       end
       schema_defn.lazy_resolve(::Promise, :sync)
     end
-
-    autoload :ExecutionStrategy, 'graphql/batch/execution_strategy'
-    autoload :MutationExecutionStrategy, 'graphql/batch/mutation_execution_strategy'
   end
 end
 
 require_relative "batch/version"
 require_relative "batch/loader"
 require_relative "batch/executor"
-require_relative "batch/promise"
 require_relative "batch/setup"
 require_relative "batch/setup_multiplex"