aerospike 2.23.0 → 2.25.0

data/lib/aerospike/command/command.rb

@@ -15,18 +15,17 @@
 # License for the specific language governing permissions and limitations under
 # the License.
 
-require 'time'
-require 'zlib'
+require "time"
+require "zlib"
 
-require 'msgpack'
-require 'aerospike/result_code'
-require 'aerospike/command/field_type'
+require "msgpack"
+require "aerospike/result_code"
+require "aerospike/command/field_type"
 
-require 'aerospike/policy/consistency_level'
-require 'aerospike/policy/commit_level'
+require "aerospike/policy/consistency_level"
+require "aerospike/policy/commit_level"
 
 module Aerospike
-
   private
 
   # Flags commented out are not supported by cmd client.
@@ -34,7 +33,8 @@ module Aerospike
   INFO1_READ = Integer(1 << 0)
   # Get all bins.
   INFO1_GET_ALL = Integer(1 << 1)
-
+  # Short query
+  INFO1_SHORT_QUERY = Integer(1 << 2)
 
   INFO1_BATCH = Integer(1 << 3)
   # Do not read the bins
@@ -76,19 +76,18 @@ module Aerospike
   # Completely replace existing record only.
   INFO3_REPLACE_ONLY = Integer(1 << 5)
 
-  MSG_TOTAL_HEADER_SIZE      = 30
-  FIELD_HEADER_SIZE          = 5
-  OPERATION_HEADER_SIZE      = 8
-  MSG_REMAINING_HEADER_SIZE  = 22
-  DIGEST_SIZE                = 20
-  COMPRESS_THRESHOLD         = 128
-  CL_MSG_VERSION             = 2
-  AS_MSG_TYPE                = 3
-  AS_MSG_TYPE_COMPRESSED     = 4
+  MSG_TOTAL_HEADER_SIZE = 30
+  FIELD_HEADER_SIZE = 5
+  OPERATION_HEADER_SIZE = 8
+  MSG_REMAINING_HEADER_SIZE = 22
+  DIGEST_SIZE = 20
+  COMPRESS_THRESHOLD = 128
+  CL_MSG_VERSION = 2
+  AS_MSG_TYPE = 3
+  AS_MSG_TYPE_COMPRESSED = 4
 
   class Command #:nodoc:
-
-    def initialize(node=nil)
+    def initialize(node = nil)
       @data_offset = 0
       @data_buffer = nil
 
@@ -116,6 +115,9 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       bins.each do |bin|
         estimate_operation_size_for_bin(bin)
       end
@@ -125,6 +127,7 @@ module Aerospike
       write_header_with_policy(policy, 0, INFO2_WRITE, field_count, bins.length)
       write_key(key, policy)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
 
       bins.each do |bin|
         write_operation_for_bin(bin, operation)
@@ -142,10 +145,14 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       size_buffer
-      write_header_with_policy(policy, 0, INFO2_WRITE|INFO2_DELETE, field_count, 0)
+      write_header_with_policy(policy, 0, INFO2_WRITE | INFO2_DELETE, field_count, 0)
       write_key(key)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
       end_cmd
     end
 
@@ -157,11 +164,15 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       estimate_operation_size
       size_buffer
       write_header_with_policy(policy, 0, INFO2_WRITE, field_count, 1)
       write_key(key)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
       write_operation_for_operation_type(Aerospike::Operation::TOUCH)
       end_cmd
     end
@@ -174,10 +185,14 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       size_buffer
-      write_header(policy, INFO1_READ|INFO1_NOBINDATA, 0, field_count, 0)
+      write_header(policy, INFO1_READ | INFO1_NOBINDATA, 0, field_count, 0)
       write_key(key)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
       end_cmd
     end
 
@@ -189,10 +204,14 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       size_buffer
-      write_header(policy, INFO1_READ|INFO1_GET_ALL, 0, field_count, 0)
+      write_header(policy, INFO1_READ | INFO1_GET_ALL, 0, field_count, 0)
       write_key(key)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
       end_cmd
     end
 
@@ -205,6 +224,8 @@ module Aerospike
         predexp_size = estimate_predexp(policy.predexp)
         field_count += 1 if predexp_size > 0
 
