redis-file 0.4.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e596eb446bbf827dad2d9251d6802817d9b3763f
+  data.tar.gz: 5491961a1adce14639bc37f9acaeae6b6ebece98
+SHA512:
+  metadata.gz: 787696710e82958c2a07a55fdb2b81973f2260d045ffe8277cc6afde71f2e92eef378ee388e9d76f1450e5fb8c4c87664dc2677d0b983382992d0fecd9562dc6
+  data.tar.gz: 8104ba9421bf5a7bbd58368bc98a75e8c152a03308194ac2c767f7092bfcfdaa69246c0b05b47d87551fdc33c8a433f3b8079f697f123f5f893f500a38776171

data/.gitignore

@@ -0,0 +1,7 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+.rvmrc
+*.rbc
+db.redis

data/.rspec

@@ -0,0 +1 @@
+-b -fn --color

data/.travis.yml

@@ -0,0 +1,8 @@
+rvm:
+  - 1.8.7
+  - 1.9.2
+  - 1.9.3
+  - 2.0.0
+  - ree
+  - jruby
+  - rbx-19mode

data/Gemfile

@@ -0,0 +1,7 @@
+source "http://rubygems.org"
+
+gem 'rake'
+gem 'rdoc'
+
+# Specify your gem's dependencies in redis-file.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,21 @@
+Copyright (c) 2013 Daniel Loureiro
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+

data/README.md

