lib/aerospike/query/scan_partition_command.rb in aerospike-2.29.0 vs lib/aerospike/query/scan_partition_command.rb in aerospike-3.0.0

@@ -21,23 +21,23 @@
 
   private
 
   class ScanPartitionCommand < StreamCommand #:nodoc:
 
-    def initialize(policy, tracker, node_partitions, namespace, set_name, bin_names, recordset)
+    def initialize(cluster, policy, tracker, node_partitions, namespace, set_name, bin_names, recordset)
       super(node_partitions.node)
-
+      @cluster = cluster
       @policy = policy
       @namespace = namespace
       @set_name = set_name
       @bin_names = bin_names
       @recordset = recordset
       @node_partitions = node_partitions
       @tracker = tracker
     end
 
     def write_buffer
-      set_scan(@policy, @namespace, @set_name, @bin_names, @node_partitions)
+      set_scan(@cluster, @policy, @namespace, @set_name, @bin_names, @node_partitions)
     end
 
     def should_retry(e)
       # !! converts nil to false
       !!@tracker&.should_retry(@node_partitions, e)