aerospike 2.11.0 → 2.16.0

data/lib/aerospike/cdt/bit_overflow_action.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+# Copyright 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
+  module CDT
+
+    # BitOverflowAction specifies the action to take when bitwise add/subtract results in overflow/underflow.
+    module BitOverflowAction
+
+      ##
+      # Fail specifies to fail operation with error.
+      FAIL  = 0
+
+      ##
+      # SATURATE specifies that in add/subtract overflows/underflows, set to max/min value.
+      # Example: MAXINT + 1 = MAXINT
+      SATURATE  = 2
+
+      ##
+      # Wrap specifies that in add/subtract overflows/underflows, WRAP the value.
+      # Example: MAXINT + 1 = -1
+      WRAP = 4
+
+      ##
+      # Default behavior
+      DEFAULT = FAIL
+
+    end
+  end
+end

data/lib/aerospike/cdt/bit_policy.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+# Copyright 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
+  module CDT
+
+    # BitPolicy determines the Bit operation policy.
+    class BitPolicy
+
+      attr_accessor :flags
+
+      def initialize(flags = BitWriteFlags::DEFAULT)
+        @flags = flags
+      end
+
+      DEFAULT = BitPolicy.new
+
+    end
+  end
+end

data/lib/aerospike/cdt/bit_resize_flags.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+# Copyright 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
+  module CDT
+
+    ##
+    # BitResizeFlags specifies the bitwise operation flags for resize.
+    module BitResizeFlags
+
+      ##
+      # Default specifies the defalt flag.
+      DEFAULT = 0
+
+      ##
+      # Adds/removes bytes from the beginning instead of the end.
+      FROM_FRONT = 1
+
+      ##
+      # only allow the byte array size to increase.
+      GROW_ONLY = 2
+
+      ##
+      # only allow the byte array size to decrease.
+      SHRINK_ONLY = 4
+    end
+  end
+end

data/lib/aerospike/cdt/bit_write_flags.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+# Copyright 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
+  module CDT
+
+    ##
+    # BitWriteFlags specify bitwise operation policy write flags.
+    module BitWriteFlags
+
+    ##
+    # Default allows create or update.
+    DEFAULT = 0
+
+    ##
+    # If the bin already exists, the operation will be denied.
+    # If the bin does not exist, a new bin will be created.
+    CREATE_ONLY = 1
+
+    ##
+    # If the bin already exists, the bin will be overwritten.
+    # If the bin does not exist, the operation will be denied.
+    UPDATE_ONLY = 2
+
+    ##
+    # Will not raise error if operation is denied.
+    NO_FAIL = 4
+
+    ##
+    # Partial allows other valid operations to be committed if this operations is
+    # denied due to flag constraints.
+    PARTIAL = 8
+    end
+  end
+end

data/lib/aerospike/cdt/context.rb

@@ -0,0 +1,101 @@
+# frozen_string_literal: true
+
+# Copyright 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
+  module CDT
+
+  	##
+	# Nested CDT context. Identifies the location of nested list/map to apply the operation.
+	# for the current level.
+	# An array of CTX identifies location of the list/map on multiple
+	# levels on nesting.
+    class Context
+
+      attr_accessor :id, :value
+
+      def initialize(id, value)
+        @id = id
+        @value = value
+      end
+
+	  ##
+	  # Lookup list by index offset.
+	  # If the index is negative, the resolved index starts backwards from end of list.
+	  # If an index is out of bounds, a parameter error will be returned.
+	  # Examples:
+	  # 0: First item.
+	  # 4: Fifth item.
+	  # -1: Last item.
+	  # -3: Third to last item.
+	  def self.list_index(index)
+	  	Context.new(0x10, index)
+	  end
+	  
+	  ##
+	  # Lookup list by rank.
+	  # 0 = smallest value
+	  # N = Nth smallest value
+	  # -1 = largest value
+	  def self.list_rank(rank)
+	  	Context.new(0x11, rank)
+	  end
+	  
+	  ##
+	  # Lookup list by value.
+	  def self.list_value(key)
+	  	Context.new(0x13, key)
+	  end
+	  
+	  ##
+	  # Lookup map by index offset.
+	  # If the index is negative, the resolved index starts backwards from end of list.
+	  # If an index is out of bounds, a parameter error will be returned.
+	  # Examples:
+	  # 0: First item.
+	  # 4: Fifth item.
+	  # -1: Last item.
+	  # -3: Third to last item.
+	  def self.map_index(index)
+	  	Context.new(0x20, index)
+	  end
+	  
+	  ##
+	  # Lookup map by rank.
+	  # 0 = smallest value
+	  # N = Nth smallest value
+	  # -1 = largest value
+	  def self.map_rank(rank)
+	  	Context.new(0x21, rank)
+	  end
+	  
+	  ##
+	  # Lookup map by key.
+	  def self.map_key(key)
+	  	Context.new(0x22, key)
+	  end
+	  
+	  ##
+	  # Lookup map by value.
+	  def self.map_value(key)
+	  	Context.new(0x23, key)
+	  end      
+
+    end
+  end
+end

