@@ -21,13 +21,12 @@ module Aerospike
2121
2222 class BatchOperateCommand < MultiCommand #:nodoc:
2323
24- attr_accessor :batch , :policy , :attr , :records
24+ attr_accessor :batch , :policy , :attr
2525
26- def initialize ( node , batch , policy , records )
26+ def initialize ( node , batch , policy )
2727 super ( node )
2828 @batch = batch
2929 @policy = policy
30- @records = records
3130 end
3231
3332 def batch_flags
@@ -50,7 +49,7 @@ def write_buffer
5049 @data_offset += FIELD_HEADER_SIZE + 4 + 1 # batch.keys.length + flags
5150
5251 prev = nil
53- @ records. each do |record |
52+ batch . records . each do |record |
5453 key = record . key
5554 @data_offset += key . digest . length + 4 # 4 byte batch offset
5655
@@ -78,7 +77,7 @@ def write_buffer
7877
7978 prev = nil
8079 attr = BatchAttr . new
81- batch . records . each_with_index do |record , index |
80+ batch . each_record_with_index do |record , index |
8281 @data_offset += @data_buffer . write_int32 ( index , @data_offset )
8382 key = record . key
8483 @data_offset += @data_buffer . write_binary ( key . digest , @data_offset )
@@ -136,13 +135,13 @@ def parse_row(result_code)
136135 op_count = @data_buffer . read_int16 ( 20 )
137136
138137 skip_key ( field_count )
139- req_key = records [ batch_index ] . key
138+ req_key = batch . record_for_index ( batch_index ) . key
140139
141- records [ batch_index ] . result_code = result_code
140+ batch . record_for_index ( batch_index ) . result_code = result_code
142141 case result_code
143142 when 0 , ResultCode ::UDF_BAD_RESPONSE
144143 record = parse_record ( req_key , op_count , generation , expiration )
145- records [ batch_index ] . record = record
144+ batch . record_for_index ( batch_index ) . record = record
146145 end
147146 end
148147
0 commit comments