@@ -0,0 +1,86 @@
+# redis-file [![Build Status](https://secure.travis-ci.org/loureirorg/redis-file.png)](http://travis-ci.org/loureirorg/redis-file)
+This a Ruby implementation of redis-rb for machines without Redis or development/test environments
+
+
+## Installation
+
+Install the gem:
+
+    gem install redis-file
+
+Add it to your Gemfile:
+
+    gem "redis-file"
+
+
+## Versions
+
+redis-file currently supports redis-rb v3.0.0 or later, if you are using
+redis-rb v2.2.x install the version 0.3.x:
+
+    gem install redis-file -v "~> 0.3.0"
+
+or use the branch 0-3-x on your Gemfile:
+
+    gem "redis-file", :git => "git://github.com/loureirorg/redis-file.git", :branch => "0-3-x"
+
+
+## Usage
+
+You can use redis-file without any changes:
+
+    require "redis-file"
+    
+    redis = Redis.new
+    
+    >> redis.set "foo", "bar"
+    => "OK"
+    
+    >> redis.get "foo"
+    => "bar"
+
+Read [redis-rb](https://github.com/ezmobius/redis-rb) documentation and
+[Redis](http://redis.io) homepage for more info about commands
+
+## Usage with RSpec
+
+Require this either in your Gemfile or in RSpec's support scripts. So either: 
+
+    # Gemfile
+    group :test do
+      gem "rspec"
+      gem "redis-file", :require => "redis-file/rspec"
+    end
+
+Or:
+
+    # spec/support/redis-file.rb
+    require 'redis-file/rspec'
+
+## Acknowledgements
+
+* [dim](https://github.com/dim)
+* [czarneckid](https://github.com/czarneckid)
+* [obrie](https://github.com/obrie)
+* [jredville](https://github.com/jredville)
+* [redsquirrel](https://github.com/redsquirrel)
+* [dpick](https://github.com/dpick)
+* [caius](https://github.com/caius) 
+* [Travis-CI](http://travis-ci.org/)
+
+
+## Contributing to redis-file
+
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it
+* Fork the project
+* Start a feature/bugfix branch
+* Commit and push until you are happy with your contribution
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+
+## Copyright
+
+Copyleft 2013 Daniel Loureiro (based on work of Guillermo Iguaran). See LICENSE for
+further details.

data/Rakefile

@@ -0,0 +1,27 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+$:.push File.expand_path("../lib", __FILE__)
+require "redis-file/version"
+
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+  spec.ruby_opts="-w"
+end
+
+task :default => :spec
+
+require 'rdoc/task'
+Rake::RDocTask.new do |rdoc|
+  version = RedisFile::VERSION
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "redis-file #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+

data/lib/redis-file.rb

@@ -0,0 +1,6 @@
+require 'redis'
+require 'redis/connection/file'
+
+module RedisFile
+  Redis = ::Redis
+end

data/lib/redis-file/expiring_hash.rb

@@ -0,0 +1,70 @@
+module RedisFile
+  # Represents a normal hash with some additional expiration information
+  # associated with each key
+  class ExpiringHash < Hash
+    attr_reader :expires
+
+    def initialize(*)
+      super
+      @expires = {}
+    end
+
+    def [](key)
+      key = normalize key
+      delete(key) if expired?(key)
+      super
+    end
+
+    def []=(key, val)
+      key = normalize key
+      expire(key)
+      super
+    end
+
+    def delete(key)
+      key = normalize key
+      expire(key)
+      super
+    end
+
+    def expire(key)
+      key = normalize key
+      expires.delete(key)
+    end
+
+    def expired?(key)
+      key = normalize key
+      expires.include?(key) && expires[key] < Time.now
+    end
+
+    def key?(key)
+      key = normalize key
+      delete(key) if expired?(key)
+      super
+    end
+
+    def values_at(*keys)
+      keys.each do |key|
+        key = normalize(key)
+        delete(key) if expired?(key)
+      end
+      super
+    end
+
+    def keys
+      super.select do |key|
+        key = normalize(key)
+        if expired?(key)
+          delete(key)
+          false
+        else
+          true
+        end
+      end
+    end
+
+    def normalize key
+      key.to_s
+    end
+  end
+end

data/lib/redis-file/rspec.rb

@@ -0,0 +1,23 @@
+# Require this either in your Gemfile or in RSpec's
+# support scripts. Examples:
+#
+#   # Gemfile
+#   group :test do
+#     gem "rspec"
+#     gem "redis-file", :require => "redis-file/rspec"
+#   end
+#
+#   # spec/support/redis-file.rb
+#   require 'redis-file/rspec'
+#
+
+require 'rspec/core'
+require 'redis-file'
+
+RSpec.configure do |c|
+
+  c.before do    
+    Redis::Connection::File.reset_all_databases
+  end
+
+end

data/lib/redis-file/sorted_set_argument_handler.rb

@@ -0,0 +1,74 @@
+module RedisFile
+  # Takes in the variable length array of arguments for a zinterstore/zunionstore method
+  # and parses them into a few attributes for the method to access.
+  #
+  # Handles throwing errors for various scenarios (matches redis):
+  #   * Custom weights specified, but not enough or too many given
+  #   * Invalid aggregate value given
+  #   * Multiple aggregate values given
+  class SortedSetArgumentHandler
+    # [Symbol] The aggregate method to use for the output values. One of %w(sum min max) expected
+    attr_reader :aggregate
+    # [Integer] Number of keys in the argument list
+    attr_accessor :number_of_keys
+    # [Array] The actual keys in the argument list
+    attr_accessor :keys
+    # [Array] integers for weighting the values of each key - one number per key expected
+    attr_accessor :weights
+
+    # Used internally
+    attr_accessor :type
+
+    # Expects all the argments for the method to be passed as an array
+    def initialize args
+      # Pull out known lengths of data
+      self.number_of_keys = args.shift
+      self.keys = args.shift(number_of_keys)
+      # Handle the variable lengths of data (WEIGHTS/AGGREGATE)
+      args.inject(self) {|handler, item| handler.handle(item) }
+
+      # Defaults for unspecified things
+      self.weights ||= Array.new(number_of_keys) { 1 }
+      self.aggregate ||= :sum
+
+      # Validate values
+      raise(Redis::CommandError, "ERR syntax error") unless weights.size == number_of_keys
+      raise(Redis::CommandError, "ERR syntax error") unless [:min, :max, :sum].include?(aggregate)
+    end
+
+    # Only allows assigning a value *once* - raises Redis::CommandError if a second is given
+    def aggregate=(str)
+      raise(Redis::CommandError, "ERR syntax error") if (defined?(@aggregate) && @aggregate)
+      @aggregate = str.to_s.downcase.to_sym
+    end
+
+    # Decides how to handle an item, depending on where we are in the arguments
+    def handle(item)
+      case item
+      when "WEIGHTS"
+        self.type = :weights
+        self.weights = []
+      when "AGGREGATE"
+        self.type = :aggregate
+      when nil
+        # This should never be called, raise a syntax error if we manage to hit it
+        raise(Redis::CommandError, "ERR syntax error")
+      else
+        send "handle_#{type}", item
+      end
+      self
+    end
+
+    def handle_weights(item)
+      self.weights << item
+    end
+
+    def handle_aggregate(item)
+      self.aggregate = item
+    end
+
+    def inject_block
+      lambda { |handler, item| handler.handle(item) }
+    end
+  end
+end

data/lib/redis-file/sorted_set_store.rb

@@ -0,0 +1,80 @@
+module RedisFile
+  class SortedSetStore
+    attr_accessor :data, :weights, :aggregate, :keys
+
+    def initialize params, data
+      self.data = data
+      self.weights = params.weights
+      self.aggregate = params.aggregate
+      self.keys = params.keys
+    end
+
+    def hashes
+      @hashes ||= keys.map do |src|
+        case data[src]
+        when ::Set
+          # Every value has a score of 1
+          Hash[data[src].map {|k,v| [k, 1]}]
+        when Hash
+          data[src]
+        else
+          {}
+        end
+      end
+    end
+
+    # Apply the weightings to the hashes
+    def computed_values
+      unless defined?(@computed_values) && @computed_values
+        # Do nothing if all weights are 1, as n * 1 is n
+        @computed_values = hashes if weights.all? {|weight| weight == 1 }
+        # Otherwise, multiply the values in each hash by that hash's weighting
+        @computed_values ||= hashes.each_with_index.map do |hash, index|
+          weight = weights[index]
+          Hash[hash.map {|k, v| [k, (v * weight)]}]
+        end
+      end
+      @computed_values
+    end
+
+    def aggregate_sum out
+      selected_keys.each do |key|
+        out[key] = computed_values.inject(0) do |n, hash|
+          n + (hash[key] || 0)
+        end
+      end
+    end
+
+    def aggregate_min out
+      selected_keys.each do |key|
+        out[key] = computed_values.map {|h| h[key] }.compact.min
+      end
+    end
+
+    def aggregate_max out
+      selected_keys.each do |key|
+        out[key] = computed_values.map {|h| h[key] }.compact.max
+      end
+    end
+
+    def selected_keys
+      raise NotImplemented, "subclass needs to implement #selected_keys"
+    end
+
+    def call
+      ZSet.new.tap {|out| send("aggregate_#{aggregate}", out) }
+    end
+  end
+
+  class SortedSetIntersectStore < SortedSetStore
+    def selected_keys
+      @values ||= hashes.inject([]) { |r, h| r.empty? ? h.keys : (r & h.keys) }
+    end
+  end
+
+  class SortedSetUnionStore < SortedSetStore
+    def selected_keys
+      @values ||= hashes.map(&:keys).flatten.uniq
+    end
+  end
+end

data/lib/redis-file/version.rb

@@ -0,0 +1,3 @@
+module RedisFile
+  VERSION = "0.4.2"
+end

data/lib/redis-file/zset.rb

@@ -0,0 +1,29 @@
+module RedisFile
+  class ZSet < Hash
+
+    def []=(key, val)
+      super(key, _floatify(val))
+    end
+
+    # Increments the value of key by val
+    def increment(key, val)
+      self[key] += _floatify(val)
+    end
+
+    def select_by_score min, max
+      min = _floatify(min)
+      max = _floatify(max)
+      reject {|_,v| v < min || v > max }
+    end
+
+    # Originally lifted from redis-rb
+    def _floatify(str)
+      if (( inf = str.to_s.match(/^([+-])?inf/i) ))
+        (inf[1] == "-" ? -1.0 : 1.0) / 0.0
+      else
+        Float str
+      end
+    end
+
+  end
+end