aerospike 0.1.0

data/lib/aerospike/ldt/large_list.rb

@@ -0,0 +1,100 @@
+# 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/ldt/large'
+
+module Aerospike
+
+  private
+
+  class LargeList < Large
+
+    def initialize(client, policy, key, bin_name, user_module=nil)
+      @PACKAGE_NAME = 'llist'
+
+      super(client, policy, key, bin_name, user_module)
+
+      self
+    end
+
+    # Add values to the list.  If the list does not exist, create it using specified user_module configuration.
+    #
+    # values      values to add
+    def add(*values)
+      if values.length == 1
+        @client.execute_udf(@key, @PACKAGE_NAME, 'add', [@bin_name, values[0], @user_module], @policy)
+      else
+        @client.execute_udf(@key, @PACKAGE_NAME, 'add_all', [@bin_name, values, @user_module], @policy)
+      end
+    end
+
+    # Update/Add each value in array depending if key exists or not.
+    # If value is a map, the key is identified by "key" entry.  Otherwise, the value is the key.
+    # If large list does not exist, create it using specified user_module configuration.
+    #
+    # values      values to update
+    def update(*values)
+      if values.length == 1
+        @client.execute_udf(@key, @PACKAGE_NAME, 'update', [@bin_name, values[0], @user_module], @policy)
+      else
+        @client.execute_udf(@key, @PACKAGE_NAME, 'update_all', [@bin_name, values, @user_module], @policy)
+      end
+    end
+
+    # Delete value from list.
+    #
+    # value       value to delete
+    def remove(value)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'remove', [@bin_name, value], @policy)
+    end
+
+    # Select values from list.
+    #
+    # value       value to select
+    # returns          list of entries selected
+    def find(value)
+      begin
+        @client.execute_udf(@key, @PACKAGE_NAME, 'find', [@bin_name, value], @policy)
+      rescue Aerospike::Exceptions::Aerospike => e
+        unless e.result_code == Aerospike::ResultCode::UDF_BAD_RESPONSE && e.message.index("Item Not Found")
+          raise e
+        end
+        nil
+      end
+    end
+
+    # Select values from list and apply specified Lua filter.
+    #
+    # value       value to select
+    # filter_name    Lua function name which applies filter to returned list
+    # filter_args    arguments to Lua function name
+    # returns          list of entries selected
+    def find_then_filter(value, filter_name, *filter_args)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'find_then_filter', [@bin_name, value, @user_module, filter_name, filter_args], @policy)
+    end
+
+    # Select values from list and apply specified Lua filter.
+    #
+    # filter_name    Lua function name which applies filter to returned list
+    # filter_args    arguments to Lua function name
+    # returns          list of entries selected
+    def filter(filter_name, *filter_args)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'filter', [@bin_name, @user_module, filter_name, filter_args], @policy)
+    end
+
+  end # class
+
+end #class

data/lib/aerospike/ldt/large_map.rb

@@ -0,0 +1,82 @@
+# 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/ldt/large'
+
+module Aerospike
+
+  private
+
+  class LargeMap < Large
+
+    def initialize(client, policy, key, bin_name, user_module=nil)
+      @PACKAGE_NAME = 'lmap'
+
+      super(client, policy, key, bin_name, user_module)
+
+      self
+    end
+
+    # Add entry to map.  If the map does not exist, create it using specified user_module configuration.
+    #
+    # name        entry key
+    # value       entry value
+    def put(name, value)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'put', [@bin_name, name, value, @user_module], @policy)
+    end
+
+    # Add map values to map.  If the map does not exist, create it using specified user_module configuration.
+    #
+    # map       map values to push
+    def put_map(the_map)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'put_all', [@bin_name, the_map, @user_module], @policy)
+    end
+
+    # Get value from map given name key.
+    #
+    # name        key.
+    # return          map of items selected
+    def get(name)
+      begin
+        @client.execute_udf(@key, @PACKAGE_NAME, 'get', [@bin_name, name, @user_module], @policy)
+      rescue Aerospike::Exceptions::Aerospike => e
+        unless e.result_code == Aerospike::ResultCode::UDF_BAD_RESPONSE && e.message.index("Item Not Found")
+          raise e
+        end
+        nil
+      end
+    end
+
+    # Get value from map given name key.
+    #
+    # name        key.
+    # return          map of items selected
+    def remove(name)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'remove', [@bin_name, name, @user_module], @policy)
+    end
+
+    # Select items from map.
+    #
+    # filter_name    Lua function name which applies filter to returned list
+    # filter_args    arguments to Lua function name
+    # return          list of items selected
+    def filter(filter_name, *filter_args)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'filter', [@bin_name, @user_module, filter_name, filter_args], @policy)
+    end
+
+  end # class
+
+end #class

