artemk-cache-money 0.2.13.2

data/TODO

@@ -0,0 +1,17 @@
+TOP PRIORITY
+
+REFACTOR
+* Clarify terminology around cache/key/index, etc.
+
+INFRASTRUCTURE
+
+NEW FEATURES
+* transactional get multi isn't really multi
+
+BUGS
+* Handle append strategy (using add rather than set?) to avoid race condition
+
+MISSING TESTS:
+* missing tests for Klass.transaction do ... end
+* non "id" pks work but lack test coverage
+* expire_cache

data/UNSUPPORTED_FEATURES

@@ -0,0 +1,12 @@
+* does not work with :dependent => nullify because
+  def nullify_has_many_dependencies(record, reflection_name, association_class, primary_key_name, dependent_conditions)
+    association_class.update_all("#{primary_key_name} = NULL", dependent_conditions)
+  end
+  This does not trigger callbacks
+* update_all, delete, update_counter, increment_counter, decrement_counter, counter_caches in general - counter caches are replaced by this gem, bear that in mind.
+* attr_readonly - no technical obstacle, just not yet supported
+* attributes before typecast behave unpredictably - hard to support
+* printf style binds: :conditions => ["name = '%s'", "37signals!"] - not too hard to support
+* objects as attributes that are serialized. story.title = {:foo => :bar}; customer.balance = Money.new(...) - these could be coerced using Column#type_cast?
+
+With a lot of these features the issue is not technical but performance. Every special case costs some overhead.

data/config/environment.rb

@@ -0,0 +1,16 @@
+require 'rubygems'
+gem 'activesupport', '~> 2.3.0'
+gem 'activerecord', '~> 2.3.0'
+gem 'actionpack', '~> 2.3.0'
+gem 'rspec', '>= 1.3.0'
+gem 'jeweler', '~> 1.4.0'
+
+require 'action_controller'
+require 'active_record'
+require 'active_record/session_store'
+require 'jeweler'
+
+ActiveRecord::Base.establish_connection(
+  :adapter => 'sqlite3',
+  :database => ':memory:'
+)

data/config/memcached.yml

@@ -0,0 +1,6 @@
+test:
+  ttl: 604800
+  namespace: cache
+  sessions: false
+  debug: false
+  servers: localhost:11211

data/db/schema.rb

@@ -0,0 +1,18 @@
+ActiveRecord::Schema.define(:version => 2) do
+  create_table "stories", :force => true do |t|
+    t.string "title", "subtitle"
+    t.string  "type"
+    t.boolean "published"
+  end
+
+  create_table "characters", :force => true do |t|
+    t.integer "story_id"
+    t.string "name"
+  end
+  
+  create_table :sessions, :force => true do |t|
+    t.string :session_id
+    t.text   :data
+    t.timestamps
+  end
+end

data/init.rb

@@ -0,0 +1 @@
+require 'cache_money'

data/lib/cache_money.rb

@@ -0,0 +1,86 @@
+require 'active_support'
+require 'active_record'
+
+require 'cash/lock'
+require 'cash/transactional'
+require 'cash/write_through'
+require 'cash/finders'
+require 'cash/buffered'
+require 'cash/index'
+require 'cash/config'
+require 'cash/accessor'
+
+require 'cash/request'
+require 'cash/fake'
+require 'cash/local'
+
+require 'cash/query/abstract'
+require 'cash/query/select'
+require 'cash/query/primary_key'
+require 'cash/query/calculation'
+
+require 'cash/util/array'
+require 'cash/util/marshal'
+
+class ActiveRecord::Base
+  def self.is_cached(options = {})
+    if options == false
+      include NoCash
+    else
+      options.assert_valid_keys(:ttl, :repository, :version)
+      include Cash unless ancestors.include?(Cash)
+      Cash::Config.create(self, options)
+    end
+  end
+
+  def <=>(other)
+    if self.id == other.id then 
+      0
+    else
+      self.id < other.id ? -1 : 1
+    end
+  end
+end
+
+module Cash
+  def self.included(active_record_class)
+    active_record_class.class_eval do
+      include Config, Accessor, WriteThrough, Finders
+      extend ClassMethods
+    end
+  end
+
+  module ClassMethods
+    def self.extended(active_record_class)
+      class << active_record_class
+        alias_method_chain :transaction, :cache_transaction
+      end
+    end
+
+    def transaction_with_cache_transaction(&block)
+      if cache_config
+        transaction_without_cache_transaction do
+          repository.transaction(&block)
+        end
+      else
+        transaction_without_cache_transaction(&block)
+      end
+    end
+
+    def cacheable?(*args)
+      true
+    end
+  end
+end
+module NoCash
+  def self.included(active_record_class)
+    active_record_class.class_eval do
+      extend ClassMethods
+    end
+  end
+  module ClassMethods
+    def cachable?(*args)
+      false
+    end
+  end
+end

