jashmenn-method_cache 0.7.1.0

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2008 Justin Balthrop
+
+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.rdoc

@@ -0,0 +1,40 @@
+= MethodCache
+
+MethodCache lets you easily cache the results of any instance method or class method in
+Ruby.
+
+== Usage:
+
+  class Foo
+    extend MethodCache
+
+    cache_method :bar
+    def bar
+      # do expensive calculation
+    end
+    
+    cache_class_method :baz, :clone => true, :expiry => 1.day
+    def self.baz
+      # do some expensive calculation that will be invalid tomorrow
+    end
+  end
+
+  foo = Foo.new
+  foo.bar # does calculation
+  foo.bar # cached
+  
+  Foo.baz # does calculation
+  Foo.baz # cached
+  
+  Foo.invalidate_cached_method(:baz)
+
+  Foo.baz # does calculation
+  Foo.baz # cached
+
+== Install:
+
+  gem install method_cache
+
+== License:
+
+Copyright (c) 2010 Justin Balthrop, Geni.com; Published under The MIT License, see LICENSE

data/Rakefile

@@ -0,0 +1,45 @@
+require 'rake'
+require 'rake/testtask'
+require 'rake/rdoctask'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |s|
+    s.name = "jashmenn-method_cache"
+    s.summary = %Q{Simple memcache-based memoization library for Ruby}
+    s.email = "code@justinbalthrop.com"
+    s.homepage = "http://github.com/ninjudd/method_cache"
+    s.description = "Simple memcache-based memoization library for Ruby"
+    s.authors = ["Justin Balthrop"]
+    # s.add_dependency('memcache', '>= 1.0.0')
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end
+
+Rake::TestTask.new do |t|
+  t.libs << ['lib', 'test']
+  t.pattern = 'test/**/*_test.rb'
+  t.verbose = false
+end
+
+Rake::RDocTask.new do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'method_cache'
+  rdoc.options << '--line-numbers' << '--inline-source'
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |t|
+    t.libs << 'test'
+    t.test_files = FileList['test/**/*_test.rb']
+    t.verbose = true
+  end
+rescue LoadError
+end
+
+task :default => :test

data/VERSION.yml

@@ -0,0 +1,5 @@
+---
+:major: 0
+:minor: 7
+:patch: 1
+:build: 0

data/lib/method_cache.rb

