aerospike 2.13.0 → 2.18.0

data/lib/aerospike/command/multi_command.rb

@@ -29,6 +29,9 @@ module Aerospike
       @valid = true
       @mutex = Mutex.new
 
+      @compressed_data_buffer = nil
+      @compressed_data_offset = nil
+
       self
     end
 
@@ -42,12 +45,41 @@ module Aerospike
       status = true
 
       while status
+        @data_offset = 0
+        @compressed_data_buffer = nil
+
         # Read header.
         read_bytes(8)
 
         size = @data_buffer.read_int64(0)
         receive_size = size & 0xFFFFFFFFFFFF
 
+        # inflate if compressed
+        compressed_sz = compressed_size
+        if compressed_sz
+          begin
+            # read compressed msg header
+            @conn.read(@data_buffer, 8)
+
+            # read compressed message
+            @conn.read(@data_buffer, compressed_sz - 8)
+
+            # inflate the results
+            # TODO: reuse the current buffer
+            uncompressed = Zlib::inflate(@data_buffer.buf)
+            receive_size = uncompressed.size - 8
+
+            @compressed_data_buffer = Buffer.new(-1, uncompressed)
+            @compressed_data_offset = 0
+
+            # waste the first 8 header bytes
+            @compressed_data_buffer.eat!(8)
+          rescue => e
+            Aerospike.logger.error("parse result error: #{e}")
+            raise e
+          end
+        end
+
         if receive_size > 0
           status = parse_group(receive_size)
         else
@@ -157,7 +189,13 @@ module Aerospike
         @data_buffer = Buffer.new(length)
       end
 
-      @conn.read(@data_buffer, length)
+      if compressed?
+        @data_buffer.write_binary(@compressed_data_buffer.buf[@compressed_data_offset...@compressed_data_offset+length], 0)
+        @compressed_data_offset += length
+      else
+        @conn.read(@data_buffer, length)
+      end
+
       @data_offset += length
     end
 
@@ -176,6 +214,11 @@ module Aerospike
       res
     end
 
+    def compressed?
+      @compressed_data_buffer != nil
+    end
+
+
   end # class
 
 end # module

data/lib/aerospike/command/read_command.rb

@@ -15,6 +15,8 @@
 # License for the specific language governing permissions and limitations under
 # the License.
 
+require 'zlib'
+
 require 'aerospike/record'
 
 require 'aerospike/command/single_command'
@@ -50,12 +52,40 @@ module Aerospike
     def parse_result
       # Read header.
       begin
-        @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+        @conn.read(@data_buffer, 8)
       rescue => e
         Aerospike.logger.error("parse result error: #{e}")
         raise e
       end
 
+      # inflate if compressed
+      compressed_sz = compressed_size
+      if compressed_sz
+        begin
+          # waste 8 size bytes
+          @conn.read(@data_buffer, 8)
+
+          # read compressed message
+          @conn.read(@data_buffer, compressed_sz - 8)
+
+          # inflate the results
+          # TODO: reuse the current buffer
+          uncompressed = Zlib::inflate(@data_buffer.buf)
+
+          @data_buffer = Buffer.new(-1, uncompressed)
+        rescue => e
+          Aerospike.logger.error("parse result error: #{e}")
+          raise e
+        end
+      else
+        begin
+          bytes_read = @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE - 8, 8)
+        rescue => e
+          Aerospike.logger.error("parse result error: #{e}")
+          raise e
+        end
+      end
+
       # A number of these are commented out because we just don't care enough to read
       # that section of the header. If we do care, uncomment and check!
       sz = @data_buffer.read_int64(0)
@@ -68,7 +98,9 @@ module Aerospike
       receive_size = (sz & 0xFFFFFFFFFFFF) - header_length
 
       # Read remaining message bytes.
-      if receive_size > 0
+      if compressed_sz 
+        @data_buffer.eat!(MSG_TOTAL_HEADER_SIZE)
+      elsif receive_size > 0
         size_buffer_sz(receive_size)
 
         begin

data/lib/aerospike/command/touch_command.rb

@@ -40,7 +40,40 @@ module Aerospike
 
     def parse_result
       # Read header.
