ruby-redis-portertech 0.0.3

data/lib/redis/version.rb

@@ -0,0 +1,5 @@
+module Redis
+
+  VERSION = '0.0.3'
+
+end

data/lib/redis/zsets.rb

@@ -0,0 +1,279 @@
+module Redis
+  
+  #WARN Time complexity may not match C version in this module yet.
+  #TODO Should add and delete manipulate @keys instead of clearing?
+    
+  module ZSets
+    
+    class ZSet
+      include Enumerable
+
+      def initialize
+        @hash = Hash.new
+  	    @keys = nil
+  	    @keys_reverse = nil
+  	  end
+
+  	  def add(o, s = 0.0)
+        @hash[o] = s
+  	    @keys = nil
+  	    @keys_reverse = nil
+        self
+      end
+
+  	  def delete(o)
+  	    @keys = nil
+  	    @keys_reverse = nil
+  	    @hash.delete(o)
+  	    self
+  	  end
+
+  	  def include?(o)
+        @hash.include?(o)
+      end
+
+  	  def score(o)
+  	    @hash[o]
+      end
+
+      def size
+        @hash.size
+      end
+
+      def empty?
+        @hash.empty?
+      end
+
+  	  def each
+  	    block_given? or return enum_for(__method__)
+  	    to_a.each { |o| yield(o) }
+  	    self
+  	  end
+
+  	  def to_a
+  	    unless @keys
+    	    (@keys = @hash.to_a).sort! do |a, b|
+    	      a.reverse <=> b.reverse
+  	      end
+  	    end
+  	    @keys
+  	  end
+
+  	  def to_a_reverse
+  	    unless @keys_reverse
+  	      @keys_reverse = to_a.reverse
+        end
+  	    @keys_reverse
+      end
+
+      def range reverse, start ,stop, conversions, withscores = false
+        start = conversions.redis_i start
+        stop = conversions.redis_i stop
+        array = reverse ? to_a_reverse : to_a
+        start = 0 if start < -size
+        return array[start..stop].flatten(1) if withscores
+        (array[start..stop]||[]).collect{|i|i.first}
+      end
+
+      def range_by_score reverse, min, max, conversions, *args
+        withscores = offset = count = nil
+        until args.empty?
+          case args.shift.upcase
+          when 'LIMIT'
+            offset = args.shift.to_i
+            count = args.shift.to_i
+          when 'WITHSCORES'
+            withscores = true
+          else
+            raise 'bad arguments'
+          end
+        end
+        result = []
+        min_exclusive = false
+        if min[0..0] == '('
+          min_exclusive = true
+          min = min[1..-1]
+        end
+        min = conversions.redis_f min
+        max_exclusive = false
+        if max[0..0] == '('
+          max_exclusive = true
+          max = max[1..-1]
+        end
+        max = conversions.redis_f max
+        if reverse
+          x = min; min = max; max = x
+        end
+        (reverse ? to_a_reverse : to_a).each do |member, score|
+          next if min > score or (min_exclusive and min >= score)
+          next if max < score or (max_exclusive and max <= score)
+          if offset
+            offset -= 1
+            next unless offset < 0
+            offset = nil
+          end
+          result << member
+          result << score if withscores
+          if count
+            count -= 1
+            break if count == 0
+          end
+        end
+        result
+      end
+
+      def self.aggregate database, is_and, destination, numkeys, conversions, *args
+        numkeys = numkeys.to_i
+        aggregate = 'SUM'
+        keys = []
+        keys << args.shift while (numkeys -= 1) >= 0
+        weights = Array.new keys.size, 1
+        until args.empty?
+          case args.shift.upcase
+          when 'WEIGHTS'
+            weights = []
+            keys.size.times {weights << conversions.redis_f(args.shift, 'weight value is not a double')}
+          when 'AGGREGATE'
+            aggregate = args.shift.upcase
+          else
+            raise 'bad arguments'
+          end
+        end
+        results = []
+        keys.zip(weights) do |key, weight|
+          inner_result = ZSet.new
+          record = database[key] || ZSet.new
+          record.each do |member, score|
+            inner_result.add member, (score||1) * weight
+          end
+          results << inner_result
+        end
+        result = results.reduce do |memo, result|
+          n = is_and ? new : memo
+          result.each do |member, score|
+            next if is_and and !memo.include?(member)
+            test = [score, memo.score(member)].compact
+            text << 0 if test.empty?
+            case aggregate
+            when 'SUM'
+              score = test.reduce :+
+            when 'MIN'
+              score = test.min
+            when 'MAX'
+              score = test.max
+            else
+              raise 'bad arguments'
+            end
+            n.add member, score 
+          end
+          n
+        end
+        database[destination] = result unless result.empty?
+        result.size
+      end
+
+    end
+
+    def redis_ZADD key, score, member
+      record = (@database[key] ||= ZSet.new)
+      result = !record.include?(member)
+      record.add member, redis_f(score)
+      result
+    end
+    
+    def redis_ZINCRBY key, increment, member
+      record = (@database[key] ||= ZSet.new)
+      increment = redis_f increment
+      if record.include?(member)
+        score = record.score(member) + increment
+      else
+        score = increment
+      end
+      raise 'NaN' if score.nan?
+      record.add member, score
+      score
+    end
+    
+    def redis_ZRANK key, member
+      record = (@database[key] || ZSet.new).to_a
+      record.index {|i| i[0]==member}
+    end
+    
+    def redis_ZREM key, member
+      record = @database[key] || []
+      return false unless record.include? member
+      record.delete member
+      @database.delete key if record.empty?
+      return true
+    end
+    
+    def redis_ZSCORE key, member
+      (@database[key] || ZSet.new).score member
+    end
+    
+    def redis_ZREVRANK key, member
+      record = (@database[key] || ZSet.new).to_a_reverse
+      record.index {|i| i[0]==member}
+    end
+    
+    def redis_ZCARD key
+      (@database[key] || []).size
+    end
+    
+    def redis_ZREMRANGEBYSCORE key, min, max
+      record = @database[key] || ZSet.new
+      range = record.range_by_score(false, min, max, self)
+      range.each do |member, score|
+        record.delete member
+      end
+      range.size
+    end
+    
+    def redis_ZCOUNT key, min, max
+      record = @database[key] || ZSet.new
+      record.range_by_score(false, min, max, self).size
+    end
+
+    def redis_ZREVRANGEBYSCORE key, min, max, *args
+      record = @database[key] || ZSet.new
+      record.range_by_score true, min, max, self, *args
+    end
+    
+    def redis_ZRANGEBYSCORE key, min, max, *args
+      record = @database[key] || ZSet.new
+      record.range_by_score false, min, max, self, *args
+    end
+  
+    def redis_ZRANGE key, start ,stop, withscores = false
+      record = @database[key] || ZSet.new
+      start = redis_i start
+      stop = redis_i stop
+      record.range false, start, stop, self, withscores
+    end
+  
+    def redis_ZREVRANGE key, start ,stop, withscores = false
+      record = @database[key] || ZSet.new
+      start = redis_i start
+      stop = redis_i stop
+      record.range true, start, stop, self, withscores
+    end
+
+    def redis_ZREMRANGEBYRANK key, start, stop
+      record = @database[key] || ZSet.new
+      range = record.range false, start, stop, self
+      range.each do |member, score|
+        record.delete member
+      end
+      range.size
+    end
+    
+    def redis_ZUNIONSTORE destination, numkeys, *args
+      ZSet.aggregate @database, false, destination, numkeys, self, *args      
+    end
+
+    def redis_ZINTERSTORE destination, numkeys, *args
+      ZSet.aggregate @database, true, destination, numkeys, self, *args      
+    end
+    
+  end
+end

