ffi-hiredis_vip 0.1.0.pre1

data/lib/ffi/hiredis_vip/exists.rb

@@ -0,0 +1,33 @@
+module FFI
+  module HiredisVip
+    class Exists
+      def initialize(client)
+        @client = client
+      end
+
+      def exists(*keys)
+        reply = nil
+        keys = keys.flatten
+        number_of_exists = keys.size
+        command = "EXISTS#{' %b' * number_of_exists}"
+        command_args = []
+        keys.each do |key|
+          command_args << :string << key << :size_t << key.size
+        end
+
+        @client.synchronize do |connection|
+          reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+        end
+
+        return nil if reply.nil? || reply.null?
+
+        case reply[:type] 
+        when :REDIS_REPLY_INTEGER
+          reply[:integer]
+        else
+          0
+        end
+      end
+    end # class Exists
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/exists_before_3.rb

@@ -0,0 +1,32 @@
+module FFI
+  module HiredisVip
+    class ExistsBefore3
+      def initialize(client)
+        @client = client
+      end
+
+      def exists(*keys)
+        keys = keys.flatten
+        number_of_exists = 0
+        command = "EXISTS %b"
+
+        keys.each do |key|
+          reply = nil
+          command_args = [ :string, key, :size_t, key.size ]
+          @client.synchronize do |connection|
+            reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+          end
+
+          next if reply.nil? || reply.null?
+
+          case reply[:type] 
+          when :REDIS_REPLY_INTEGER
+            number_of_exists = number_of_exists + reply[:integer]
+          end
+        end
+
+        number_of_exists
+      end
+    end # class ExistsBefore3
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/info.rb

@@ -0,0 +1,32 @@
+module FFI
+  module HiredisVip
+    class Info
+      attr_reader :original_info, :info_hash
+
+      def initialize(info_string)
+        @info_hash = {}
+        @original_info = info_string
+        process_original_info
+      end
+      
+      def [](key)
+        @info_hash[key]
+      end
+
+      def method_missing(method_name, *args, &block)
+        @info_hash[method_name] || super
+      end
+
+      private
+
+      def process_original_info
+        original_info.each_line do |info_line|
+          next unless info_line.include?(":")
+
+          parts = info_line.split(":")
+          @info_hash[parts.shift.strip] = parts.join(":").strip
+        end
+      end
+    end # Info
+  end # HiredisVip
+end # FFI

data/lib/ffi/hiredis_vip/keys.rb

@@ -0,0 +1,54 @@
+module FFI
+  module HiredisVip
+    class Keys
+      def initialize(client)
+        @client = client
+      end
+
+      def keys(pattern)
+        reply = nil
+        pattern = "#{pattern}"
+        command = "KEYS %b"
+        command_args = [ :string, pattern, :size_t, pattern.size ]
+
+        synchronize do |connection|
+          reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+        end
+
+        return nil if reply.nil? || reply.null?
+
+        # TODO: more error checking here? what is correct response on nothing?
+        case reply[:type]
+        when :REDIS_REPLY_ARRAY
+          keys_results_to_array(reply)
+        else
+          []
+        end
+      end
+
+      private
+
+      def keys_results_to_array(array_reply)
+        keys_results = []
+
+        0.upto(array_reply[:elements] - 1) do |element_number|
+          result = ::FFI::HiredisVip::Core.redisReplyElement(array_reply, element_number)
+
+          case result[:type]
+          when :REDIS_REPLY_STRING
+            keys_results << result[:str]
+          end
+        end
+
+        keys_results
+      end
+
+      def synchronize
+        @client.synchronize do |connection|
+          yield(connection)
+        end
+      end
+
+    end # class Keys
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/mget.rb

@@ -0,0 +1,60 @@
+module FFI
+  module HiredisVip
+    class Mget
+      def initialize(client)
+        @client = client
+      end
+
+      def mget(*keys)
+        reply = nil
+        keys = keys.flatten
+        command = "MGET"
+        command_args = []
+        keys.each do |key|
+          command << " %b"
+          command_args << :string << key << :size_t << key.size
+        end
+
+        synchronize do |connection|
+          reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+        end
+
+        return nil if reply.nil? || reply.null?
+
+        # TODO: more error checking here? what is correct response on nothing?
+        case reply[:type]
+        when :REDIS_REPLY_ARRAY
+          mget_results_to_array(reply)
+        else
+          []
+        end
+      end
+
+      private
+
+      def mget_results_to_array(array_reply)
+        mget_results = []
+
+        0.upto(array_reply[:elements] - 1) do |element_number|
+          result = ::FFI::HiredisVip::Core.redisReplyElement(array_reply, element_number)
+
+          case result[:type]
+          when :REDIS_REPLY_STRING
+            mget_results << result[:str]
+          when :REDIS_REPLY_NIL
+            mget_results << nil
+          end
+        end
+
+        mget_results
+      end
+
+      def synchronize
+        @client.synchronize do |connection|
+          yield(connection)
+        end
+      end
+
+    end # class Mget
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/persist.rb