data/lib/cash/accessor.rb

@@ -0,0 +1,83 @@
+module Cash
+  module Accessor
+    def self.included(a_module)
+      a_module.module_eval do
+        extend ClassMethods
+        include InstanceMethods
+      end
+    end
+
+    module ClassMethods
+      def fetch(keys, options = {}, &block)
+        case keys
+        when Array
+          return {} if keys.empty?
+          
+          keys = keys.collect { |key| cache_key(key) }
+          hits = repository.get_multi(*keys)
+          if (missed_keys = keys - hits.keys).any?
+            missed_values = block.call(missed_keys)
+            hits.merge!(missed_keys.zip(Array(missed_values)).to_hash_without_nils)
+          end
+          hits
+        else
+          repository.get(cache_key(keys), options[:raw]) || (block ? block.call : nil)
+        end
+      end
+
+      def get(keys, options = {}, &block)
+        case keys
+        when Array
+          fetch(keys, options, &block)
+        else
+          fetch(keys, options) do
+            if block_given?
+              add(keys, result = yield(keys), options)
+              result
+            end
+          end
+        end
+      end
+
+      def add(key, value, options = {})
+        if repository.add(cache_key(key), value, options[:ttl] || cache_config.ttl, options[:raw]) == "NOT_STORED\r\n"
+          yield if block_given?
+        end
+      end
+
+      def set(key, value, options = {})
+        repository.set(cache_key(key), value, options[:ttl] || cache_config.ttl, options[:raw])
+      end
+
+      def incr(key, delta = 1, ttl = nil)
+        ttl ||= cache_config.ttl
+        repository.incr(cache_key = cache_key(key), delta) || begin
+          repository.add(cache_key, (result = yield).to_s, ttl, true) { repository.incr(cache_key) }
+          result
+        end
+      end
+
+      def decr(key, delta = 1, ttl = nil)
+        ttl ||= cache_config.ttl
+        repository.decr(cache_key = cache_key(key), delta) || begin
+          repository.add(cache_key, (result = yield).to_s, ttl, true) { repository.decr(cache_key) }
+          result
+        end
+      end
+
+      def expire(key)
+        repository.delete(cache_key(key))
+      end
+
+      def cache_key(key)
+        "#{name}:#{cache_config.version}/#{key.to_s.gsub(' ', '+')}"
+      end
+    end
+
+    module InstanceMethods
+      def expire
+        self.class.expire(id)
+      end
+    end
+  end
+end

data/lib/cash/buffered.rb