data/lib/aerospike/ldt/large_set.rb

@@ -0,0 +1,78 @@
+# 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/ldt/large'
+
+module Aerospike
+
+  private
+
+  class LargeSet < Large
+
+    def initialize(client, policy, key, bin_name, user_module=nil)
+      @PACKAGE_NAME = 'lset'
+
+      super(client, policy, key, bin_name, user_module)
+
+      self
+    end
+
+    # Add values to the list.  If the list does not exist, create it using specified user_module configuration.
+    #
+    # values      values to add
+    def add(*values)
+      if values.length == 1
+        @client.execute_udf(@key, @PACKAGE_NAME, 'add', [@bin_name, values[0], @user_module], @policy)
+      else
+        @client.execute_udf(@key, @PACKAGE_NAME, 'add_all', [@bin_name, values, @user_module], @policy)
+      end
+    end
+
+    # Delete value from list.
+    #
+    # value       value to delete
+    def remove(value)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'remove', [@bin_name, value], @policy)
+    end
+
+    # Select values from list.
+    #
+    # value       value to select
+    # returns          list of entries selected
+    def get(value)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'get', [@bin_name, value], @policy)
+    end
+
+    # Check existence of value in the set.
+    #
+    # value       value to check
+    # returns          true if found, otherwise false
+    def exists(value)
+      1 == @client.execute_udf(@key, @PACKAGE_NAME, 'exists', [@bin_name, value], @policy)
+    end
+
+    # Select values from list and apply specified Lua filter.
+    #
+    # filter_name    Lua function name which applies filter to returned list
+    # filter_args    arguments to Lua function name
+    # returns          list of entries selected
+    def filter(filter_name, *filter_args)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'filter', [@bin_name, @user_module, filter_name, filter_args], @policy)
+    end
+
+  end # class
+
+end #class

data/lib/aerospike/ldt/large_stack.rb

@@ -0,0 +1,72 @@
+# 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/ldt/large'
+
+module Aerospike
+
+  private
+
+  class LargeStack < Large
+
+    def initialize(client, policy, key, bin_name, user_module=nil)
+      @PACKAGE_NAME = 'lstack'
+
+      super(client, policy, key, bin_name, user_module)
+
+      self
+    end
+
+    # Push values onto stack.  If the stack does not exist, create it using specified user_module configuration.
+    #
+    # values      values to push
+    def push(*values)
+      if values.length == 1
+        @client.execute_udf(@key, @PACKAGE_NAME, 'push', [@bin_name, values[0], @user_module], @policy)
+      else
+        @client.execute_udf(@key, @PACKAGE_NAME, 'push_all', [@bin_name, values, @user_module], @policy)
+      end
+    end
+
+    # Select items from top of stack.
+    #
+    # peek_count     number of items to select.
+    # returns          list of items selected
+    def peek(peek_count)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'peek', [@bin_name, peek_count], @policy)
+    end
+
+    # Select items from top of stack.
+    #
+    # peek_count     number of items to select.
+    # returns          list of items selected
+    def pop(count)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'pop', [@bin_name, count], @policy)
+    end
+
+    # Select items from top of stack.
+    #
+    # peek_count     number of items to select.
+    # filter_name    Lua function name which applies filter to returned list
+    # filter_args    arguments to Lua function name
+    # returns          list of items selected
+    def filter(peek_count, filter_name, *filter_args)
+      @client.execute_udf(@key, @PACKAGE_NAME, 'filter', [@bin_name, peek_count, @user_module, filter_name, filter_args], @policy)
+    end
+
+  end # class
+
+end #class

