aerospike 2.19.0 → 2.26.0

data/lib/aerospike/role.rb

@@ -0,0 +1,55 @@
+# encoding: utf-8
+# Copyright 2014-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
+
+  # Role provides granular access to database entities for users.
+  class Role
+
+    # Role name
+    attr_accessor :name
+
+    # List of assigned privileges
+    attr_accessor :privileges
+
+    # List of allowable IP addresses
+    attr_accessor :allowlist
+
+    # Maximum reads per second limit for the role
+    attr_accessor :read_quota
+
+    # Maximum writes per second limit for the role
+    attr_accessor :write_quota
+
+    # The following aliases are for backward compatibility reasons
+    USER_ADMIN = Privilege::USER_ADMIN # :nodoc:
+    SYS_ADMIN = Privilege::SYS_ADMIN # :nodoc:
+    DATA_ADMIN = Privilege::DATA_ADMIN # :nodoc:
+    UDF_ADMIN = Privilege::UDF_ADMIN # :nodoc:
+    SINDEX_ADMIN = Privilege::SINDEX_ADMIN # :nodoc:
+    READ_WRITE_UDF = Privilege::READ_WRITE_UDF # :nodoc:
+    READ_WRITE = Privilege::READ_WRITE # :nodoc:
+    READ = Privilege::READ # :nodoc:
+    WRITE = Privilege::WRITE # :nodoc:
+    TRUNCATE = Privilege::TRUNCATE # :nodoc:
+
+    def to_s
+      "Role [name=#{@name}, privileges=#{@privileges}, allowlist=#{@allowlist}, readQuota=#{@read_quota}, writeQuota=#{@write_quota}]";
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/task/execute_task.rb

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 module Aerospike
-
   private
 
   # ExecuteTask is used to poll for long running server execute job completion.
@@ -29,19 +28,24 @@ module Aerospike
       self
     end
 
-    # IsDone queries all nodes for task completion status.
+    # queries all nodes for task completion status.
     def all_nodes_done?
-
-      if @scan
-        command = 'scan-list'
-      else
-        command = 'query-list'
-      end
+      modul = @scan ? "scan" : "query"
+      cmd1 = "query-show:trid=#{@task_id}"
+      cmd2 = modul + "-show:trid=#{@task_id}"
+      cmd3 = "jobs:module=" + modul + ";cmd=get-job;trid=#{@task_id}"
 
       nodes = @cluster.nodes
       done = false
 
       nodes.each do |node|
+        command = cmd3
+        if node.supports_feature?(Aerospike::Features::PARTITION_QUERY)
+          command = cmd1
+        elsif node.supports_feature?(Aerospike::Features::QUERY_SHOW)
+          command = cmd2
+        end
+
         conn = node.get_connection(0)
         responseMap, _ = Info.request(conn, command)
         node.put_connection(conn)
@@ -58,28 +62,27 @@ module Aerospike
 
         b = index + find.length
         response = response[b, response.length]
-        find = 'job_status='
+        find = "job_status="
         index = response.index(find)
 
         next unless index
 
         b = index + find.length
         response = response[b, response.length]
-        e = response.index(':')
+        e = response.index(":")
         status = response[0, e]
 
         case status
-        when 'ABORTED'
-          raise raise Aerospike::Exceptions::QueryTerminated
-        when 'IN PROGRESS'
+        when "ABORTED"
+          raise Aerospike::Exceptions::QueryTerminated
+        when "IN PROGRESS"
           return false
-        when 'DONE'
+        when "DONE"
           done = true
         end
       end
 
       done
     end
-
   end
-end
+end

data/lib/aerospike/task/index_task.rb

@@ -25,7 +25,7 @@ module Aerospike
 
   class IndexTask < Task
 
-    MATCHER = /.*load_pct=(?<load_pct>\d+(\.\d+)?).*/
+    MATCHER = /.*load_pct=(?<load_pct>\d+(\.\d+)?).*/.freeze
 
     def initialize(cluster, namespace, index_name, done=false)
       super(cluster, done)

data/lib/aerospike/user_role.rb

@@ -25,6 +25,31 @@ module Aerospike
 		# List of assigned roles.
 		attr_accessor :roles
 
