aerospike 0.1.0

data/lib/aerospike/value/particle_type.rb

@@ -0,0 +1,45 @@
+# 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
+
+  module ParticleType
+
+    # Server particle types. Unsupported types are commented out.
+    NULL = 0
+    INTEGER = 1
+    #BIGNUM = 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
+    #LUA_BLOB = 18
+    MAP = 19
+    LIST = 20
+
+  end # module
+
+end # module

data/lib/aerospike/value/value.rb

@@ -0,0 +1,380 @@
+# 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 'msgpack'
+
+require 'aerospike/utils/pool'
+
+require 'aerospike/aerospike_exception'
+
+module Aerospike
+
+  private
+
+  # Polymorphic value classes used to efficiently serialize objects into the wire protocol.
+  class Value
+
+    @@packer_pool = Pool.new
+    @@packer_pool.create_block = Proc.new { MessagePack::Packer.new }
+
+    def self.get_packer
+      res = @@packer_pool.poll
+      res.clear
+      res
+    end
+
+    def self.put_packer(packer)
+      @@packer_pool.offer(packer)
+    end
+
+    def self.of(value)
+      case value
+      when nil
+        res = NullValue.new
+      when Integer
+        if value < 2**63
+          res = IntegerValue.new(value)
+        else
+          # big nums > 2**63 are not supported
+          raise Aerospike::Exceptions::Aerospike.new(TYPE_NOT_SUPPORTED, "Value type #{value.class} not supported.")
+        end
+      when String
+        res = StringValue.new(value)
+      when Value
+        res = value
+      when Hash
+        res = MapValue.new(value)
+      when Array
+        res = ListValue.new(value)
+      else
+        # throw an exception for anything that is not supported.
+        raise Aerospike::Exceptions::Aerospike.new(TYPE_NOT_SUPPORTED, "Value type #{value.class} not supported.")
+      end
+
+      res
+    end
+
+  end # Value
+
+
+  # Empty value.
+  class NullValue < Value
+
+    def initialize
+      self
+    end
+
+    def type
+      Aerospike::ParticleType::NULL
+    end
+
+    def get
+      nil
+    end
+
+    def to_s
+      ''
+    end
+
+
+    def estimate_size
+      0
+    end
+
+    def write(buffer, offset)
+      0
+    end
+
+    def pack(packer)
+      packer.write_nil
+    end
+
+    def to_bytes
+      ''
+    end
+  end
+
+  # Byte array value.
+  class BytesValue < Value
+
+    def initialize(value)
+      @bytes = value
+      @bytes.force_encoding('binary')
+
+      self
+    end
+
+    def type
+      Aerospike::ParticleType::BLOB
+    end
+
+    def get
+      @bytes
+    end
+
+    def to_s
+      @bytes.to_s
+    end
+
+    def to_bytes
+      @bytes.bytes
+    end
+
+    def estimate_size
+      @bytes.bytesize
+    end
+
+    def write(buffer, offset)
+      buffer.write_binary(bytes, offset)
+      buffer.length
+    end
+
+    def pack(packer)
+      packer.write(@bytes.bytes)
+    end
+
+  end # BytesValue
+
+  #######################################
+
+  # value string.
+  class StringValue < Value
+
+    def initialize(val)
+      @value = val || ''
+      self
+    end
+
+    def estimate_size
+      @value.bytesize
+    end
+
+    def write(buffer, offset)
+      buffer.write_binary(@value, offset)
+    end
+
+    def pack(packer)
+      packer.write(@value)
+    end
+
+    def type
+      Aerospike::ParticleType::STRING
+    end
+
+    def get
+      @value
+    end
+
+    def to_bytes
+      @value
+    end
+
+    def to_s
+      @value
+    end
+
+  end # StringValue
+
+  #######################################
+
+  # Integer value.
+  class IntegerValue < Value
+
+    def initialize(val)
+      @value = val || 0
+      self
+    end
+
+    def estimate_size
+      8
+    end
+
+    def write(buffer, offset)
+      buffer.write_int64(@value, offset)
+      8
+    end
+
+    def pack(packer)
+      packer.write(@value)
+    end
+
+    def type
+      Aerospike::ParticleType::INTEGER
+    end
+
+    def get
+      @value
+    end
+
+    def to_bytes
+      [@value].pack('Q<')
+    end
+
+    def to_s
+      @value.to_s
+    end
+
+  end # IntegerValue
+
+  # List value.
+  # Supported by Aerospike 3 servers only.
+  class ListValue < Value
+
+    def initialize(list)
+      @list = list || nil
+      packer = Value.get_packer
+      pack(packer)
+      @bytes = packer.to_s.force_encoding('binary')
+      Value.put_packer(packer)
+
+      self
+    end
+
+    def estimate_size
+      @bytes.bytesize
+    end
+
+    def write(buffer, offset)
+      buffer.write_binary(@bytes, offset)
+      @bytes.bytesize
+    end
+
+    def pack(packer)
+      packer.write_array_header(@list.length)
+      # @list.each do |val|
+      for val in @list
+        Value.of(val).pack(packer)
+      end
+    end
+
+    def type
+      Aerospike::ParticleType::LIST
+    end
+
+    def get
+      @list
+    end
+
+    def to_bytes
+      @bytes
+    end
+
+    def to_s
+      @list.map{|v| v.to_s}.to_s
+    end
+
+  end
+
+  # #######################################/
+
+  # Map value.
+  # Supported by Aerospike 3 servers only.
+  class MapValue < Value
+
+    def initialize(vmap)
+      @vmap = vmap || {}
+
+      packer = Value.get_packer
+      pack(packer)
+      @bytes = packer.to_s.force_encoding('binary')
+      Value.put_packer(packer)
+
+      self
+    end
+
+    def estimate_size
+      @bytes.bytesize
+    end
+
+    def write(buffer, offset)
+      buffer.write_binary(@bytes, offset)
+      @bytes.bytesize
+    end
+
+    def pack(packer)
+      packer.write_map_header(@vmap.length)
+      # @vmap.each do |key, val|
+      for key, val in @vmap
+        Value.of(key).pack(packer)
+        Value.of(val).pack(packer)
+      end
+    end
+
+    def type
+      Aerospike::ParticleType::MAP
+    end
+
+    def get
+      @vmap
+    end
+
+    def to_bytes
+      @bytes
+    end
+
+    def to_s
+      @vmap.map{|k, v| "#{k.to_s} => #{v.to_s}" }.to_s
+    end
+
+  end
+
+  #######################################
+
+  protected
+
+  def self.bytes_to_particle(type, buf , offset, length)
+
+    case type
+    when Aerospike::ParticleType::STRING
+      # StringValue.new(buf.read(offset, length))
+      buf.read(offset, length)
+
+    when Aerospike::ParticleType::INTEGER
+      # IntegerValue.new(buf.read_int64(offset))
+      buf.read_int64(offset)
+
+    when Aerospike::ParticleType::BLOB
+      buf.read(offse,length)
+
+    when Aerospike::ParticleType::LIST
+      MessagePack.unpack(buf.read(offset, length))
+
+    when Aerospike::ParticleType::MAP
+      MessagePack.unpack(buf.read(offset, length))
+
+    else
+      nil
+    end
+  end
+
+  def self.bytes_to_key_value(type, buf, offset, len)
+
+    case type
+    when Aerospike::ParticleType::STRING
+      StringValue.new(buf.read(offset, len))
+
+    when Aerospike::ParticleType::INTEGER
+      IntegerValue.new(buf.read_var_int64(offset, len))
+
+    when Aerospike::ParticleType::BLOB
+      BytesValue.new(buf.read(offset,len))
+
+    else
+      nil
+    end
+  end
+
+end # module

