aerospike 0.1.0

data/lib/aerospike/command/delete_command.rb

@@ -0,0 +1,57 @@
+# encoding: utf-8
+# Copyright 2014 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/command/single_command'
+require 'aerospike/result_code'
+
+module Aerospike
+
+  private
+
+  class DeleteCommand < SingleCommand
+
+    attr_reader :existed
+
+    def initialize(cluster, policy, key)
+      super(cluster, key)
+
+      @policy = policy
+
+      self
+    end
+
+    def write_buffer
+      set_delete(@policy, @key)
+    end
+
+    def parse_result
+      # Read header.
+      @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+
+      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)
+      end
+
+      @existed = (result_code == 0)
+
+      empty_socket
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/execute_command.rb

@@ -0,0 +1,42 @@
+# encoding: utf-8
+# Copyright 2014 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/value/value'
+require 'aerospike/command/read_command'
+
+module Aerospike
+
+  private
+
+  class ExecuteCommand < ReadCommand
+
+    def initialize(cluster, policy, key, package_name, function_name, args)
+      super(cluster, policy, key, nil)
+
+      @package_name = package_name
+      @function_name = function_name
+      @args = ListValue.new(args)
+
+      self
+    end
+
+    def write_buffer
+      set_udf(@key, @package_name, @function_name, @args)
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/exists_command.rb

@@ -0,0 +1,57 @@
+# encoding: utf-8
+# Copyright 2014 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/command/single_command'
+require 'aerospike/result_code'
+
+module Aerospike
+
+  private
+
+  class ExistsCommand < SingleCommand
+
+    attr_reader :exists
+
+    def initialize(cluster, policy, key)
+      super(cluster, key)
+
+      @policy = policy
+
+      self
+    end
+
+    def write_buffer
+      set_exists(@key)
+    end
+
+    def parse_result
+      # Read header.
+      @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+
+      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)
+      end
+
+      @exists = (result_code == 0)
+
+      empty_socket
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/field_type.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+# Copyright 2014 Aerospike, Inc.
+#
+# 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
+
+  private
+
+  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
+    INDEX_NAME = 21
+    INDEX_RANGE = 22
+    INDEX_FILTER = 23
+    INDEX_LIMIT = 24
+    INDEX_ORDER_BY = 25
+    UDF_PACKAGE_NAME = 30
+    UDF_FUNCTION = 31
+    UDF_ARGLIST = 32
+    UDF_OP = 33
+    QUERY_BINLIST = 40
+
+  end # module
+
+end # module

data/lib/aerospike/command/operate_command.rb

@@ -0,0 +1,37 @@
+# encoding: utf-8
+# Copyright 2014 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/command/read_command'
+
+module Aerospike
+
+  private
+
+  class OperateCommand < ReadCommand
+
+    def initialize(cluster, policy, key, operations)
+      super(cluster, policy, key, nil)
+
+      @operations = operations
+    end
+
+    def write_buffer
+      set_operate(@policy, @key, @operations)
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/read_command.rb