+		# List of read statistics. List may be nil.
+		# Current statistics by offset are:
+		#
+		# 0: read quota in records per second
+		# 1: single record read transaction rate (TPS)
+		# 2: read scan/query record per second rate (RPS)
+		# 3: number of limitless read scans/queries
+		#
+		# Future server releases may add additional statistics.
+		attr_accessor :read_info
+
+		# List of write statistics. List may be nil.
+		# Current statistics by offset are:
+		#
+		# 0: write quota in records per second
+		# 1: single record write transaction rate (TPS)
+		# 2: write scan/query record per second rate (RPS)
+		# 3: number of limitless write scans/queries
+		#
+		# Future server releases may add additional statistics.
+		attr_accessor :write_info
+
+		# Number of currently open connections for the user
+		attr_accessor :conns_in_use
+
 	end
 
-end
+end

data/lib/aerospike/utils/buffer.rb

@@ -17,34 +17,34 @@
 # License for the specific language governing permissions and limitations under
 # the License.
 
-require 'aerospike/utils/pool'
+require "aerospike/utils/pool"
 
 module Aerospike
-
   private
 
   # Buffer class to ease the work around
   class Buffer #:nodoc:
-
     @@buf_pool = Pool.new
     @@buf_pool.create_proc = Proc.new { Buffer.new }
 
     attr_accessor :buf
 
-    INT16 = 's>'
-    UINT16 = 'n'
-    INT32 = 'l>'
-    UINT32 = 'N'
-    INT64 = 'q>'
-    UINT64 = 'Q>'
-    DOUBLE = 'G'
+    INT16 = "s>"
+    UINT16 = "n"
+    UINT16LE = "v"
+    INT32 = "l>"
+    UINT32 = "N"
+    INT64 = "q>"
+    UINT64 = "Q>"
+    UINT64LE = "Q"
+    DOUBLE = "G"
 
     DEFAULT_BUFFER_SIZE = 16 * 1024
     MAX_BUFFER_SIZE = 10 * 1024 * 1024
 
-    def initialize(size=DEFAULT_BUFFER_SIZE, buf = nil)
+    def initialize(size = DEFAULT_BUFFER_SIZE, buf = nil)
       @buf = (buf ? buf : ("%0#{size}d" % 0))
-      @buf.force_encoding('binary')
+      @buf.force_encoding("binary")
       @slice_end = @buf.bytesize
     end
 
@@ -59,6 +59,7 @@ module Aerospike
     def size
       @buf.bytesize
     end
+
     alias_method :length, :size
 
     def eat!(n)
@@ -66,6 +67,12 @@ module Aerospike
     end
 
     def resize(length)
+      # Corrupted data streams can result in a hug.length.
+      # Do a sanity check here.
+      if length > MAX_BUFFER_SIZE
+        raise Aerospike::Exceptions::Parse.new("Invalid size for buffer: #{length}")
+      end
+
       if @buf.bytesize < length
         @buf.concat("%0#{length - @buf.bytesize}d" % 0)
       end
@@ -92,6 +99,11 @@ module Aerospike
       2
     end
 
+    def write_uint16_little_endian(i, offset)
+      @buf[offset, 2] = [i].pack(UINT16LE)
+      2
+    end
+
     def write_int32(i, offset)
       @buf[offset, 4] = [i].pack(INT32)
       4
@@ -112,12 +124,17 @@ module Aerospike
       8
     end
 
+    def write_uint64_little_endian(i, offset)
+      @buf[offset, 8] = [i].pack(UINT64LE)
+      8
+    end
+
     def write_double(f, offset)
       @buf[offset, 8] = [f].pack(DOUBLE)
       8
     end
 
-    def read(offset, len=nil)
+    def read(offset, len = nil)
       if len
         @buf[offset, len]
       else
@@ -126,56 +143,103 @@ module Aerospike
     end
 
     def read_int16(offset)
-      vals = @buf[offset..offset+1]
+      vals = @buf[offset..offset + 1]
       vals.unpack(INT16)[0]
     end
 
+    def read_uint16(offset)
+      vals = @buf[offset..offset + 1]
+      vals.unpack(UINT16)[0]
+    end
+
     def read_int32(offset)
-      vals = @buf[offset..offset+3]
+      vals = @buf[offset..offset + 3]
       vals.unpack(INT32)[0]
     end
 
+    def read_uint32(offset)
+      vals = @buf[offset..offset + 3]
+      vals.unpack(UINT32)[0]
+    end
+
     def read_int64(offset)