@@ -0,0 +1,186 @@
+$:.unshift(File.dirname(__FILE__))
+require 'method_cache/proxy'
+
+module MethodCache
+  def cache_method(method_name, opts = {})
+    method_name = method_name.to_sym
+    proxy = opts.kind_of?(Proxy) ? opts : Proxy.new(method_name, opts)
+
+    if self.class == Class
+      return if instance_methods.include?(proxy.method_name_without_caching)
+
+      if cached_instance_methods.empty?
+        include(InvalidationMethods)
+        extend(MethodAdded)
+      end
+
+      cached_instance_methods[method_name] = nil
+      if method_defined?(method_name) or private_method_defined?(method_name)
+        if proxy.opts[:counter]
+          define_method "increment_#{method_name}", proxy.counter_method(:increment)
+          define_method "decrement_#{method_name}", proxy.counter_method(:decrement)
+        end
+
+        # Replace instance method.
+        alias_method proxy.method_name_without_caching, method_name
+        define_method method_name, proxy.method_with_caching
+      end
+      cached_instance_methods[method_name] = proxy
+
+    elsif self.class == Module
+      # We will alias all methods when the module is mixed-in.
+      extend(ModuleAdded) if cached_module_methods.empty?
+      cached_module_methods[method_name.to_sym] = proxy
+    end
+  end
+
+  def cache_class_method(method_name, opts = {})
+    method_name = method_name.to_sym
+    proxy = opts.kind_of?(Proxy) ? opts : Proxy.new(method_name, opts)
+
+    return if methods.include?(proxy.method_name_without_caching)
+
+    if cached_class_methods.empty?
+      extend(InvalidationMethods)
+      extend(SingletonMethodAdded)
+    end
+
+    method_name = method_name.to_sym
+    cached_class_methods[method_name] = nil
+    if class_method_defined?(method_name)
+      (class << self; self; end).module_eval do
+        if proxy.opts[:counter]
+          define_method "increment_#{method_name}", proxy.counter_method(:increment)
+          define_method "decrement_#{method_name}", proxy.counter_method(:decrement)
+        end
+
+        # Replace class method.
+        alias_method proxy.method_name_without_caching, method_name
+        define_method method_name, proxy.method_with_caching
+      end
+    end
+    cached_class_methods[method_name] = proxy
+  end
+
+  def class_method_defined?(method_name)
+    method(method_name)
+    true
+  rescue NameError
+    false
+  end
+
+  def self.default_cache
+    @default_cache ||= {}
+  end
+
+  def cached_instance_methods(method_name = nil)
+    if method_name
+      method_name = method_name.to_sym
+      ancestors.each do |klass|
+        next unless klass.kind_of?(MethodCache)
+        proxy = klass.cached_instance_methods[method_name]
+        return proxy if proxy
+      end
+      nil
+    else
+      @cached_instance_methods ||= {}
+    end
+  end
+
+  def cached_class_methods(method_name = nil)
+    if method_name
+      method_name = method_name.to_sym
+      ancestors.each do |klass|
+        next unless klass.kind_of?(MethodCache)
+        proxy = klass.cached_class_methods[method_name]
+        return proxy if proxy
+      end
+      nil
+    else
+      @cached_class_methods ||= {}
+    end
+  end
+
+  def cached_module_methods(method_name = nil)
+    if method_name
+      cached_module_methods[method_name.to_sym]
+    else
+      @cached_module_methods ||= {}
+    end
+  end
+
+  def self.disable(&block)
+    @disabled, old = true, @disabled
+    yield
+  ensure
+    @disabled = old
+  end
+
+  def self.disabled?
+    @disabled
+  end
+
+  module InvalidationMethods
+    def invalidate_cached_method(method_name, *args, &block)
+      cached_method(method_name, args).invalidate(&block)
+    end
+
+    def method_value_cached?(method_name, *args)
+      cached_method(method_name, args).cached?
+    end
+
+    def update_cached_method(method_name, *args, &block)
+      cached_method(method_name, args).update(&block)
+    end
+
+    def without_method_cache(&block)
+      MethodCache.disable(&block)
+    end
+
+  private
+
+    def cached_method(method_name, args)
+      if self.kind_of?(Class) or self.kind_of?(Module)
+        proxy = cached_class_methods(method_name)
+      else
+        proxy = self.class.send(:cached_instance_methods, method_name)
+      end
+      raise "method '#{method_name}' not cached" unless proxy
+      proxy.bind(self, args)
+    end
+  end
+
+  module MethodAdded
+    def method_added(method_name)
+      if proxy = cached_instance_methods(method_name)
+        cache_method(method_name, proxy)
+      end
+      super
+    end
+  end
+
+  module SingletonMethodAdded
+    def singleton_method_added(method_name)
+      if proxy = cached_class_methods(method_name)
+        cache_class_method(method_name, proxy)
+      end
+      super
+    end
+  end
+
+  module ModuleAdded
+    def extended(mod)
+      mod.extend(MethodCache)
+      cached_module_methods.each do |method_name, proxy|
+        mod.cache_class_method(method_name, proxy)
+      end
+    end
+
+    def included(mod)
+      mod.extend(MethodCache)
+      cached_module_methods.each do |method_name, proxy|
+        mod.cache_method(method_name, proxy)
+      end
+    end
+  end
+end

data/lib/method_cache/proxy.rb