data/lib/aerospike/loggable.rb

@@ -0,0 +1,55 @@
+# encoding: utf-8
+module Aerospike
+
+  module Loggable
+
+    def self.log_operations(prefix, ops, runtime)
+      indent  = " "*prefix.length
+      if ops.length == 1
+        Aerospike.logger.debug([ prefix, ops.first.log_inspect, "runtime: #{runtime}" ].join(' '))
+      else
+        first, *middle, last = ops
+        Aerospike.logger.debug([ prefix, first.log_inspect ].join(' '))
+        middle.each { |m| Aerospike.logger.debug([ indent, m.log_inspect ].join(' ')) }
+        Aerospike.logger.debug([ indent, last.log_inspect, "runtime: #{runtime}" ].join(' '))
+      end
+    end
+
+    def self.debug(prefix, payload, runtime)
+      Aerospike.logger.debug([ prefix, payload, "runtime: #{runtime}" ].join(' '))
+    end
+
+    def self.warn(prefix, payload, runtime)
+      Aerospike.logger.warn([ prefix, payload, "runtime: #{runtime}" ].join(' '))
+    end
+
+    def self.info(prefix, payload, runtime)
+      Aerospike.logger.info([ prefix, payload, "runtime: #{runtime}" ].join(' '))
+    end
+
+    def self.error(prefix, payload, runtime)
+      Aerospike.logger.error([ prefix, payload, "runtime: #{runtime}" ].join(' '))
+    end
+
+    def logger
+      return @logger if defined?(@logger)
+      @logger = rails_logger || default_logger
+    end
+
+    def rails_logger
+      defined?(Rails) && Rails.respond_to?(:logger) && Rails.logger
+    end
+
+    def default_logger
+      logger = Logger.new(STDOUT)
+      logger.level = Logger::ERROR
+      logger
+    end
+
+    def logger=(logger)
+      @logger = logger
+    end
+
+  end # module
+
+end # module

data/lib/aerospike/operation.rb

@@ -0,0 +1,70 @@
+# 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'
+
+module Aerospike
+
+  class Operation
+
+    attr_reader :op_type, :bin_name, :bin_value
+
+    READ        = 1
+    READ_HEADER = 1
+    WRITE       = 2
+    ADD         = 5
+    APPEND      = 9
+    PREPEND     = 10
+    TOUCH       = 11
+
+    def initialize(op_type, bin_name=nil, bin_value=NullValue.new)
+      @op_type = op_type
+      @bin_name = bin_name
+      @bin_value = Value.of(bin_value)
+      self
+    end
+
+    def self.get(bin_name=nil)
+      Operation.new(READ, bin_name)
+    end
+
+    def self.get_header(bin_name=nil)
+      Operation.new(READ_HEADER, bin_name)
+    end
+
+    def self.put(bin)
+      Operation.new(WRITE, bin.name, bin.value)
+    end
+
+    def self.append(bin)
+      Operation.new(APPEND, bin.name, bin.value)
+    end
+
+    def self.prepend(bin)
+      Operation.new(PREPEND, bin.name, bin.value)
+    end
+
+    def self.add(bin)
+      Operation.new(ADD, bin.name, bin.value)
+    end
+
+    def self.touch
+      Operation.new(TOUCH)
+    end
+
+  end
+
+end # module