@@ -0,0 +1,174 @@
+# encoding: utf-8
+# Copyright 2014 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/record'
+
+require 'aerospike/command/single_command'
+require 'aerospike/utils/epoc'
+require 'aerospike/value/value'
+
+module Aerospike
+
+  private
+
+  class ReadCommand < SingleCommand
+
+    attr_reader :record
+
+    def initialize(cluster, policy, key, bin_names)
+      super(cluster, key)
+
+      @bin_names = bin_names
+      @policy = policy
+
+      self
+    end
+
+    def write_buffer
+      set_read(@key, @bin_names)
+    end
+
+    def parse_result
+      # Read header.
+      begin
+        @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+      rescue => e
+        Aerospike.logger.error("parse result error: #{e}")
+        raise e
+      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)
+      header_length = @data_buffer.read(8).ord
+      result_code = @data_buffer.read(13).ord & 0xFF
+      generation = @data_buffer.read_int32(14)
+      expiration = Aerospike.TTL(@data_buffer.read_int32(18))
+      field_count = @data_buffer.read_int16(26) # almost certainly 0
+      op_count = @data_buffer.read_int16(28)
+      receive_size = (sz & 0xFFFFFFFFFFFF) - header_length
+
+      # Read remaining message bytes.
+      if receive_size > 0
+        size_buffer_sz(receive_size)
+
+        begin
+          @conn.read(@data_buffer, receive_size)
+        rescue => e
+          Aerospike.logger.warn("parse result error: #{e}")
+          raise e
+        end
+
+      end
+
+      if result_code != 0
+        return if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
+
+        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.warn("UDF execution error: #{e}")
+            raise e
+          end
+
+        end
+
+        raise Aerospike::Exceptions::Aerospike.new(result_code)
+      end
+
+      if op_count == 0
+        # data Bin was not returned.
+        @record = Record.new(@node, @key, nil, nil, generation, expiration)
+        return
+      end
+
+      @record = parse_record(op_count, field_count, generation, expiration)
+    end
+
+    def handle_udf_error(result_code)
+      ret = @record.bins['FAILURE']
+      raise Aerospike::Exceptions::Aerospike.new(result_code, ret) if ret
+      raise Aerospike::Exceptions::Aerospike.new(result_code)
+    end
+
+    def parse_record(op_count, field_count, generation, expiration)
+      bins = nil
+      duplicates = nil
+      receive_offset = 0
+
+      # There can be fields in the response (setname etc).
+      # But for now, ignore them. Expose them to the API if needed in the future.
+      if field_count != 0
+        # Just skip over all the fields
+        for i in 0...field_count
+          field_size = @data_buffer.read_int32(receive_offset)
+          receive_offset += (4 + field_size)
+        end
+      end
+
+      for i in 0...op_count
+        op_size = @data_buffer.read_int32(receive_offset)
+        particle_type = @data_buffer.read(receive_offset+5).ord
+        version = @data_buffer.read(receive_offset+6).ord
+        name_size = @data_buffer.read(receive_offset+7).ord
+        name = @data_buffer.read(receive_offset+8, name_size).force_encoding('utf-8')
+        receive_offset += 4 + 4 + name_size
+
+
+        particle_bytes_size = op_size - (4 + name_size)
+        value = Aerospike.bytes_to_particle(particle_type, @data_buffer, receive_offset, particle_bytes_size)
+        receive_offset += particle_bytes_size
+
+        vmap = {}
+
+        if version > 0 || duplicates != nil
+          unless duplicates
+            duplicates = []
+            duplicates << bins
+            bins = nil
+
+            for j in 0..version-1
+              duplicates << nil
+            end
+          else
+            for j in duplicates.length..version
+              duplicates << nil
+            end
+          end
+
+          vmap = duplicates[version]
+          unless vmap
+            vmap = {}
+            duplicates[version] = vmap
+          end
+        else
+          bins = {} unless bins
+          vmap = bins
+        end
+        vmap[name] = value
+      end
+
+      # Remove nil duplicates just in case there were holes in the version number space.
+      duplicates.compact! if duplicates
+
+      Record.new(@node, @key, bins, duplicates, generation, expiration)
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/read_header_command.rb

@@ -0,0 +1,63 @@
+# encoding: utf-8
+# Copyright 2014 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/command/single_command'
+require 'aerospike/result_code'
+
+module Aerospike
+
+  private
+
+  class ReadHeaderCommand < SingleCommand
+
+    attr_reader :record
+
+    def initialize(cluster, policy, key)
+      super(cluster, key)
+
+      @policy = policy
+
+      self
+    end
+
+    def write_buffer
+      set_read_header(@key)
+    end
+
+    def parse_result
+      # Read header.
+      @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+
+      result_code = @data_buffer.read(13).ord & 0xFF
+
+      if result_code == 0
+        generation = @data_buffer.read_int32(14)
+        expiration = Aerospike.TTL(@data_buffer.read_int32(18))
+        @record = Record.new(@node, @key, nil, nil, generation, expiration)
+      else
+        if result_code == Aerospike::ResultCode::KEY_NOT_FOUND_ERROR
+          @record = nil
+        else
+          raise Aerospike::Exceptions::Aerospike.new(result_code)
+        end
+      end
+
+      empty_socket
+    end
+
+  end # class
+
+end # module