aerospike 2.9.0 → 2.13.0

data/lib/aerospike/command/delete_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -33,6 +33,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @cluster.master_node(@partition)
+    end
+
     def write_buffer
       set_delete(@policy, @key)
     end
@@ -43,13 +47,25 @@ module Aerospike
 
       result_code = @data_buffer.read(13).ord & 0xFF
 
-      if (result_code != 0) && (result_code != Aerospike::ResultCode::KEY_NOT_FOUND_ERROR)
-        raise Aerospike::Exceptions::Aerospike.new(result_code)
+      if result_code == 0
+        @existed = true
+        return
       end
 
-      @existed = (result_code == 0)
+      if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
+        @existed = false
+        return
+      end
+
+      if result_code == Aerospike::ResultCode::FILTERED_OUT
+        if @policy.fail_on_filtered_out
+          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        end
+        @existed = true
+        return
+      end
 
-      empty_socket
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
     end
 
   end # class

data/lib/aerospike/command/execute_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.

data/lib/aerospike/command/exists_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -33,6 +33,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @cluster.read_node(@partition, @policy.replica, @sequence)
+    end
+
     def write_buffer
       set_exists(@policy, @key)
     end
@@ -43,13 +47,25 @@ module Aerospike
 
       result_code = @data_buffer.read(13).ord & 0xFF
 
-      if (result_code != 0) && (result_code != Aerospike::ResultCode::KEY_NOT_FOUND_ERROR)
-        raise Aerospike::Exceptions::Aerospike.new(result_code)
+      if result_code == 0
+        @exists = true
+        return
       end
 
-      @exists = (result_code == 0)
+      if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
+        @exists = false
+        return
+      end
+
+      if result_code == Aerospike::ResultCode::FILTERED_OUT
+        if @policy.fail_on_filtered_out
+          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        end
+        @exists = true
+        return
+      end
 
-      empty_socket
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
     end
 
   end # class

data/lib/aerospike/command/field_type.rb

@@ -1,4 +1,4 @@
-# Copyright 2014-2018 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -31,6 +31,7 @@ module Aerospike
     TRAN_ID = 7    # user supplied transaction id, which is simply passed back
     SCAN_OPTIONS = 8
     SCAN_TIMEOUT = 9
+    RECORDS_PER_SECOND = 10
     INDEX_NAME = 21
     INDEX_RANGE = 22
     INDEX_FILTER = 23
@@ -43,6 +44,7 @@ module Aerospike
     UDF_OP = 33
     QUERY_BINLIST = 40
     BATCH_INDEX = 41
+    PREDEXP = 43
 
   end # module
 

data/lib/aerospike/command/multi_command.rb

@@ -1,4 +1,4 @@
-# Copyright 2014-2018 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -32,6 +32,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @node
+    end
+
     def parse_result
       # Read socket into receive buffer one record at a time.  Do not read entire receive size
       # because the receive buffer would be too big.
@@ -59,10 +63,13 @@ module Aerospike
         read_bytes(MSG_REMAINING_HEADER_SIZE)
         result_code = @data_buffer.read(5).ord & 0xFF
 
-        # The only valid server return codes are "ok" and "not found".
+        # The only valid server return codes are "ok", "not found" and "filtered out".
         # If other return codes are received, then abort the batch.
-        if result_code != 0 && result_code != Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
-          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        if result_code != 0 
+            if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR || result_code == Aerospike::ResultCode::FILTERED_OUT
+            else
+              raise Aerospike::Exceptions::Aerospike.new(result_code)
+            end
         end
 
         # If cmd is the end marker of the response, do not proceed further

data/lib/aerospike/command/operate_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -28,6 +28,11 @@ module Aerospike
       @operations = operations
     end
 
+    def get_node
+      @cluster.master_node(@partition)
+    end
+
+
     def write_bins
       @operations.select{|op| op.op_type == Aerospike::Operation::WRITE}.map(&:bin).compact
     end

data/lib/aerospike/command/read_command.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -39,6 +39,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @cluster.read_node(@partition, @policy.replica, @sequence)
+    end
+
     def write_buffer
       set_read(@policy, @key, @bin_names)
     end
@@ -76,29 +80,36 @@ module Aerospike
 
       end
 
-      if result_code != 0
-        return nil if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
-        return nil if result_code == Aerospike::ResultCode::LARGE_ITEM_NOT_FOUND
-        if result_code == Aerospike::ResultCode::UDF_BAD_RESPONSE
-          begin
-            @record = parse_record(op_count, field_count, generation, expiration)
-            handle_udf_error(result_code)
-          rescue => e
-            Aerospike.logger.error("UDF execution error: #{e}")
-            raise e
-          end
-
+      if result_code == 0
+        if op_count == 0
+          @record = Record.new(@node, @key, nil, generation, expiration)
+          return
         end
-
-        raise Aerospike::Exceptions::Aerospike.new(result_code)
+  
+        @record = parse_record(op_count, field_count, generation, expiration)
+        return
       end
 
-      if op_count == 0
-        @record = Record.new(@node, @key, nil, generation, expiration)
+      return nil if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
+
+      if result_code == Aerospike::ResultCode::FILTERED_OUT
+        if @policy.fail_on_filtered_out
+          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        end
         return
       end
 