metadata

@@ -0,0 +1,120 @@
+--- !ruby/object:Gem::Specification 
+name: ruby-redis-portertech
+version: !ruby/object:Gem::Version 
+  hash: 25
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 3
+  version: 0.0.3
+platform: ruby
+authors: 
+- David Turnbull
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-11-21 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: minitest
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+description: 
+email: 
+- dturnbull@gmail.com
+executables: 
+- ruby-redis
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/ruby-redis
+- lib/redis/bin.rb
+- lib/redis/client.rb
+- lib/redis/config.rb
+- lib/redis/connection.rb
+- lib/redis/database.rb
+- lib/redis/hashes.rb
+- lib/redis/keys.rb
+- lib/redis/lists.rb
+- lib/redis/logger.rb
+- lib/redis/protocol.rb
+- lib/redis/pubsub.rb
+- lib/redis/reader.rb
+- lib/redis/sender.rb
+- lib/redis/server.rb
+- lib/redis/sets.rb
+- lib/redis/strict.rb
+- lib/redis/strings.rb
+- lib/redis/synchrony.rb
+- lib/redis/version.rb
+- lib/redis/zsets.rb
+- lib/redis.rb
+- LICENSE
+- README.md
+has_rdoc: true
+homepage: https://github.com/dturnbull/ruby-redis
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 23
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: ruby-redis
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Ruby implementation of a Redis server and client
+test_files: []
+