+        exp_size = estimate_expression_size(@policy.filter_exp)
+        field_count += 1 if exp_size > 0
 
         bin_names.each do |bin_name|
           estimate_operation_size_for_bin_name(bin_name)
@@ -214,6 +235,7 @@ module Aerospike
         write_header(policy, INFO1_READ, 0, field_count, bin_names.length)
         write_key(key)
         write_predexp(policy.predexp, predexp_size)
+        write_filter_exp(@policy.filter_exp, exp_size)
 
         bin_names.each do |bin_name|
           write_operation_for_bin_name(bin_name, Aerospike::Operation::READ)
@@ -233,7 +255,10 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
-      estimate_operation_size_for_bin_name('')
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
+      estimate_operation_size_for_bin_name("")
       size_buffer
 
       # The server does not currently return record header data with _INFO1_NOBINDATA attribute set.
@@ -244,70 +269,35 @@ module Aerospike
 
       write_key(key)
       write_predexp(policy.predexp, predexp_size)
-      write_operation_for_bin_name('', Aerospike::Operation::READ)
+      write_filter_exp(@policy.filter_exp, exp_size)
+      write_operation_for_bin_name("", Aerospike::Operation::READ)
       end_cmd
     end
 
     # Implements different command operations
-    def set_operate(policy, key, operations)
+    def set_operate(policy, key, args)
       begin_cmd
       field_count = estimate_key_size(key, policy)
 
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
-      read_attr = 0
-      write_attr = 0
-      read_header = false
-      record_bin_multiplicity = policy.record_bin_multiplicity == RecordBinMultiplicity::ARRAY
-
-      operations.each do |operation|
-        case operation.op_type
-        when Aerospike::Operation::READ, Aerospike::Operation::CDT_READ,
-              Aerospike::Operation::HLL_READ, Aerospike::Operation::BIT_READ
-          read_attr |= INFO1_READ
-
-          # Read all bins if no bin is specified.
-          read_attr |= INFO1_GET_ALL unless operation.bin_name
-
-        when Aerospike::Operation::READ_HEADER
-          # The server does not currently return record header data with _INFO1_NOBINDATA attribute set.
-          # The workaround is to request a non-existent bin.
-          # TODO: Fix this on server.
-          # read_attr |= _INFO1_READ | _INFO1_NOBINDATA
-          read_attr |= INFO1_READ
-          read_header = true
-
-        else
-          write_attr = INFO2_WRITE
-        end
+      exp_size = estimate_expression_size(policy.filter_exp)
+      field_count += 1 if exp_size > 0
 
-        if [Aerospike::Operation::HLL_MODIFY, Aerospike::Operation::HLL_READ,
-              Aerospike::Operation::BIT_MODIFY, Aerospike::Operation::BIT_READ].include?(operation.op_type)
-          record_bin_multiplicity = true
-        end
+      @data_offset += args.size
 
-        estimate_operation_size_for_operation(operation)
-      end
       size_buffer
 
-
-      write_attr |= INFO2_RESPOND_ALL_OPS if write_attr != 0 && record_bin_multiplicity
-
-      if write_attr == 0
-        write_header(policy, read_attr, write_attr, field_count, operations.length)
-      else
-        write_header_with_policy(policy, read_attr, write_attr, field_count, operations.length)
-      end
+      write_header_with_policy(policy, args.read_attr, args.write_attr, field_count, args.operations.length)
       write_key(key, policy)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(policy.filter_exp, exp_size)
 
-      operations.each do |operation|
+      args.operations.each do |operation|
         write_operation_for_operation(operation)
       end
 
-      write_operation_for_bin(nil, Aerospike::Operation::READ) if read_header
-
       end_cmd
       mark_compressed(policy)
     end
@@ -319,6 +309,9 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       arg_bytes = args.to_bytes
 
       field_count += estimate_udf_size(package_name, function_name, arg_bytes)