-      @record = parse_record(op_count, field_count, generation, expiration)
+      if result_code == Aerospike::ResultCode::UDF_BAD_RESPONSE
+        begin
+          @record = parse_record(op_count, field_count, generation, expiration)
+          handle_udf_error(result_code)
+        rescue => e
+          Aerospike.logger.error("UDF execution error: #{e}")
+          raise e
+        end
+      end
+
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
     end
 
     def handle_udf_error(result_code)

data/lib/aerospike/command/read_header_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -33,6 +33,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @cluster.read_node(@partition, @policy.replica, @sequence)
+    end
+
     def write_buffer
       set_read_header(@policy, @key)
     end
@@ -47,15 +51,23 @@ module Aerospike
         generation = @data_buffer.read_int32(14)
         expiration = @data_buffer.read_int32(18)
         @record = Record.new(@node, @key, nil,  generation, expiration)
-      else
-        if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
-          @record = nil
-        else
+        return
+      end
+
+      if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
+        @record = nil
+        return
+      end
+
+      if result_code == Aerospike::ResultCode::FILTERED_OUT
+        @record = nil
+        if @policy.fail_on_filtered_out
           raise Aerospike::Exceptions::Aerospike.new(result_code)
         end
+        return
       end
 
-      empty_socket
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
     end
 
   end # class

data/lib/aerospike/command/roles.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.

data/lib/aerospike/command/single_command.rb

@@ -1,4 +1,4 @@
-# Copyright 2014-2018 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -27,12 +27,18 @@ module Aerospike
       @cluster = cluster
       @key = key
 
-      node = cluster.get_node_for_key(key)
-      super(node)
+      @partition = Partition.new_by_key(key)
+
+      super()
     end
 
     protected
 
+    def get_node
+      # abstract method
+      raise Aerospike::Exceptions::InvalidNode
+    end
+
     def empty_socket
       # There should not be any more bytes.
       # Empty the socket to be safe.

data/lib/aerospike/command/touch_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -30,6 +30,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @cluster.master_node(@partition)
+    end
+
     def write_buffer
       set_touch(@policy, @key)
     end
@@ -40,9 +44,16 @@ module Aerospike
 
       result_code = @data_buffer.read(13).ord & 0xFF
 
-      raise Aerospike::Exceptions::Aerospike.new(result_code) if result_code != 0
+      return if result_code == 0
+
+      if result_code == Aerospike::ResultCode::FILTERED_OUT
+        if @policy.fail_on_filtered_out
+          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        end
+        return
+      end
 
-      empty_socket
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
     end
 
   end # class

data/lib/aerospike/command/unsupported_particle_type_validator.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2015-2017 Aerospike, Inc.
+# Copyright 2015-2020 Aerospike, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License")
 # you may not use this file except in compliance with the License.

data/lib/aerospike/command/write_command.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -33,6 +33,10 @@ module Aerospike
       self
     end
 
+    def get_node
+      @cluster.master_node(@partition)
+    end
+
     def write_bins
       @bins
     end
@@ -52,11 +56,16 @@ module Aerospike
 
       result_code = @data_buffer.read(13).ord & 0xFF
 
-      if result_code != 0
-        raise Aerospike::Exceptions::Aerospike.new(result_code)
+      return if result_code == 0
+
+      if result_code == Aerospike::ResultCode::FILTERED_OUT
+        if @policy.fail_on_filtered_out
+          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        end
+        return
       end
 
-      empty_socket
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
     end
 
   end # class

data/lib/aerospike/connection/create.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-# Copyright 2014-2018 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.

data/lib/aerospike/features.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-# Copyright 2014-2017 Aerospike, Inc.
+# Copyright 2014-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.

data/lib/aerospike/geo_json.rb

@@ -1,5 +1,5 @@
 # encoding: utf-8
-# Copyright 2015-2017 Aerospike, Inc.
+# Copyright 2015-2020 Aerospike, Inc.
 #
 # Portions may be licensed to Aerospike, Inc. under one or more contributor
 # license agreements.
@@ -52,6 +52,75 @@ module Aerospike
       other.to_json == self.to_json
     end
 
+    def lng
+      case type
+      when 'Point'
+        coordinates.first
+      when 'AeroCircle'
+        coordinates.first.first
+      end
+    end
+
+    def lat
+      case type
+      when 'Point'
+        coordinates.last
+      when 'AeroCircle'
+        coordinates.first.last
+      end
+    end
+
+    def radius
+      return nil unless circle?
+
+      coordinates.last
+    end
+
+    def coordinates
+      to_h['coordinates']
+    end
+
+    def type
+      to_h['type']
+    end
+
+    def point?
+      type == 'Point'
+    end
+
+    def circle?
+      type == 'AeroCircle'
+    end
+
+    def polygon?
+      type == 'Polygon'
+    end
+
+    def self.point(lng, lat)
+      new(type: 'Point', coordinates: [lng, lat])
+    end
+
+    def self.circle(lng, lat, radius)
+      new(type: 'AeroCircle', coordinates: [[lng, lat], radius])
+    end
+
+    def self.polygon(coordinates)
+      new(type: 'Polygon', coordinates: coordinates)
+    end
+
+    def to_circle(radius)
+      raise TypeError, 'Cannot create a Circle from a Polygon' if polygon?
+
+      self.class.circle(lng, lat, radius)
+    end
+
+    def to_point
+      return self if point?
+      raise TypeError, 'Cannot create a Point from a Polygon' if polygon?
+
+      self.class.point(lng, lat)
+    end
+
     protected
 
     attr_accessor :json_data