-      @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+      begin
+        @conn.read(@data_buffer, 8)
+      rescue => e
+        Aerospike.logger.error("parse result error: #{e}")
+        raise e
+      end
+
+      # inflate if compressed
+      compressed_sz = compressed_size
+      if compressed_sz
+        begin
+          #waste 8 size bytes
+          @conn.read(@data_buffer, 8)
+
+          # read compressed message
+          @conn.read(@data_buffer, sz - 8)
+
+          # inflate the results
+          # TODO: reuse the current buffer
+          uncompressed = Zlib::inflate(@data_buffer.buf)
+
+          @data_buffer = Buffer.new(-1, uncompressed)
+        rescue => e
+          Aerospike.logger.error("parse result error: #{e}")
+          raise e
+        end
+      else
+        begin
+          bytes_read = @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE - 8, 8)
+        rescue => e
+          Aerospike.logger.error("parse result error: #{e}")
+          raise e
+        end
+      end
 
       result_code = @data_buffer.read(13).ord & 0xFF
 

data/lib/aerospike/features.rb

@@ -40,5 +40,10 @@ module Aerospike
     # Server supports the new "peers" protocol for automatic node discovery
     PEERS = :peers
 
+    # Server supports the "truncate-namespace" command
+    TRUNCATE_NAMESPACE = :"truncate-namespace"
+
+    # Server supports the "blob-bits" command
+    BLOB_BITS = :"blob-bits"
   end
 end

data/lib/aerospike/node.rb

@@ -22,7 +22,7 @@ require 'aerospike/atomic/atomic'
 module Aerospike
   class Node
 
-    attr_reader :reference_count, :responded, :name, :features, :cluster_name, :partition_generation, :peers_generation, :failures, :cluster, :peers_count, :host
+    attr_reader :reference_count, :responded, :name, :features, :cluster_name, :partition_generation, :rebalance_generation, :peers_generation, :failures, :cluster, :peers_count, :host
 
     PARTITIONS = 4096
     FULL_HEALTH = 100
@@ -46,16 +46,29 @@ module Aerospike
       @peers_count = Atomic.new(0)
       @peers_generation = ::Aerospike::Node::Generation.new
       @partition_generation = ::Aerospike::Node::Generation.new
+      @rebalance_generation = ::Aerospike::Node::Rebalance.new
       @reference_count = Atomic.new(0)
       @responded = Atomic.new(false)
       @active = Atomic.new(true)
       @failures = Atomic.new(0)
 
       @replica_index = Atomic.new(0)
+      @racks = Atomic.new(nil)
 
       @connections = ::Aerospike::ConnectionPool.new(cluster, host)
     end
 
+    def update_racks(parser)
+      new_racks = parser.update_racks
+      @racks.value = new_racks if new_racks
+    end
+
+    def has_rack(ns, rack_id)
+      racks = @racks.value
+      return false if !racks
+      racks[ns] == rack_id
+    end
+
     # Get a connection to the node. If no cached connection is not available,
     # a new connection will be created
     def get_connection(timeout)
@@ -199,6 +212,10 @@ module Aerospike
       Node::Refresh::Partitions.(self, peers)
     end
 
+    def refresh_racks()
+      Node::Refresh::Racks.(self)
+    end
+
     def refresh_peers(peers)
       Node::Refresh::Peers.(self, peers)
     end

data/lib/aerospike/node/rebalance.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+# Copyright 2018 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.
+
+module Aerospike
+  class Node
+    # generic class for representing changes in eg. peer and partition generation
+    class Rebalance
+      attr_reader :generation
+
+      def initialize(generation = -1)
+        @generation = ::Aerospike::Atomic.new(generation)
+        @changed = ::Aerospike::Atomic.new(false)
+      end
+
+      def changed?
+        @changed.value == true
+      end
+
+      def eql?(generation)
+        @generation.value == generation
+      end
+
+      def reset_changed!
+        @changed.value = false
+      end
+
+      def update(new_generation)
+        return if @generation.value == new_generation
+        @generation.value = new_generation
+        @changed.value = true
+      end
+    end
+  end
+end

data/lib/aerospike/node/refresh/info.rb

@@ -24,14 +24,17 @@ module Aerospike
         CMDS_BASE = %w[node partition-generation cluster-name].freeze
         CMDS_PEERS = (CMDS_BASE + ['peers-generation']).freeze
         CMDS_SERVICES = (CMDS_BASE + ['services']).freeze
+        CMDS_REBALANCE = (CMDS_PEERS + ['rebalance-generation']).freeze
 
         class << self
           def call(node, peers)
             conn = node.tend_connection
             if peers.use_peers?