@@ -0,0 +1,129 @@
+module Cash
+  class Buffered
+    def self.push(cache, lock)
+      if cache.is_a?(Buffered)
+        cache.push
+      else
+        Buffered.new(cache, lock)
+      end
+    end
+
+    def initialize(memcache, lock)
+      @buffer = {}
+      @commands = []
+      @cache = memcache
+      @lock = lock
+    end
+
+    def pop
+      @cache
+    end
+
+    def push
+      NestedBuffered.new(self, @lock)
+    end
+
+    def get(key, *options)
+      if @buffer.has_key?(key)
+        @buffer[key]
+      else
+        @buffer[key] = @cache.get(key, *options)
+      end
+    end
+
+    def set(key, value, *options)
+      @buffer[key] = value
+      buffer_command Command.new(:set, key, value, *options)
+    end
+
+    def incr(key, amount = 1)
+      return unless value = get(key, true)
+
+      @buffer[key] = value.to_i + amount
+      buffer_command Command.new(:incr, key, amount)
+      @buffer[key]
+    end
+
+    def decr(key, amount = 1)
+      return unless value = get(key, true)
+
+      @buffer[key] = [value.to_i - amount, 0].max
+      buffer_command Command.new(:decr, key, amount)
+      @buffer[key]
+    end
+
+    def add(key, value, *options)
+      @buffer[key] = value
+      buffer_command Command.new(:add, key, value, *options)
+    end
+
+    def delete(key, *options)
+      @buffer[key] = nil
+      buffer_command Command.new(:delete, key, *options)
+    end
+
+    def get_multi(*keys)
+      values = keys.collect { |key| get(key) }
+      keys.zip(values).to_hash_without_nils
+    end
+
+    def flush
+      sorted_keys = @commands.select(&:requires_lock?).collect(&:key).uniq.sort
+      sorted_keys.each do |key|
+        @lock.acquire_lock(key)
+      end
+      perform_commands
+    ensure
+      @buffer = {}
+      sorted_keys.each do |key|
+        @lock.release_lock(key)
+      end
+    end
+
+    def respond_to?(method)
+      @cache.respond_to?(method)
+    end
+
+    protected
+
+    def perform_commands
+      @commands.each do |command|
+        command.call(@cache)
+      end
+    end
+
+    def buffer_command(command)
+      @commands << command
+    end
+    
+    private
+    
+    def method_missing(method, *args, &block)
+      @cache.send(method, *args, &block)
+    end
+  end
+
+  class NestedBuffered < Buffered
+    def flush
+      perform_commands
+    end
+  end
+
+  class Command
+    attr_accessor :key
+
+    def initialize(name, key, *args)
+      @name = name
+      @key = key
+      @args = args
+    end
+
+    def requires_lock?
+      @name == :set
+    end
+
+    def call(cache)
+      cache.send @name, @key, *@args
+    end
+  end
+end

data/lib/cash/config.rb

@@ -0,0 +1,82 @@
+module Cash
+  module Config
+    def self.create(active_record, options, indices = [])
+      active_record.cache_config = Cash::Config::Config.new(active_record, options)
+      indices.each { |i| active_record.index i.attributes, i.options }
+    end
+    
+    def self.included(a_module)
+      a_module.module_eval do
+        extend ClassMethods
+        delegate :repository, :to => "self.class"
+      end
+    end
+
+    module ClassMethods
+      def self.extended(a_class)
+        class << a_class
+          def cache_config
+            @cache_config ? @cache_config : superclass.cache_config
+          end
+          
+          delegate :repository, :indices, :to => :cache_config
+          alias_method_chain :inherited, :cache_config
+        end
+      end
+
+      def inherited_with_cache_config(subclass)
+        inherited_without_cache_config(subclass)
+        @cache_config.inherit(subclass)
+      end
+
+      def index(attributes, options = {})
+        options.assert_valid_keys(:ttl, :order, :limit, :buffer, :order_column)
+        (@cache_config.indices.unshift(Index.new(@cache_config, self, attributes, options))).uniq!
+      end
+
+      def version(number)
+        @cache_config.options[:version] = number
+      end
+
+      def cache_config=(config)
+        @cache_config = config
+      end
+    end
+
+    class Config
+      attr_reader :active_record, :options
+
+      def initialize(active_record, options = {})
+        @active_record, @options = active_record, options
+      end
+
+      def repository
+        @options[:repository]
+      end
+
+      def ttl
+        @ttl ||= @options[:ttl] || default_ttl
+      end
+
+      def version
+        @options[:version] || 1
+      end
+
+      def indices
+        @indices ||= active_record == ActiveRecord::Base ? [] : [Index.new(self, active_record, active_record.primary_key)]
+      end
+
+      def inherit(active_record)
+        Cash::Config.create(active_record, @options, indices)
+      end
+      
+      private
+      
+        def default_ttl
+          default = repository.default_ttl if repository.respond_to?(:default_ttl)
+          default ||= 1.day
+          default
+        end 
+    end
+  end
+end