@@ -0,0 +1,220 @@
+require 'digest/sha1'
+
+class Object
+  def metaclass; class << self; self; end; end
+end
+
+module MethodCache
+  class Proxy
+    attr_reader :method_name, :opts, :args, :target
+    NULL = 'NULL'
+
+    def initialize(method_name, opts)
+      opts[:cache] ||= :counters if opts[:counter]
+      @method_name = method_name
+      @opts        = opts      
+    end
+
+    def bind(target, args)
+      self.clone.bind!(target, args)
+    end
+
+    def bind!(target, args)
+      @target = target
+      @args   = args
+      @key    = nil
+      self
+    end
+
+    def invalidate
+      if block_given?
+        # Only invalidate if the block returns true.
+        value = cache[key]
+        return if value and not yield(value)
+      end
+      cache.delete(key)
+    end
+
+    def context
+      opts[:context]
+    end
+
+    def version
+      dynamic_opt(:version)
+    end
+
+    def cached?
+      not cache[key].nil?
+    end
+
+    def update
+      value = block_given? ? yield(cache[key]) : target.send(method_name_without_caching, *args)
+      write_to_cache(key, value)
+      value
+    end
+
+    def value
+      value = opts[:counter] ? cache.count(key) : cache[key] unless MethodCache.disabled?
+      value = nil unless valid?(:load, value)
+
+      if value.nil?
+        value = target.send(method_name_without_caching, *args)
+        raise "non-integer value returned by counter method" if opts[:counter] and not value.kind_of?(Fixnum)
+        write_to_cache(key, value) if valid?(:save, value)
+      end
+
+      value = nil if value == NULL
+      if clone? and value
+        value.clone
+      else
+        value
+      end
+    end
+
+    def method_with_caching
+      proxy = self # Need access to the proxy in the closure.
+
+      lambda do |*args|
+        proxy.bind(self, args).value
+      end
+    end
+
+    def counter_method(method_name)
+      proxy = self # Need access to the proxy in the closure.
+
+      lambda do |*args|
+        if args.last.kind_of?(Hash) and args.last.keys == [:by]
+          amount = args.last[:by]
+          args.pop
+        end
+        proxy.bind(self, args).send(method_name, amount || 1)
+      end
+    end
+
+    def method_name_without_caching
+      @method_name_without_caching ||= begin
+        base_name, punctuation = method_name.to_s.sub(/([?!=])$/, ''), $1
+        "#{base_name}_without_caching#{punctuation}"
+      end
+    end
+
+    def cache
+      if @cache.nil?
+        @cache = opts[:cache] || MethodCache.default_cache
+        @cache = Memcache.pool[@cache] if @cache.kind_of?(Symbol)
+        if not @cache.respond_to?(:[]) and @cache.respond_to?(:get)
+          @cache.metaclass.module_eval do
+            define_method :[] do |key|
+              get(key)
+            end
+          end
+        end
+      end
+      @cache
+    end
+
+    def local?
+      cache.kind_of?(Hash)
+    end
+
+    def clone?
+      !!opts[:clone]
+    end
+
+    def key
+      if @key.nil?
+        arg_string = ([method_name, target] + args).collect do |arg|
+          object_key(arg)
+        end.join('|')
+        @key = ['m', version, arg_string].compact.join('|')
+        @key = "m|#{Digest::SHA1.hexdigest(@key)}" if @key.length > 250
+      end
+      @key
+    end
+
+  private
+
+    def expiry(value)
+      value = dynamic_opt(:expiry, value).to_i
+      if defined?(Memcache) and cache.kind_of?(Memcache)
+        {:expiry => value}
+      else
+        value
+      end
+    end
+
+    def valid?(type, value)
+      name = "#{type}_validation".to_sym
+      return true unless opts[name]
+      return unless value
+
+      dynamic_opt(name, value)
+    end
+
+    def dynamic_opt(name, value = nil)
+      if opts[name].kind_of?(Proc)
+        proc = opts[name].bind(target)
+        case proc.arity
+        when 0 then proc.call()
+        when 1 then proc.call(value)
+        else
+          proc.call(value, *args)
+        end
+      else
+        opts[name]
+      end
+    end
+
+    def write_to_cache(key, value)
+      unless opts[:counter]
+        value = value.nil? ? NULL : value
+      end
+      if cache.kind_of?(Hash)
+        raise 'expiry not permitted when cache is a Hash'        if opts[:expiry]
+        raise 'counter cache not permitted when cache is a Hash' if opts[:counter]
+        cache[key] = value
+      elsif opts[:counter]
+        cache.write(key, value.to_s, expiry(value))
+      else
+        cache.set(key, value, expiry(value))
+      end
+    end
+
+    def increment(amount)
+      raise "cannot increment non-counter method" unless opts[:counter]
+      cache.incr(key, amount)
+    end
+
+    def decrement(amount)
+      raise "cannot decrement non-counter method" unless opts[:counter]
+      cache.decr(key, amount)
+    end
+
+    def object_key(arg)
+      return "#{class_key(arg.class)}-#{arg.string_hash}" if arg.respond_to?(:string_hash)
+
+      case arg
+      when NilClass      then 'nil'
+      when TrueClass     then 'true'
+      when FalseClass    then 'false'
+      when Numeric       then arg.to_s
+      when Symbol        then ":#{arg}"
+      when String        then "'#{arg}'"
+      when Class, Module then class_key(arg)
+      when Hash
+        '{' + arg.collect {|key, value| "#{object_key(key)}=#{object_key(value)}"}.sort.join(',') + '}'
+      when Array
+        '[' + arg.collect {|item| object_key(item)}.join(',') + ']'
+      when defined?(ActiveRecord::Base) && ActiveRecord::Base
+        "#{class_key(arg.class)}-#{arg.id}"
+      else
+        hash = local? ? arg.hash : Marshal.dump(arg).hash
+        "#{class_key(arg.class)}-#{hash}"
+      end
+    end
+
+    def class_key(klass)
+      klass.respond_to?(:version) ? "#{klass.name}_#{klass.version(context)}" : klass.name
+    end
+  end
+end