@@ -327,6 +320,7 @@ module Aerospike
       write_header(policy, 0, INFO2_WRITE, field_count, 0)
       write_key(key, policy)
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
       write_field_string(package_name, Aerospike::FieldType::UDF_PACKAGE_NAME)
       write_field_string(function_name, Aerospike::FieldType::UDF_FUNCTION)
       write_field_bytes(arg_bytes, Aerospike::FieldType::UDF_ARGLIST)
@@ -377,6 +371,9 @@ module Aerospike
       predexp_size = estimate_predexp(policy.predexp)
       field_count += 1 if predexp_size > 0
 
+      exp_size = estimate_expression_size(@policy.filter_exp)
+      field_count += 1 if exp_size > 0
+
       # Estimate scan options size.
       # @data_offset += 2 + FIELD_HEADER_SIZE
       # field_count += 1
@@ -418,7 +415,7 @@ module Aerospike
 
         node_partitions.parts_full.each do |part|
           @data_buffer.write_uint16_little_endian(part.id, @data_offset)
-          @data_offset+=2
+          @data_offset += 2
         end
       end
 
@@ -427,7 +424,7 @@ module Aerospike
 
         node_partitions.parts_partial.each do |part|
           @data_buffer.write_binary(part.digest, @data_offset)
-          @data_offset+=part.digest.length
+          @data_offset += part.digest.length
         end
       end
 
@@ -440,6 +437,7 @@ module Aerospike
       end
 
       write_predexp(policy.predexp, predexp_size)
+      write_filter_exp(@policy.filter_exp, exp_size)
 
       # write_field_header(2, Aerospike::FieldType::SCAN_OPTIONS)
 
@@ -454,7 +452,7 @@ module Aerospike
       # @data_buffer.write_byte(policy.scan_percent.to_i.ord, @data_offset)
       # @data_offset += 1
 
-      write_field_header(4, Aerospike::FieldType::SCAN_TIMEOUT)
+      write_field_header(4, Aerospike::FieldType::SOCKET_TIMEOUT)
       @data_buffer.write_uint32(policy.socket_timeout.to_i, @data_offset)
       @data_offset += 4
 
@@ -477,7 +475,7 @@ module Aerospike
       while true
         # too many retries
         iterations += 1
-        break if (@policy.max_retries > 0) && (iterations > @policy.max_retries+1)
+        break if (@policy.max_retries > 0) && (iterations > @policy.max_retries + 1)
 
         # Sleep before trying again, after the first iteration
         sleep(@policy.sleep_between_retries) if iterations > 1 && @policy.sleep_between_retries > 0
@@ -565,7 +563,6 @@ module Aerospike
         ensure
           Buffer.put(@data_buffer)
         end
-
       end # while
 
       # execution timeout
@@ -574,7 +571,6 @@ module Aerospike
 
     protected
 
-
     def estimate_key_size(key, policy = nil)
       field_count = 0
 
@@ -644,6 +640,15 @@ module Aerospike
       return 0
     end
 
+    def estimate_expression_size(exp)
+      unless exp.nil?
+        @data_offset += FIELD_HEADER_SIZE
+        @data_offset += exp.size
+        return exp.size
+      end
+      0
+    end
+
     # Generic header write.
     def write_header(policy, read_attr, write_attr, field_count, operation_count)
       read_attr |= INFO1_CONSISTENCY_ALL if policy.consistency_level == Aerospike::ConsistencyLevel::CONSISTENCY_ALL
@@ -715,7 +720,6 @@ module Aerospike
       @data_buffer.write_byte(0, 24)
       @data_buffer.write_byte(0, 25)
 
-
       @data_buffer.write_int16(field_count, 26)
       @data_buffer.write_int16(operation_count, 28)
 
@@ -737,15 +741,14 @@ module Aerospike
       if policy && policy.respond_to?(:send_key) && policy.send_key == true
         write_field_value(key.user_key_as_value, Aerospike::FieldType::KEY)
       end
-
     end
 
     def write_operation_for_bin(bin, operation)
-      name_length = @data_buffer.write_binary(bin.name, @data_offset+OPERATION_HEADER_SIZE)
-      value_length = bin.value_object.write(@data_buffer, @data_offset+OPERATION_HEADER_SIZE+name_length)
+      name_length = @data_buffer.write_binary(bin.name, @data_offset + OPERATION_HEADER_SIZE)
+      value_length = bin.value_object.write(@data_buffer, @data_offset + OPERATION_HEADER_SIZE + name_length)
 
       # Buffer.Int32ToBytes(name_length+value_length+4, @data_buffer, @data_offset)