-              info_map = ::Aerospike::Info.request(conn, *CMDS_PEERS)
+              cmds = node.cluster.rack_aware ? CMDS_REBALANCE : CMDS_PEERS 
+              info_map = ::Aerospike::Info.request(conn, *cmds)
               Verify::PeersGeneration.(node, info_map, peers)
               Verify::PartitionGeneration.(node, info_map)
+              Verify::RebalanceGeneration.(node, info_map) if node.cluster.rack_aware
               Verify::Name.(node, info_map)
               Verify::ClusterName.(node, info_map)
             else

data/lib/aerospike/node/refresh/racks.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+# Copyright 2018-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.
+
+module Aerospike
+  class Node
+    module Refresh
+      module Racks
+        class << self
+          def call(node)
+            return unless should_refresh?(node)
+
+            Aerospike.logger.info("Updating racks for node #{node.name}")
+            conn = node.tend_connection
+            parser = RackParser.new(node, conn)
+            node.update_racks(parser)
+          rescue ::Aerospike::Exceptions::Aerospike => e
+            conn.close
+            Refresh::Failed.(node, e)
+          end
+
+          # Do not refresh racks when node connection has already failed
+          # during this cluster tend iteration.
+          def should_refresh?(node)
+            return false if node.failed? || !node.active?
+            true
+          end
+        end
+      end
+    end
+  end
+end

data/lib/aerospike/node/refresh/reset.rb

@@ -28,6 +28,7 @@ module Aerospike
             node.reset_reference_count!
             node.reset_responded!
             node.partition_generation.reset_changed!
+            node.rebalance_generation.reset_changed!
           end
         end
       end

data/lib/aerospike/node/verify/rebalance_generation.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+# Copyright 2018 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.
+
+module Aerospike
+  class Node
+    module Verify
+      # Fetch and set rebalance generation. If racks needs to be refreshed
+      # this will be indicated in node.rebalance_changed
+      module RebalanceGeneration
+        class << self
+          def call(node, info_map)
+            gen_string = info_map.fetch('rebalance-generation', nil)
+
+            raise Aerospike::Exceptions::Parse.new('rebalance-generation is empty') if gen_string.to_s.empty?
+
+            generation = gen_string.to_i
+
+            node.rebalance_generation.update(generation)
+
+            return unless node.rebalance_generation.changed?
+            Aerospike.logger.info("Node #{node.name} rebalance generation #{generation} changed")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/aerospike/operation.rb

@@ -20,7 +20,7 @@ module Aerospike
 
   class Operation
 
-    attr_reader :op_type, :bin_name, :bin_value
+    attr_reader :op_type, :bin_name, :bin_value, :ctx
 
     READ        = 1
     READ_HEADER = 1
@@ -31,12 +31,17 @@ module Aerospike
     APPEND      = 9
     PREPEND     = 10
     TOUCH       = 11
+    BIT_READ    = 12
+    BIT_MODIFY  = 13
     DELETE      = 14
+    HLL_READ    = 15
+    HLL_MODIFY  = 16
 
-    def initialize(op_type, bin_name=nil, bin_value=NullValue.new)
+    def initialize(op_type, bin_name=nil, bin_value=NullValue.new, ctx = nil)
       @op_type = op_type
       @bin_name = bin_name
       @bin_value = Value.of(bin_value)
+      @ctx = ctx
       self
     end
 

data/lib/aerospike/policy/client_policy.rb

@@ -27,6 +27,7 @@ module Aerospike
     attr_accessor :cluster_name
     attr_accessor :tls
     attr_accessor :policies
+    attr_accessor :rack_aware, :rack_id
 
     def initialize(opt={})
       # Initial host connection timeout in seconds. The timeout when opening a connection
@@ -56,6 +57,20 @@ module Aerospike
 
       # Default Policies
       @policies = opt.fetch(:policies) { Hash.new }
+
+      # Track server rack data.  This field is useful when directing read commands to the server node
+      # that contains the key and exists on the same rack as the client.  This serves to lower cloud
+      # provider costs when nodes are distributed across different racks/data centers.
+      #
+      # ClientPolicy#rack_id, Replica#PREFER_RACK and server rack
+      # configuration must also be set to enable this functionality.
+      @rack_aware = opt[:rack_aware] || false
+
+      # Rack where this client instance resides.
+      #
+      # ClientPolicy#rack_aware, Replica#PREFER_RACK and server rack
+      # configuration must also be set to enable this functionality.
+      @rack_id = opt[:rack_id] || 0
     end
 
     def requires_authentication