lib/graphql/batch.rb in graphql-batch-0.4.3 vs lib/graphql/batch.rb in graphql-batch-0.5.0

@@ -14,38 +14,25 @@
         GraphQL::Batch::Executor.end_batch
       end
     end
 
     def self.use(schema_defn, executor_class: GraphQL::Batch::Executor)
-      # Support 1.10+ which passes the class instead of the definition proxy
-      schema = schema_defn.is_a?(Class) ? schema_defn : schema_defn.target
-      current_gem_version = Gem::Version.new(GraphQL::VERSION)
-      if current_gem_version >= Gem::Version.new("1.6.0")
-        instrumentation = GraphQL::Batch::SetupMultiplex.new(schema, executor_class: executor_class)
-        schema_defn.instrument(:multiplex, instrumentation)
-        if schema.mutation
-          if current_gem_version >= Gem::Version.new('1.9.0.pre3') &&
-              (schema.mutation.is_a?(Class) || schema.mutation.metadata[:type_class])
-            require_relative "batch/mutation_field_extension"
-            schema.mutation.fields.each do |name, f|
-              field = f.respond_to?(:type_class) ? f.type_class : f.metadata[:type_class]
-              field.extension(GraphQL::Batch::MutationFieldExtension)
-            end
-          else
-            schema_defn.instrument(:field, instrumentation)
-          end
+      instrumentation = GraphQL::Batch::SetupMultiplex.new(schema_defn, executor_class: executor_class)
+      schema_defn.instrument(:multiplex, instrumentation)
+
+      if schema_defn.mutation
+        require_relative "batch/mutation_field_extension"
+
+        schema_defn.mutation.fields.each do |name, field|
+          field.extension(GraphQL::Batch::MutationFieldExtension)
         end
-      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
   end
 end
 
 require_relative "batch/version"
 require_relative "batch/loader"
 require_relative "batch/executor"
-require_relative "batch/setup"
 require_relative "batch/setup_multiplex"