data/test/method_cache_remote_test.rb

@@ -0,0 +1,25 @@
+require 'test_helper'
+require 'dalli'
+
+PORT    = 19112
+$client = Dalli::Client.new("localhost:#{PORT}")
+
+class FooBar
+  extend MethodCache
+
+  cache_method :foo, :cache => $client 
+  def foo
+    'bar'
+  end
+end
+
+class MethodCacheRemoteTest < Test::Unit::TestCase
+
+  should 'work with dalli client' do
+    start_memcache(PORT)
+    $client.flush
+
+    f = FooBar.new
+    assert_equal 'bar', f.foo
+  end
+end

data/test/method_cache_test.rb

@@ -0,0 +1,201 @@
+require 'test_helper'
+require 'memcache'
+
+class Foo
+  extend MethodCache
+
+  def foo(i)
+    @i ||= 0
+    @i  += i
+  end
+  cache_method :foo
+
+  cache_method :bar
+  def bar
+    @i ||= 0
+    @i  += 1
+  end
+
+  @@i = 0
+  def baz(i)
+    @@i += i
+  end
+  cache_method :baz, :cache => :remote
+
+  cache_class_method :bap
+  def self.bap(i)
+    @i ||= 0
+    @i  += i
+  end
+  
+  cache_class_method :zap, :counter => true
+  def self.zap
+    0
+  end
+
+  attr_accessor :z
+  def zang
+    self.z ||= 1
+    self.z  += 1
+    nil
+  end
+  cache_method :zang
+end
+
+module Bar
+  extend MethodCache
+
+  cache_method :foo
+  def foo(i)
+    @i ||= 0
+    @i  += i
+  end
+
+  cache_method :foo_count, :counter => true, :cache => :default
+  def foo_count(key)
+    100
+  end
+end
+
+class Baz
+  include Bar
+  extend  Bar
+end
+
+class TestMethodCache < Test::Unit::TestCase
+  should 'cache methods locally' do
+    a = Foo.new
+    f1 = a.foo(1)
+    f2 = a.foo(2)
+
+    assert_equal 1, f1
+    assert_equal 3, f2
+
+    assert f1 == a.foo(1)
+    assert f1 != f2
+    assert f2 == a.foo(2)
+
+    b = a.bar
+    assert b == a.bar
+    assert b == a.bar
+  end
+
+  should 'disable method_cache' do
+    a = Foo.new
+    f1 = a.foo(1)
+
+    f2 = a.without_method_cache do
+      a.foo(1)
+    end
+
+    f3 = MethodCache.disable do
+      a.foo(1)
+    end
+
+    assert f1 != f2
+    assert f1 != f3
+    assert f2 != f3
+  end
+
+  should 'cache methods remotely' do
+    a = Foo.new
+    b1 = a.baz(1)
+    b2 = a.baz(2)
+
+    assert_equal 1, b1
+    assert_equal 3, b2
+
+    assert b1 == a.baz(1)
+    assert b1 != b2
+    assert b2 == a.baz(2)
+  end
+
+  should 'cache class methods' do
+    assert_equal 10, Foo.bap(10)
+    assert_equal 23, Foo.bap(13)
+    assert_equal 10, Foo.bap(10)
+    assert_equal 23, Foo.bap(13)
+  end
+
+  should 'cache methods for mixins' do
+    a = Baz.new
+
+    assert_equal 1, a.foo(1)
+    assert_equal 1, a.foo(1)
+    assert_equal 3, a.foo(2)
+    assert_equal 3, a.foo(2)
+  end
+
+  should 'cache class methods for mixins' do
+    assert_equal 1, Baz.foo(1)
+    assert_equal 1, Baz.foo(1)
+    assert_equal 3, Baz.foo(2)
+    assert_equal 3, Baz.foo(2)
+  end
+
+  should 'invalidate cached method' do
+    a = Foo.new
+
+    assert_equal 1, a.foo(1)
+    assert_equal 3, a.foo(2)
+
+    a.invalidate_cached_method(:foo, 1)
+
+    assert_equal 4, a.foo(1)
+    assert_equal 3, a.foo(2)
+  end
+
+  should 'cache counters' do
+    b = Baz.new
+
+    assert_equal 100, b.foo_count(:bar)
+    b.increment_foo_count(:bar, :by => 42)
+    assert_equal 142, b.foo_count(:bar)
+    b.decrement_foo_count(:bar, :by => 99)
+    assert_equal 43, b.foo_count(:bar)
+    b.increment_foo_count(:bar)
+    assert_equal 44, b.foo_count(:bar)
+
+    assert_equal 100, b.foo_count(:baz)
+    b.increment_foo_count(:baz)
+    assert_equal 101, b.foo_count(:baz)
+    assert_equal 44,  b.foo_count(:bar) # make sure :bar wasn't affected
+
+    assert_equal 0, Foo.zap
+    Foo.increment_zap(:by => 3)
+    assert_equal 3, Foo.zap
+    Foo.decrement_zap
+    assert_equal 2, Foo.zap
+  end
+
+  should 'use consistent local keys' do
+    a = Foo.new
+    o = Object.new
+    a_hash = a.hash
+    o_hash = o.hash
+
+    5.times do
+      key = a.send(:cached_method, :bar, [{'a' => 3, 'b' => [5,6], 'c' => o}, [1,nil,{:o => o}]]).key
+      assert_equal "m|:bar|Foo-#{a_hash}|{'a'=3,'b'=[5,6],'c'=Object-#{o_hash}}|[1,nil,{:o=Object-#{o_hash}}]", key
+    end
+  end
+
+  should 'use consistent remote keys' do
+    a = Foo.new
+    o = Object.new
+    a_hash = Marshal.dump(a).hash
+    o_hash = Marshal.dump(o).hash
+
+    5.times do
+      key = a.send(:cached_method, :baz, [{:a => 3, :b => [5,6], :c => o}, [false,true,{:o => o}]]).key
+      assert_equal "m|:baz|Foo-#{a_hash}|{:a=3,:b=[5,6],:c=Object-#{o_hash}}|[false,true,{:o=Object-#{o_hash}}]", key
+    end
+  end
+  
+  should 'cache nil locally' do
+    a = Foo.new
+    a.zang
+    a.zang
+    assert_equal 2, a.z
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+require 'mocha'
+require 'pp'
+
+$LOAD_PATH.unshift File.dirname(__FILE__) + "/../lib"
+$LOAD_PATH.unshift File.dirname(__FILE__) + "/../../memcache/lib"
+$LOAD_PATH.unshift File.dirname(__FILE__) + "/../../cache_version/lib"
+require 'method_cache'
+
+class Test::Unit::TestCase
+  def start_memcache(port)
+    system("memcached -p #{port} -U 0 -d -P /tmp/memcached_#{port}.pid")
+    sleep 1
+    File.read("/tmp/memcached_#{port}.pid")
+  end
+end

metadata

@@ -0,0 +1,55 @@
+--- !ruby/object:Gem::Specification
+name: jashmenn-method_cache
+version: !ruby/object:Gem::Version
+  version: 0.7.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Justin Balthrop
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-17 00:00:00.000000000Z
+dependencies: []
+description: Simple memcache-based memoization library for Ruby
+email: code@justinbalthrop.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE
+- README.rdoc
+files:
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION.yml
+- lib/method_cache.rb
+- lib/method_cache/proxy.rb
+- test/method_cache_remote_test.rb
+- test/method_cache_test.rb
+- test/test_helper.rb
+homepage: http://github.com/ninjudd/method_cache
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Simple memcache-based memoization library for Ruby
+test_files: []