-      vals = @buf[offset..offset+7]
+      vals = @buf[offset..offset + 7]
       vals.unpack(INT64)[0]
     end
 
+    def read_uint64_little_endian(offset)
+      vals = @buf[offset..offset + 7]
+      vals.unpack(UINT64LE)[0]
+    end
+
+    def read_uint64(offset)
+      vals = @buf[offset..offset + 7]
+      vals.unpack(UINT64)[0]
+    end
+
     def read_var_int64(offset, len)
       val = 0
       i = 0
       while i < len
         val <<= 8
-        val |= @buf[offset+i].ord & 0xFF
+        val |= @buf[offset + i].ord & 0xFF
         i = i.succ
       end
       val
     end
 
     def read_double(offset)
-      vals = @buf[offset..offset+7]
+      vals = @buf[offset..offset + 7]
       vals.unpack(DOUBLE)[0]
     end
 
+    def read_bool(offset, length)
+      length <= 0 ? false : @buf[offset].ord != 0
+    end
+
     def to_s
-      @buf[0..@slice_end-1]
+      @buf[0..@slice_end - 1]
     end
 
     def reset
-      for i in 0..@buf.size-1
-        @buf[i] = ' '
+      for i in 0..@buf.size - 1
+        @buf[i] = " "
       end
     end
 
-    def dump(from=nil, to=nil)
-      from ||= 0
-      to ||= @slice_end - 1
+    def dump(start = 0, finish = nil)
+      buf ||= @buf.bytes
+      finish ||= @slice_end - 1
+      width = 16
+
+      ascii = "|"
+      counter = 0
+
+      print "%08x  " % start
+      @buf.bytes[start...finish].each do |c|
+        if counter >= start
+          print "%02x " % c
+          ascii << (c.between?(32, 126) ? c : ?.)
+          print " " if ascii.length == (width / 2 + 1)
+          if ascii.length > width
+            ascii << "|"
+            puts ascii
+            ascii = "|"
+            print "%08x  " % (counter + 1)
+          end
+        end
+        counter += 1
+      end
 
-      @buf.bytes[from...to].each do |c|
-        print c.ord.to_s(16)
-        putc ' '
+      # print the remainder in buffer
+      if ascii.length.positive?
+        fill_size = ((width - ascii.length + 1) * 3)
+        fill_size += 1 if ascii.length <= (width / 2)
+        filler = " " * fill_size
+        print filler
+        ascii << "|"
+        puts ascii
       end
     end
-
   end # buffer
-
 end # module

data/lib/aerospike/utils/packer.rb

@@ -14,13 +14,11 @@
 # License for the specific language governing permissions and limitations under
 # the License.
 
-require 'msgpack'
-require 'aerospike/utils/pool'
+require "msgpack"
+require "aerospike/utils/pool"
 
 module Aerospike
-
   class Packer < MessagePack::Packer #:nodoc:
-
     AS_EXT_TYPE = -1
 
     @@pool = Pool.new
@@ -44,9 +42,12 @@ module Aerospike
       buffer << [val].pack("S>")
     end
 
+    def write_raw(buf)
+      buffer.write(buf)
+    end
+
     def bytes
-      self.to_s.force_encoding('binary')
+      self.to_s.force_encoding("binary")
     end
   end
-
 end

data/lib/aerospike/utils/pool.rb

@@ -53,7 +53,7 @@ module Aerospike
     end
 
     def empty?
-      @pool.length == 0
+      @pool.empty?
     end
 
     def length

data/lib/aerospike/value/particle_type.rb

@@ -23,19 +23,8 @@ module Aerospike
     DOUBLE = 2
     STRING = 3
     BLOB = 4
-    #TIMESTAMP = 5
-    #DIGEST = 6
-    #JBLOB = 7
-    #CSHARP_BLOB = 8
-    #PYTHON_BLOB = 9
     RUBY_BLOB = 10
-    #PHP_BLOB = 11
-    #ERLANG_BLOB = 12
-    #SEGMENT_POINTER = 13
-    #RTA_LIST = 14
-    #RTA_DICT = 15
-    #RTA_APPEND_DICT = 16
-    #RTA_APPEND_LIST = 17
+    BOOL = 17
     HLL = 18
     MAP = 19
     LIST = 20