-      @data_buffer.write_int32(name_length+value_length+4, @data_offset)
+      @data_buffer.write_int32(name_length + value_length + 4, @data_offset)
 
       @data_offset += 4
       @data_buffer.write_byte(operation, @data_offset)
@@ -762,13 +765,13 @@ module Aerospike
     def write_operation_for_operation(operation)
       name_length = 0
       if operation.bin_name
-        name_length = @data_buffer.write_binary(operation.bin_name, @data_offset+OPERATION_HEADER_SIZE)
+        name_length = @data_buffer.write_binary(operation.bin_name, @data_offset + OPERATION_HEADER_SIZE)
       end
 
-      value_length = operation.bin_value.write(@data_buffer, @data_offset+OPERATION_HEADER_SIZE+name_length)
+      value_length = operation.bin_value.write(@data_buffer, @data_offset + OPERATION_HEADER_SIZE + name_length)
 
       # Buffer.Int32ToBytes(name_length+value_length+4, @data_buffer, @data_offset)
-      @data_buffer.write_int32(name_length+value_length+4, @data_offset)
+      @data_buffer.write_int32(name_length + value_length + 4, @data_offset)
 
       @data_offset += 4
       @data_buffer.write_byte(operation.op_type, @data_offset)
@@ -783,9 +786,9 @@ module Aerospike
     end
 
     def write_operation_for_bin_name(name, operation)
-      name_length = @data_buffer.write_binary(name, @data_offset+OPERATION_HEADER_SIZE)
+      name_length = @data_buffer.write_binary(name, @data_offset + OPERATION_HEADER_SIZE)
       # Buffer.Int32ToBytes(name_length+4, @data_buffer, @data_offset)
-      @data_buffer.write_int32(name_length+4, @data_offset)
+      @data_buffer.write_int32(name_length + 4, @data_offset)
 
       @data_offset += 4
       @data_buffer.write_byte(operation, @data_offset)
@@ -824,37 +827,37 @@ module Aerospike
     end
 
     def write_field_string(str, ftype)
-      len = @data_buffer.write_binary(str, @data_offset+FIELD_HEADER_SIZE)
+      len = @data_buffer.write_binary(str, @data_offset + FIELD_HEADER_SIZE)
       write_field_header(len, ftype)
       @data_offset += len
     end
 
     def write_u16_little_endian(i, ftype)
-      @data_buffer.write_uint16_little_endian(i, @data_offset+FIELD_HEADER_SIZE)
+      @data_buffer.write_uint16_little_endian(i, @data_offset + FIELD_HEADER_SIZE)
       write_field_header(2, ftype)
       @data_offset += 2
     end
 
     def write_field_int(i, ftype)
-      @data_buffer.write_int32(i, @data_offset+FIELD_HEADER_SIZE)
+      @data_buffer.write_int32(i, @data_offset + FIELD_HEADER_SIZE)
       write_field_header(4, ftype)
       @data_offset += 4
     end
 
     def write_field_int64(i, ftype)
-      @data_buffer.write_int64(i, @data_offset+FIELD_HEADER_SIZE)
+      @data_buffer.write_int64(i, @data_offset + FIELD_HEADER_SIZE)
       write_field_header(8, ftype)
       @data_offset += 8
     end
 
     def write_field_bytes(bytes, ftype)
-      @data_buffer.write_binary(bytes, @data_offset+FIELD_HEADER_SIZE)
+      @data_buffer.write_binary(bytes, @data_offset + FIELD_HEADER_SIZE)
       write_field_header(bytes.bytesize, ftype)
       @data_offset += bytes.bytesize
     end
 
     def write_field_header(size, ftype)
-      @data_buffer.write_int32(size+1, @data_offset)
+      @data_buffer.write_int32(size + 1, @data_offset)
       @data_offset += 4
       @data_buffer.write_byte(ftype, @data_offset)
       @data_offset += 1
