aerospike 0.1.0

data/lib/aerospike/command/single_command.rb

@@ -0,0 +1,60 @@
+# 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/cluster/partition'
+require 'aerospike/command/command'
+
+module Aerospike
+
+  private
+
+  class SingleCommand < Command
+
+    def initialize(cluster, key)
+      @cluster = cluster
+      @key = key
+      @partition = Partition.new_by_key(key)
+
+      super(@cluster.get_node(@partition))
+
+      self
+    end
+
+
+    protected
+
+    def empty_socket
+      # There should not be any more bytes.
+      # Empty the socket to be safe.
+      sz = @data_buffer.read_int64( 0)
+      header_length = @data_buffer.read(8).ord
+      receive_size = Integer(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.error("#{e}")
+          raise e
+        end
+      end
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/touch_command.rb

@@ -0,0 +1,50 @@
+# 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'
+
+module Aerospike
+
+  private
+
+  class TouchCommand < SingleCommand
+
+    def initialize(cluster, policy, key)
+      super(cluster, key)
+
+      @policy = policy
+
+      self
+    end
+
+    def write_buffer
+      set_touch(@policy, @key)
+    end
+
+    def parse_result
+      # Read header.
+      @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+
+      result_code = @data_buffer.read(13).ord & 0xFF
+
+      raise Aerospike::Exceptions::Aerospike.new(result_code) if result_code != 0
+
+      empty_socket
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/command/write_command.rb

@@ -0,0 +1,60 @@
+# 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'
+
+module Aerospike
+
+  private
+
+  class WriteCommand < SingleCommand
+
+    def initialize(cluster, policy, key, bins, operation)
+
+      super(cluster, key)
+
+      @bins =          bins
+      @operation =     operation
+      @policy = policy
+
+      self
+    end
+
+    def write_buffer
+      set_write(@policy, @operation, @key, @bins)
+    end
+
+    def parse_result
+      # Read header.
+      begin
+        @conn.read(@data_buffer, MSG_TOTAL_HEADER_SIZE)
+      rescue => e
+        Aerospike.logger.error("#{e}")
+        raise e
+      end
+
+      result_code = @data_buffer.read(13).ord & 0xFF
+
+      if result_code != 0
+        raise Aerospike::Exceptions::Aerospike.new(result_code)
+      end
+
+      empty_socket
+    end
+
+  end # class
+
+end # module

data/lib/aerospike/host.rb

@@ -0,0 +1,43 @@
+# 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.
+
+module Aerospike
+
+  class Host
+
+    attr_accessor :name, :port
+
+    def initialize(host_name, host_port)
+      @name = host_name
+      @port = host_port
+    end
+
+    def to_s
+      "#{@name}:#{@port.to_s}"
+    end
+
+    def ==(other)
+      other && other.is_a?(Host) && other.name == @name && other.port == @port
+    end
+    alias eql? ==
+
+    def hash
+      to_s.hash
+    end
+
+  end
+
+end

data/lib/aerospike/info.rb

@@ -0,0 +1,96 @@
+# 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/utils/buffer'
+
+module Aerospike
+
+  private
+
+  # Polymorphic value classes used to efficiently serialize objects into the wire protocol.
+  class Info
+
+    def self.request(conn, *commands)
+      buffer = Buffer.get
+
+      # If conservative estimate may be exceeded, get exact estimate
+      # to preserve memory and resize buffer.
+      offset = 8
+
+      commands.each do |command|
+        offset += command.bytesize + 1
+      end
+
+      buffer.resize(offset)
+
+      offset = 8 # Skip size field.
+
+      # The command format is: <name1>\n<name2>\n...
+      commands.each do |command|
+        buffer.write_binary(command, offset)
+        offset += command.bytesize
+        buffer.write_byte("\n", offset)
+        offset += 1
+      end
+
+      begin
+        buf_length = send_command(conn, offset, buffer)
+        parse_multiple_response(buf_length, buffer)
+      rescue => e
+        Aerospike.logger.error("#{e}")
+      ensure
+        Buffer.put(buffer)
+      end
+    end
+
+    def self.parse_multiple_response(buf_length, buffer)
+      res = {}
+      buffer.read(0, buf_length).split("\n").each do |vstr|
+        k, v = vstr.split("\t")
+        res[k] = v
+      end
+      res
+    end
+
+    private
+
+    def self.send_command(conn, offset, buffer)
+      begin
+        # Write size field.
+        size = (offset - 8) | (2 << 56) | (1 << 48)
+
+        buffer.write_int64(size, 0)
+        conn.write(buffer, offset)
+
+        # Read - reuse input buffer.
+        conn.read(buffer, 8)
+
+        size = buffer.read_int64(0)
+        length = size & 0xFFFFFFFFFFFF
+
+        buffer.resize(length)
+
+        conn.read(buffer, length)
+        return length
+      rescue => e
+        conn.close
+        raise e
+      end
+    end
+
+  end
+
+end

data/lib/aerospike/key.rb

@@ -0,0 +1,99 @@
+# 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 'digest'
+
+require 'aerospike/value/value'
+
+module Aerospike
+
+  class Key
+
+    @@digest_pool = Pool.new
+    @@digest_pool.create_block = Proc.new do
+      unless RUBY_PLATFORM == 'java'
+        Digest::RMD160.new
+      else
+        h = OpenSSL::Digest::RIPEMD160.new
+      end
+    end
+
+
+    attr_reader :namespace, :set_name, :digest
+
+    def initialize(ns, set, val, digest=nil)
+      @namespace = ns
+      @set_name = set
+      @user_key = Value.of(val)
+
+      unless digest
+        compute_digest
+      else
+        @digest = digest
+      end
+
+      self
+    end
+
+    def to_s
+      "#{@namespace}:#{@set_name}:#{@user_key}:#{@digest.nil? ? '' : @digest.bytes}"
+    end
+
+    def user_key
+      @user_key.get if @user_key
+    end
+
+    def user_key_as_value
+      @user_key
+    end
+
+    def ==(other)
+      other && other.is_a?(Key) &&
+        other.digest == @digest &&
+        other.namespace == @namespace
+    end
+    alias eql? ==
+
+    def hash
+      @digest.hash
+    end
+
+    private
+
+    def compute_digest
+      key_type = @user_key.type
+
+      if key_type == Aerospike::ParticleType::NULL
+        raise Aerospike::Exceptions::Aerospike.new(Aerospike::ResultCode::PARAMETER_ERROR, "Invalid key: nil")
+      end
+
+      # get a hash from pool and make it ready for work
+      h = @@digest_pool.poll
+      h.reset
+
+      # Compute a complete digest
+      h.update(@set_name)
+      h.update(key_type.chr)
+      h.update(@user_key.to_bytes)
+      @digest = h.digest
+
+      # put the hash object back to the pool
+      @@digest_pool.offer(h)
+    end
+
+  end
+
+end

data/lib/aerospike/language.rb

@@ -0,0 +1,25 @@
+# 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.
+
+module Aerospike
+
+  module Language
+
+    LUA = 'LUA'
+
+  end
+
+end

data/lib/aerospike/ldt/large.rb

@@ -0,0 +1,69 @@
+# 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/particle_type'
+
+module Aerospike
+
+  private
+
+  class Large
+
+    def initialize(client, policy, key, bin_name, user_module=nil)
+      @client = client
+      @policy = policy
+      @key = key
+      @bin_name = Aerospike::ParticleType::STRING.chr + bin_name
+      @user_module = Aerospike::ParticleType::STRING.chr + user_module unless user_module.nil?
+
+      self
+    end
+
+    # Delete bin containing the object.
+    def destroy
+      @client.execute_udf(@key, @PACKAGE_NAME, 'destroy', [@bin_name], @policy)
+    end
+
+    # Return size of object.
+    def size
+      @client.execute_udf(@key, @PACKAGE_NAME, 'size', [@bin_name], @policy)
+    end
+
+    # Return map of object configuration parameters.
+    def config
+      @client.execute_udf(@key, @PACKAGE_NAME, 'get_config', [@bin_name], @policy)
+    end
+
+    # Set maximum number of entries in the object.
+    #
+    # capacity      max entries in set
+    def capacity=(capacity)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'set_capacity', [@bin_name, capacity], @policy)
+    end
+
+    # Return maximum number of entries in the object.
+    def capacity
+      @client.execute_udf(@key, @PACKAGE_NAME, 'get_capacity', [@bin_name], @policy)
+    end
+
+    # Return list of all objects on the stack.
+    def scan
+      @client.execute_udf(@key, @PACKAGE_NAME, 'scan', [@bin_name], @policy)
+    end
+
+  end # class
+
+end #class