vinted-memcached 1.8.0

data/lib/memcached/auth.rb

@@ -0,0 +1,16 @@
+class Memcached
+  def destroy_credentials
+    if options[:credentials] != nil
+      check_return_code(Lib.memcached_destroy_sasl_auth_data(@struct))
+    end
+  end
+
+  def set_credentials
+    # If credentials aren't provided, try to get them from the environment
+    if options[:credentials] != nil
+      username, password = options[:credentials]
+      check_return_code(Lib.memcached_set_sasl_auth_data(@struct, username, password))
+    end
+  end
+end
+

data/lib/memcached/behaviors.rb

@@ -0,0 +1,78 @@
+
+class Memcached
+
+#:stopdoc:
+
+  def self.load_constants(prefix, hash = {})
+    Lib.constants.grep(/^#{prefix}/).each do |const_name|
+      hash[const_name[prefix.length..-1].downcase.to_sym] = Lib.const_get(const_name)
+    end
+    hash
+  end
+
+  BEHAVIORS = load_constants("MEMCACHED_BEHAVIOR_")
+
+  BEHAVIOR_VALUES = {
+    false => 0,
+    true => 1
+  }
+
+  HASH_VALUES = {}
+  BEHAVIOR_VALUES.merge!(load_constants("MEMCACHED_HASH_", HASH_VALUES))
+
+  DISTRIBUTION_VALUES = {}
+  BEHAVIOR_VALUES.merge!(load_constants("MEMCACHED_DISTRIBUTION_", DISTRIBUTION_VALUES))
+
+  DIRECT_VALUE_BEHAVIORS = [:retry_timeout, :connect_timeout, :rcv_timeout, :socket_recv_size, :poll_timeout, :socket_send_size, :server_failure_limit, :snd_timeout, :poll_max_retries]
+
+  CONVERSION_FACTORS = {
+    :rcv_timeout => 1_000_000,
+    :snd_timeout => 1_000_000,
+    :poll_timeout => 1_000,
+    :connect_timeout => 1_000
+  }
+
+#:startdoc:
+
+  private
+
+  # Set a behavior option for this Memcached instance. Accepts a Symbol <tt>behavior</tt> and either <tt>true</tt>, <tt>false</tt>, or a Symbol for <tt>value</tt>. Arguments are validated and converted into integers for the struct setter method.
+  def set_behavior(behavior, value) #:doc:
+    raise ArgumentError, "No behavior #{behavior.inspect}" unless b_id = BEHAVIORS[behavior]
+
+    # Scoped validations; annoying
+    msg =  "Invalid behavior value #{value.inspect} for #{behavior.inspect}"
+    case behavior
+      when :hash then raise(ArgumentError, msg) unless HASH_VALUES[value]
+      when :distribution then raise(ArgumentError, msg) unless DISTRIBUTION_VALUES[value]
+      when *DIRECT_VALUE_BEHAVIORS then raise(ArgumentError, msg) unless value.is_a?(Numeric) and value >= 0
+      else
+        raise(ArgumentError, msg) unless BEHAVIOR_VALUES[value]
+    end
+
+    lib_value = BEHAVIOR_VALUES[value] || (value * (CONVERSION_FACTORS[behavior] || 1)).to_i
+    #STDERR.puts "Setting #{behavior}:#{b_id} => #{value} (#{lib_value})"
+    Lib.memcached_behavior_set(@struct, b_id, lib_value)
+    #STDERR.puts " -> set to #{get_behavior(behavior).inspect}"
+  end
+
+  # Get a behavior value for this Memcached instance. Accepts a Symbol.
+  def get_behavior(behavior)
+    raise ArgumentError, "No behavior #{behavior.inspect}" unless b_id = BEHAVIORS[behavior]
+    value = Lib.memcached_behavior_get(@struct, b_id)
+
+    if BEHAVIOR_VALUES.invert.has_key?(value)
+      # False, nil are valid values so we can not rely on direct lookups
+      case behavior
+        # Scoped values; still annoying
+        when :hash then HASH_VALUES.invert[value]
+        when :distribution then DISTRIBUTION_VALUES.invert[value]
+        else
+          value
+      end
+    else
+      value
+    end
+  end
+
+end

data/lib/memcached/exceptions.rb

@@ -0,0 +1,84 @@
+
+class Memcached
+
+=begin rdoc
+
+Superclass for all Memcached runtime exceptions.
+
+Subclasses correspond one-to-one with server response strings or libmemcached errors. For example, raising <b>Memcached::NotFound</b> means that the server returned <tt>"NOT_FOUND\r\n"</tt>.
+
+== Subclasses
+
+* Memcached::ABadKeyWasProvidedOrCharactersOutOfRange
+* Memcached::AKeyLengthOfZeroWasProvided
+* Memcached::ATimeoutOccurred
+* Memcached::ActionNotSupported
+* Memcached::ActionQueued
+* Memcached::ClientError
+* Memcached::ConnectionBindFailure
+* Memcached::ConnectionDataDoesNotExist
+* Memcached::ConnectionDataExists
+* Memcached::ConnectionFailure
+* Memcached::ConnectionSocketCreateFailure
+* Memcached::CouldNotOpenUnixSocket
+* Memcached::EncounteredAnUnknownStatKey
+* Memcached::Failure
+* Memcached::FetchWasNotCompleted
+* Memcached::HostnameLookupFailure
+* Memcached::ItemValue
+* Memcached::MemoryAllocationFailure
+* Memcached::NoServersDefined
+* Memcached::NotFound
+* Memcached::NotStored
+* Memcached::PartialRead
+* Memcached::ProtocolError
+* Memcached::ReadFailure
+* Memcached::ServerDelete
+* Memcached::ServerEnd
+* Memcached::ServerError
+* Memcached::ServerIsMarkedDead
+* Memcached::ServerValue
+* Memcached::SomeErrorsWereReported
+* Memcached::StatValue
+* Memcached::SystemError
+* Memcached::TheHostTransportProtocolDoesNotMatchThatOfTheClient
+* Memcached::UnknownReadFailure
+* Memcached::WriteFailure
+
+=end
+  class Error < RuntimeError
+    attr_accessor :no_backtrace
+
+    def set_backtrace(*args)
+      @no_backtrace ? [] : super
+    end
+
+    def backtrace(*args)
+      @no_backtrace ? [] : super
+    end
+  end
+
+#:stopdoc:
+
+  class << self
+    private
+    def camelize(string)
+      string.downcase.gsub('/', ' or ').split(' ').map {|s| s.capitalize}.join
+    end
+  end
+
+  ERRNO_HASH = Hash[*Errno.constants.grep(/^E/).map{ |c| [Errno.const_get(c)::Errno, Errno.const_get(c).new.message] }.flatten]
+
+  EXCEPTIONS = []
+  empty_struct = Lib.memcached_create(nil)
+  Lib.memcached_create(empty_struct)
+
+  # Generate exception classes
+  Lib::MEMCACHED_MAXIMUM_RETURN.times do |index|
+    description = Lib.memcached_strerror(empty_struct, index).gsub("!", "")
+    exception_class = eval("class #{camelize(description)} < Error; self; end")
+    EXCEPTIONS << exception_class
+  end
+
+#:startdoc:
+end

data/lib/memcached/experimental.rb

@@ -0,0 +1,48 @@
+class Memcached
+  module Experimental
+
+    # TOUCH is used to set a new expiration time for an existing item
+    def touch(key, ttl=@default_ttl)
+      check_return_code(
+        Lib.memcached_touch(@struct, key, ttl),
+        key
+      )
+    rescue => e
+      tries ||= 0
+      raise unless tries < options[:exception_retry_limit] && should_retry(e)
+      tries += 1
+      retry
+    end
+
+    def get_len(bytes, keys)
+      if keys.is_a? ::Array
+        # Multi get
+        ret = Lib.memcached_mget_len(@struct, keys, bytes);
+        check_return_code(ret, keys)
+
+        hash = {}
+        keys.each do
+          value, key, flags, ret = Lib.memcached_fetch_rvalue(@struct)
+          break if ret == Lib::MEMCACHED_END
+          if ret != Lib::MEMCACHED_NOTFOUND
+            check_return_code(ret, key)
+            # Assign the value
+            hash[key] = value
+          end
+        end
+        hash
+      else
+        # Single get_len
+        value, flags, ret = Lib.memcached_get_len_rvalue(@struct, keys, bytes)
+        check_return_code(ret, keys)
+        value
+      end
+    rescue => e
+      tries ||= 0
+      raise unless tries < options[:exception_retry_limit] && should_retry(e)
+      tries += 1
+      retry
+    end
+
+  end
+end

data/lib/memcached/marshal_codec.rb

@@ -0,0 +1,10 @@
+class Memcached
+  module MarshalCodec
+      def self.encode(key, value, flags)
+        [Marshal.dump(value), flags]
+      end
+      def self.decode(key, value, flags)
+        Marshal.load(value)
+      end
+  end
+end