Contents

class Redis
  class Mutex
    module Macro
      def self.included(base)
        base.extend ClassMethods
        base.class_eval do
          class << self
            attr_accessor :auto_mutex_methods
          end
          @auto_mutex_methods = {}
        end
      end

      module ClassMethods
        def auto_mutex(target, options={})
          self.auto_mutex_methods[target] = options
        end

        def method_added(target)
          return if target.to_s =~ /^auto_mutex_methods/
          return unless self.auto_mutex_methods[target]
          without_method  = "#{target}_without_auto_mutex"
          with_method     = "#{target}_with_auto_mutex"
          after_method    = "#{target}_after_failure"
          return if method_defined?(without_method)
          options = self.auto_mutex_methods[target]

          if options[:after_failure].is_a?(Proc)
            define_method(after_method, &options[:after_failure])
          end

          define_method(with_method) do |*args|
            key = self.class.name << '#' << target.to_s
            response = nil

            success = Redis::Mutex.lock(key, options) do
              response = send(without_method, *args)
            end

            if !success and respond_to?(after_method)
              response = send(after_method, *args)
            end

            response
          end

          alias_method without_method, target
          alias_method target, with_method
        end
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
redis-mutex-1.3.5	lib/redis/mutex/macro.rb
redis-mutex-1.3.4	lib/redis/mutex/macro.rb
redis-mutex-1.3.3	lib/redis/mutex/macro.rb
redis-mutex-1.3.2	lib/redis/mutex/macro.rb
redis-mutex-1.3.1	lib/redis/mutex/macro.rb
redis-mutex-1.3.0	lib/redis/mutex/macro.rb
redis-mutex-1.2.3	lib/redis/mutex/macro.rb
redis-mutex-1.2.2	lib/redis/mutex/macro.rb