data/lib/aerospike/version.rb

@@ -0,0 +1,4 @@
+# encoding: utf-8
+module Aerospike
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,132 @@
+--- !ruby/object:Gem::Specification
+name: aerospike
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Khosrow Afroozeh
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-10-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: atomic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: msgpack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+description: Access your Aerospike cluster with ease of ruby.
+email:
+- khosrow@aerospike.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE
+- README.md
+- lib/aerospike.rb
+- lib/aerospike/aerospike_exception.rb
+- lib/aerospike/bin.rb
+- lib/aerospike/client.rb
+- lib/aerospike/cluster/cluster.rb
+- lib/aerospike/cluster/connection.rb
+- lib/aerospike/cluster/node.rb
+- lib/aerospike/cluster/node_validator.rb
+- lib/aerospike/cluster/partition.rb
+- lib/aerospike/cluster/partition_tokenizer_new.rb
+- lib/aerospike/cluster/partition_tokenizer_old.rb
+- lib/aerospike/command/batch_command.rb
+- lib/aerospike/command/batch_command_exists.rb
+- lib/aerospike/command/batch_command_get.rb
+- lib/aerospike/command/batch_item.rb
+- lib/aerospike/command/batch_node.rb
+- lib/aerospike/command/command.rb
+- lib/aerospike/command/delete_command.rb
+- lib/aerospike/command/execute_command.rb
+- lib/aerospike/command/exists_command.rb
+- lib/aerospike/command/field_type.rb
+- lib/aerospike/command/operate_command.rb
+- lib/aerospike/command/read_command.rb
+- lib/aerospike/command/read_header_command.rb
+- lib/aerospike/command/single_command.rb
+- lib/aerospike/command/touch_command.rb
+- lib/aerospike/command/write_command.rb
+- lib/aerospike/host.rb
+- lib/aerospike/info.rb
+- lib/aerospike/key.rb
+- lib/aerospike/language.rb
+- lib/aerospike/ldt/large.rb
+- lib/aerospike/ldt/large_list.rb
+- lib/aerospike/ldt/large_map.rb
+- lib/aerospike/ldt/large_set.rb
+- lib/aerospike/ldt/large_stack.rb
+- lib/aerospike/loggable.rb
+- lib/aerospike/operation.rb
+- lib/aerospike/policy/client_policy.rb
+- lib/aerospike/policy/generation_policy.rb
+- lib/aerospike/policy/policy.rb
+- lib/aerospike/policy/priority.rb
+- lib/aerospike/policy/record_exists_action.rb
+- lib/aerospike/policy/write_policy.rb
+- lib/aerospike/record.rb
+- lib/aerospike/result_code.rb
+- lib/aerospike/task/index_task.rb
+- lib/aerospike/task/task.rb
+- lib/aerospike/task/udf_register_task.rb
+- lib/aerospike/task/udf_remove_task.rb
+- lib/aerospike/udf.rb
+- lib/aerospike/utils/buffer.rb
+- lib/aerospike/utils/epoc.rb
+- lib/aerospike/utils/pool.rb
+- lib/aerospike/value/particle_type.rb
+- lib/aerospike/value/value.rb
+- lib/aerospike/version.rb
+homepage: http://www.aerospike.com
+licenses:
+- Apache2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: An Aerospike driver for Ruby.
+test_files: []