lib/graphql/batch.rb in graphql-batch-0.3.5 vs lib/graphql/batch.rb in graphql-batch-0.3.7

@@ -11,23 +11,23 @@
 
     def self.batch
       raise NestedError if GraphQL::Batch::Executor.current
       begin
         GraphQL::Batch::Executor.current = GraphQL::Batch::Executor.new
-        Promise.sync(yield)
+        ::Promise.sync(yield)
       ensure
         GraphQL::Batch::Executor.current = nil
       end
     end
 
-    def self.use(schema_defn)
+    def self.use(schema_defn, executor_class: GraphQL::Batch::Executor)
       schema = schema_defn.target
       if GraphQL::VERSION >= "1.6.0"
-        instrumentation = GraphQL::Batch::SetupMultiplex.new(schema)
+        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)
+        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