@@ -862,13 +865,20 @@ module Aerospike
 
     def write_predexp(predexp, predexp_size)
       if predexp && predexp.size > 0
-        write_field_header(predexp_size, Aerospike::FieldType::PREDEXP)
+        write_field_header(predexp_size, Aerospike::FieldType::FILTER_EXP)
         @data_offset = Aerospike::PredExp.write(
           predexp, @data_buffer, @data_offset
         )
       end
     end
 
+    def write_filter_exp(exp, exp_size)
+      unless exp.nil?
+        write_field_header(exp_size, Aerospike::FieldType::FILTER_EXP)
+        @data_offset += exp.write(@data_buffer, @data_offset)
+      end
+    end
+
     def begin_cmd
       @data_offset = MSG_TOTAL_HEADER_SIZE
     end
@@ -882,7 +892,7 @@ module Aerospike
     end
 
     def end_cmd
-      size = (@data_offset-8) | Integer(CL_MSG_VERSION << 56) | Integer(AS_MSG_TYPE << 48)
+      size = (@data_offset - 8) | Integer(CL_MSG_VERSION << 56) | Integer(AS_MSG_TYPE << 48)
       @data_buffer.write_int64(size, 0)
     end
 
@@ -896,13 +906,13 @@ module Aerospike
 
         # write original size as header
         proto_s = "%08d" % 0
-        proto_s[0, 8] = [@data_offset].pack('q>')
+        proto_s[0, 8] = [@data_offset].pack("q>")
         compressed.prepend(proto_s)
 
         # write proto
-        proto = (compressed.size+8) | Integer(CL_MSG_VERSION << 56) | Integer(AS_MSG_TYPE << 48)
+        proto = (compressed.size + 8) | Integer(CL_MSG_VERSION << 56) | Integer(AS_MSG_TYPE << 48)
         proto_s = "%08d" % 0
-        proto_s[0, 8] = [proto].pack('q>')
+        proto_s[0, 8] = [proto].pack("q>")
         compressed.prepend(proto_s)
 
         @data_buffer = Buffer.new(-1, compressed)
@@ -927,7 +937,5 @@ module Aerospike
     def mark_compressed(policy)
       @compress = policy.use_compression
     end
-
   end # class
-
 end # module

data/lib/aerospike/command/field_type.rb

@@ -21,34 +21,31 @@ module Aerospike
 
   module FieldType
 
-    NAMESPACE = 0
-    TABLE = 1
-    KEY = 2
-    #BIN = 3
-    DIGEST_RIPE = 4
-    #GU_TID = 5
-    DIGEST_RIPE_ARRAY = 6
-    TRAN_ID = 7    # user supplied transaction id, which is simply passed back
-    SCAN_OPTIONS = 8
-    SCAN_TIMEOUT = 9
-    RECORDS_PER_SECOND = 10
-    PID_ARRAY = 11
-    DIGEST_ARRAY = 12
-    MAX_RECORDS = 13
-    BVAL_ARRAY = 15
-    INDEX_NAME = 21
-    INDEX_RANGE = 22
-    INDEX_FILTER = 23
-    INDEX_LIMIT = 24
-    INDEX_ORDER_BY = 25
-    INDEX_TYPE = 26
-    UDF_PACKAGE_NAME = 30
-    UDF_FUNCTION = 31
-    UDF_ARGLIST = 32
-    UDF_OP = 33
-    QUERY_BINLIST = 40
-    BATCH_INDEX = 41
-    PREDEXP = 43
+    NAMESPACE  = 0
+    TABLE      = 1
+    KEY        = 2
+    DIGEST_RIPE  = 4
+    DIGEST_RIPE_ARRAY     = 6
+    TRAN_ID               = 7 # user supplied transaction id, which is simply passed back
+    SCAN_OPTIONS          = 8
+    SOCKET_TIMEOUT        = 9
+    RECORDS_PER_SECOND    = 10
+    PID_ARRAY             = 11
+    DIGEST_ARRAY          = 12
+    MAX_RECORDS           = 13
+    BVAL_ARRAY            = 15
+    INDEX_NAME            = 21
+    INDEX_RANGE           = 22
+    INDEX_CONTEXT         = 23
+    INDEX_TYPE            = 26
+    UDF_PACKAGE_NAME      = 30
+    UDF_FUNCTION          = 31
+    UDF_ARGLIST           = 32
+    UDF_OP                = 33
+    QUERY_BINLIST         = 40
+    BATCH_INDEX           = 41
+    BATCH_INDEX_WITH_SET  = 42
+    FILTER_EXP            = 43
 
   end # module
 

