lib/graphql/dataloader.rb in graphql-2.0.24 vs lib/graphql/dataloader.rb in graphql-2.0.25

@@ -109,12 +109,12 @@
       prev_queue = @pending_jobs
       prev_pending_keys = {}
       @source_cache.each do |source_class, batched_sources|
         batched_sources.each do |batch_args, batched_source_instance|
           if batched_source_instance.pending?
-            prev_pending_keys[batched_source_instance] = batched_source_instance.pending_keys.dup
-            batched_source_instance.pending_keys.clear
+            prev_pending_keys[batched_source_instance] = batched_source_instance.pending.dup
+            batched_source_instance.pending.clear
           end
         end
       end
 
       @pending_jobs = []
@@ -125,11 +125,11 @@
       }
       run
       res
     ensure
       @pending_jobs = prev_queue
-      prev_pending_keys.each do |source_instance, pending_keys|
-        source_instance.pending_keys.concat(pending_keys)
+      prev_pending_keys.each do |source_instance, pending|
+        source_instance.pending.merge!(pending)
       end
     end
 
     # @api private Move along, move along
     def run