ruby-redis 0.0.1

data/lib/redis/strings.rb

@@ -0,0 +1,144 @@
+require File.expand_path '../redis', File.dirname(__FILE__)
+
+class Redis
+  
+  # Note: Redis does not have a dedicated integer type.
+  # Integer operations will convert values to integers.
+  
+  module Strings
+    
+    def redis_GET key
+      value = @database[key]
+      redis_t NilClass, String, Numeric, value
+      value
+    end
+    
+    def redis_APPEND key, value
+      new_value = redis_GET(key).to_s + value
+      @database[key] = new_value
+      new_value.size
+    end
+    
+    def redis_SETRANGE key, offset, value
+      data = (redis_GET(key)||'').to_s
+      return data.size if value.empty?
+      offset = redis_pos_i offset, 'out of range'
+      raise 'maximum allowed size' if offset + value.size > 512*1024*1024
+      if data.size <= offset
+        data.concat 0.chr * (offset - data.size + value.size)
+      end
+      data[offset,value.size] = value
+      @database[key] = data
+      data.size
+    end
+    
+    def redis_GETRANGE key, first, last
+      first = redis_i first
+      last = redis_i last
+      value = redis_GET(key) || ''
+      first = 0 if first < -value.size
+      value[first..last]
+    end
+    
+    def redis_GETBIT key, offset
+      data = redis_GET key
+      return 0 unless data
+      data = data.to_s
+      offset = redis_pos_i offset
+      byte = offset / 8
+      bit = 0x80 >> offset % 8
+      return 0 if data.size <= byte
+      original_byte = data[byte].ord
+      original_bit = original_byte & bit
+      original_bit != 0
+    end
+    
+    def redis_SETBIT key, offset, value
+      data = (redis_GET(key)||'').to_s
+      offset = redis_pos_i offset, 'out of range'
+      raise 'out of range' if offset >= 4*1024*1024*1024
+      byte = offset / 8
+      bit = 0x80 >> offset % 8
+      if data.size <= byte
+        data += 0.chr * (byte - data.size + 1)
+      end
+      original_byte = data[byte].ord
+      original_bit = original_byte & bit
+      if value == '0'
+        data[byte] = (original_byte & ~bit).chr
+      elsif value == '1'
+        data[byte] = (original_byte | bit).chr
+      else
+        raise 'out of range'
+      end
+      @database[key] = data
+      original_bit != 0
+    end
+  
+    def redis_STRLEN key
+      value = redis_GET key
+      value ? value.size : 0
+    end
+
+    def redis_MGET *keys
+      keys.collect do |key|
+        value = @database[key]
+        (String === value) ? value : nil
+      end
+    end
+    
+    def redis_GETSET key, value
+      old = redis_GET key
+      @database[key] = value
+      old
+    end
+    
+    def redis_SET key, value
+      @database[key] = value
+      Response::OK
+    end
+
+    def redis_SETEX key, seconds, value
+      @database[key] = value
+      @database.expire key, redis_pos_i(seconds, 'invalid expire time in SETEX')
+      Response::OK
+    end
+    
+    def redis_MSET *args
+      Hash[*args].each do |key, value|
+        @database[key] = value
+      end
+      Response::OK
+    end
+    
+    def redis_MSETNX *args
+      hash = Hash[*args]
+      hash.each do |key, value|
+        return false if @database.has_key? key
+      end
+      hash.each do |key, value|
+        @database[key] = value
+      end
+      true
+    end
+
+    def redis_SETNX key, value
+      return false if @database.has_key? key
+      @database[key] = value
+      true
+    end
+    
+    def redis_INCR key
+      @database[key] = redis_i(@database[key] || 0) + 1
+    end
+
+    def redis_INCRBY key, value
+      @database[key] = redis_i(@database[key] || 0) + redis_i(value)
+    end
+
+    def redis_DECRBY key, value
+      @database[key] = redis_i(@database[key] || 0) - redis_i(value)
+    end
+      
+  end
+end