data/lib/aerospike/command/operate_args.rb

@@ -0,0 +1,99 @@
+# encoding: utf-8
+# Copyright 2016-2020 Aerospike, Inc.
+#
+# Portions may be licensed to Aerospike, Inc. under one or more contributor
+# license agreements.
+#
+# Licensed under the Apache License, Version 2.0 (the "License") you may not
+# use this file except in compliance with the License. You may obtain a copy of
+# the License at http:#www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations under
+# the License.
+
+require "aerospike/operation"
+
+module Aerospike
+  private
+
+  class OperateArgs
+    attr_reader :write_policy, :operations, :partition
+    attr_reader :size, :read_attr, :write_attr, :has_write
+
+    RESPOND_ALL_OPS_READ_CMDS = [Operation::BIT_READ, Operation::EXP_READ, Operation::HLL_READ, Operation::CDT_READ]
+    READ_CMDS = [Operation::BIT_READ, Operation::EXP_READ, Operation::HLL_READ, Operation::CDT_READ, Operation::CDT_READ, Operation::READ]
+    MODIFY_CMDS = [Operation::BIT_MODIFY, Operation::EXP_MODIFY, Operation::HLL_MODIFY, Operation::CDT_MODIFY]
+
+    def initialize(cluster, policy, write_default, read_default, key, operations)
+      @operations = operations
+
+      data_offset = 0
+      rattr = 0
+      wattr = 0
+      write = false
+      read_bin = false
+      read_header = false
+      respond_all_ops = false
+
+      @operations.each do |operation|
+        if READ_CMDS.include?(operation.op_type)
+          if RESPOND_ALL_OPS_READ_CMDS.include?(operation.op_type)
+            # Map @operations require respond_all_ops to be true.
+            respond_all_ops = true
+          end
+
+          rattr |= Aerospike::INFO1_READ
+
+          # Read all bins if no bin is specified.
+          rattr |= Aerospike::INFO1_GET_ALL if operation.bin_name.nil?
+          read_bin = true
+        elsif operation.op_type == Operation::READ_HEADER
+          rattr |= Aerospike::INFO1_READ
+          read_header = true
+        elsif MODIFY_CMDS.include?(operation.op_type)
+          # Map @operations require respond_all_ops to be true.
+          respond_all_ops = true
+
+          wattr = Aerospike::INFO2_WRITE
+          write = true
+        else
+          wattr = Aerospike::INFO2_WRITE
+          write = true
+        end
+        data_offset += operation.bin_name.bytesize + Aerospike::OPERATION_HEADER_SIZE unless operation.bin_name.nil?
+        data_offset += operation.bin_value.estimate_size
+      end
+
+      @size = data_offset
+      @has_write = write
+
+      if read_header && !read_bin
+        rattr |= Aerospike::INFO1_NOBINDATA
+      end
+      @read_attr = rattr
+
+      if policy.nil?
+        @write_policy = write ? write_default : read_default
+      else
+        @write_policy = policy
+      end
+
+      # When GET_ALL is specified, RESPOND_ALL_OPS must be disabled.
+      if (respond_all_ops && policy.record_bin_multiplicity) && (rattr & Aerospike::INFO1_GET_ALL) == 0
+        wattr |= Aerospike::INFO2_RESPOND_ALL_OPS
+      end
+      @write_attr = wattr
+
+      if write
+        # @partition = Partition.write(cluster, @write_policy, key)
+        @partition = Partition.new_by_key(key)
+      else
+        # @partition = Partition.read(cluster, @write_policy, key)
+        @partition = Partition.new_by_key(key)
+      end
+    end
+  end
+end