data/lib/aerospike/cdt/hll_operation.rb

@@ -0,0 +1,200 @@
+# 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.
+
+module Aerospike
+  module CDT
+
+    # HyperLogLog (HLL) operations.
+    # Requires server versions >= 4.9.
+    #
+    # HyperLogLog operations on HLL items nested in lists/maps are not currently
+    # supported by the server.
+    class HLLOperation < Operation
+
+      INIT            = 0
+      ADD             = 1
+      SET_UNION       = 2
+      SET_COUNT       = 3
+      FOLD            = 4
+      COUNT           = 50
+      UNION           = 51
+      UNION_COUNT     = 52
+      INTERSECT_COUNT = 53
+      SIMILARITY      = 54
+      DESCRIBE        = 55
+
+      attr_reader :hll_op, :values, :return_type, :policy, :index_bit_count, :minhash_bit_count
+
+      def initialize(op_type, hll_op, bin_name, values: nil, index_bit_count: nil, minhash_bit_count: nil, policy: nil)
+        @policy = policy
+        @op_type = op_type
+        @bin_name = bin_name
+        @bin_value = nil
+        @hll_op = hll_op
+        @index_bit_count = index_bit_count
+        @minhash_bit_count = minhash_bit_count
+        @values = values
+
+        self
+      end
+
+      ##
+      # Create HLL init operation with minhash bits.
+      # Server creates a new HLL or resets an existing HLL.
+      # Server does not return a value.
+      #
+      # policy      write policy, HLLPolicy::DEFAULT is default
+      # bin_name     name of bin
+      # index_bit_count   number of index bits. Must be between 4 and 16 inclusive.
+      # minhash_bit_count   number of min hash bits. Must be between 4 and 58 inclusive.
+      def self.init(bin_name, index_bit_count, minhash_bit_count, policy = HLLPolicy::DEFAULT)
+        HLLOperation.new(Operation::HLL_MODIFY, INIT, bin_name, index_bit_count: index_bit_count, minhash_bit_count: minhash_bit_count, policy: policy)
+      end
+
+      ##
+      # Create HLL add operation with minhash bits.
+      # Server adds values to HLL set. If HLL bin does not exist, use index_bit_count and minhash_bit_count
+      # to create HLL bin. Server returns number of entries that caused HLL to update a register.
+      #
+      # policy      write policy, HLLPolicy::DEFAULT is default
+      # bin_name     name of bin
+      # list        list of values to be added
+      # index_bit_count   number of index bits. Must be between 4 and 16 inclusive.
+      # minhash_bit_count   number of min hash bits. Must be between 4 and 58 inclusive.
+      def self.add(bin_name, *values, policy: HLLPolicy::DEFAULT, index_bit_count: -1, minhash_bit_count: -1)
+        HLLOperation.new(Operation::HLL_MODIFY, ADD, bin_name, index_bit_count: index_bit_count, minhash_bit_count: minhash_bit_count, values: values, policy: policy)
+      end
+
+      ##
+      # Create HLL set union operation.
+      # Server sets union of specified HLL objects with HLL bin.
+      # Server does not return a value.
+      #
+      # policy      write policy, HLLPolicy::DEFAULT is default
+      # bin_name     name of bin
+      # list        list of HLL objects
+      def self.set_union(bin_name, *values, policy: HLLPolicy::DEFAULT)
+        HLLOperation.new(Operation::HLL_MODIFY, SET_UNION, bin_name, values: values, policy: policy)
+      end
+
+      ##
+      # Create HLL refresh operation.
+      # Server updates the cached count (if stale) and returns the count.
+      #
+      # bin_name     name of bin
+      def self.refresh_count(bin_name)
+        HLLOperation.new(Operation::HLL_MODIFY, SET_COUNT, bin_name)
+      end
+
+      ##
+      # Create HLL fold operation.
+      # Servers folds index_bit_count to the specified value.
+      # This can only be applied when minhash_bit_count on the HLL bin is 0.
+      # Server does not return a value.
+      #
+      # bin_name     name of bin
+      # index_bit_count   number of index bits. Must be between 4 and 16 inclusive.
+      def self.fold(bin_name, index_bit_count)
+        HLLOperation.new(Operation::HLL_MODIFY, FOLD, bin_name, index_bit_count: index_bit_count)
+      end
+
+      ##
+      # Create HLL getCount operation.
+      # Server returns estimated number of elements in the HLL bin.
+      #
+      # bin_name     name of bin
+      def self.get_count(bin_name)
+        HLLOperation.new(Operation::HLL_READ, COUNT, bin_name)
+      end
+
+      ##
+      # Create HLL getUnion operation.
+      # Server returns an HLL object that is the union of all specified HLL objects in the list
+      # with the HLL bin.
+      #
+      # bin_name     name of bin
+      # list        list of HLL objects
+      def self.get_union(bin_name, *values)
+        HLLOperation.new(Operation::HLL_READ, UNION, bin_name, values: values)
+      end
+
+      ##
+      # Create HLL getUnionCount operation.
+      # Server returns estimated number of elements that would be contained by the union of these
+      # HLL objects.
+      # bin_name     name of bin
+      # list        list of HLL objects
+      def self.get_union_count(bin_name, *values)
+        HLLOperation.new(Operation::HLL_READ, UNION_COUNT, bin_name, values: values)
+      end
+
+      ##
+      # Create HLL getIntersectCount operation.
+      # Server returns estimated number of elements that would be contained by the intersection of
+      # these HLL objects.
+      #
+      # bin_name     name of bin
+      # list        list of HLL objects
+      def self.get_intersect_count(bin_name, *values)
+        HLLOperation.new(Operation::HLL_READ, INTERSECT_COUNT, bin_name, values: values)
+      end
+
+      ##
+      # Create HLL getSimilarity operation.
+      # Server returns estimated similarity of these HLL objects. Return type is a double.
+      #
+      # bin_name     name of bin
+      # list        list of HLL objects
+      def self.get_similarity(bin_name, *values)
+        HLLOperation.new(Operation::HLL_READ, SIMILARITY, bin_name, values: values)
+      end
+
+      ##
+      # Create HLL describe operation.
+      # Server returns index_bit_count and minhash_bit_count used to create HLL bin in a list of longs.
+      # The list size is 2.
+      #
+      # bin_name     name of bin
+      def self.describe(bin_name)
+        HLLOperation.new(Operation::HLL_READ, DESCRIBE, bin_name)
+      end
+
+      def bin_value
+        @bin_value ||= pack_bin_value
+      end
+
+      private
+
+      def pack_bin_value
+        bytes = nil
+        Packer.use do |packer|
+          args = [hll_op]
+          args << values if values
+          args << index_bit_count if index_bit_count
+          args << minhash_bit_count if minhash_bit_count
+          args << policy.flags if policy
+
+          vv = ListValue.new(args)
+          vv.pack(packer)
+          bytes = packer.bytes
+        end
+        BytesValue.new(bytes)
+      end
+
+    end
+
+  end
+end