@@ -0,0 +1,37 @@
+module FFI
+  module HiredisVip
+    class Persist
+      def initialize(client)
+        @client = client
+      end
+
+      def persist(key)
+        reply = nil
+        command = "PERSIST %b"
+        command_args = [ :string, key, :size_t, key.size ]
+        synchronize do |connection|
+          reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+        end
+
+        return nil if reply.nil? || reply.null?
+
+        case reply[:type]
+        when :REDIS_REPLY_INTEGER
+          reply[:integer]
+        end
+      end
+
+      def supports_persist?
+        true
+      end
+
+      private
+
+      def synchronize
+        @client.synchronize do |connection|
+          yield(connection)
+        end
+      end
+    end # class Persist
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/persist_before_2_2.rb

@@ -0,0 +1,23 @@
+module FFI
+  module HiredisVip
+    class PersistBefore22
+      def initialize(client)
+        @client = client
+      end
+
+      def persist(key)
+        raise <<-SCAN_ERROR
+          PERSIST Command in Redis is only available on Servers >= 2.2.0
+          The Redis Server you are connecting to is using a version that is not supported.
+
+          == > INFO
+            #{@client.info}
+        SCAN_ERROR
+      end
+
+      def supports_persist?
+        false
+      end
+    end # class PersistBefore22
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/sadd.rb

@@ -0,0 +1,42 @@
+module FFI
+  module HiredisVip
+    class Sadd
+      def initialize(client)
+        @client = client
+      end
+
+      def sadd(key, *values)
+        reply = nil
+        values = values.flatten
+        number_of_values = values.size
+        command = "SADD %b#{' %b' * number_of_values}"
+        command_args = [ :string, key, :size_t, key.size ]
+        values.each do |value|
+          command_args << :string << value << :size_t << value.size
+        end
+
+        synchronize do |connection|
+          reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+        end
+
+        return nil if reply.nil? || reply.null?
+
+        case reply[:type] 
+        when :REDIS_REPLY_INTEGER
+          reply[:integer]
+        else
+          0
+        end
+      end
+
+      private
+
+      def synchronize
+        @client.synchronize do |connection|
+          yield(connection)
+        end
+      end
+
+    end # class Sadd
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/sadd_before_2_4.rb

@@ -0,0 +1,40 @@
+module FFI
+  module HiredisVip
+    class SaddBefore24
+      def initialize(client)
+        @client = client
+      end
+
+      def sadd(key, *values)
+        values = values.flatten
+        number_added_to_set = 0
+        command = "SADD %b %b"
+
+        values.each do |value|
+          command_args = [ :string, key, :size_t, key.size, :string, value, :size_t, value.size ]
+          synchronize do |connection|
+            reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+
+            next if reply.nil? || reply.null?
+
+            case reply[:type] 
+            when :REDIS_REPLY_INTEGER
+              number_added_to_set = number_added_to_set + reply[:integer]
+            end
+          end
+        end
+
+        number_added_to_set
+      end
+
+      private
+
+      def synchronize
+        @client.synchronize do |connection|
+          yield(connection)
+        end
+      end
+
+    end # class SaddBefore24
+  end # module HiredisVip
+end # module FFI

data/lib/ffi/hiredis_vip/scan.rb

@@ -0,0 +1,78 @@
+module FFI
+  module HiredisVip
+    class Scan
+      def initialize(client)
+        @client = client
+      end
+
+      def scan(cursor, options = {})
+        reply = nil
+        command = "SCAN %b"
+        command_args = [ :string, cursor, :size_t, cursor.size ]
+
+        if options[:match]
+          matcher = "#{options[:match]}"
+          command << " MATCH %b"
+          command_args << :string << matcher << :size_t << matcher.size
+        end
+
+        if options[:count]
+          count = "#{options[:count]}"
+          command << " COUNT %b"
+          command_args << :string << count << :size_t << count.size
+        end
+
+        synchronize do |connection|
+          reply = ::FFI::HiredisVip::Core.command(connection, command, *command_args)
+        end
+
+        return nil if reply.nil?
+
+        # TODO: more error checking here?
+        case reply[:type]
+        when :REDIS_REPLY_ARRAY
+          [ scan_results_cursor(reply), scan_results_to_array(reply) ]
+        end
+      end
+
+      def supports_scan?
+        true
+      end
+
+      private
+
+      def scan_results_cursor(reply)
+        zeroth_result = ::FFI::HiredisVip::Core.redisReplyElement(reply, 0)
+
+        if !zeroth_result.null? && zeroth_result[:type] == :REDIS_REPLY_STRING
+          zeroth_result[:str]
+        else
+          raise "probs" # TODO: what do we do here
+        end
+      end
+
+      def scan_results_to_array(reply)
+        scan_results = []
+        array_reply = ::FFI::HiredisVip::Core.redisReplyElement(reply, 1)
+
+        if !array_reply.null? && array_reply[:type] == :REDIS_REPLY_ARRAY
+          0.upto(array_reply[:elements] - 1) do |element_number|
+            result = ::FFI::HiredisVip::Core.redisReplyElement(array_reply, element_number)
+            scan_results << result[:str] if result[:type] == :REDIS_REPLY_STRING
+          end
+
+          scan_results
+        else
+          raise "probs" # TODO: what do we do here
+        end
+      end
+
+      def synchronize
+        @client.synchronize do |connection|
+          yield(connection)
+        end
+      end
+
+    end # class Scan
+  end # module HiredisVip
+end # module FFI