Sha256: 8061c7507ec2196f0b4bdd2f442a8cd6d4995c7509e81276ff2b4f633018583d

Contents?: true

Size: 1.96 KB

Versions: 75

Compression:

Stored size: 1.96 KB

Contents

require 'monitor'

module Puppet::Util::Cacher
  # Our module has been extended in a class; we can only add the Instance methods,
  # which become *class* methods in the class.
  def self.extended(other)
    class << other
      extend ClassMethods
      include InstanceMethods
    end
  end

  # Our module has been included in a class, which means the class gets the class methods
  # and all of its instances get the instance methods.
  def self.included(other)
    other.extend(ClassMethods)
    other.send(:include, InstanceMethods)
  end

  # Methods that can get added to a class.
  module ClassMethods
    # Provide a means of defining an attribute whose value will be cached.
    # Must provide a block capable of defining the value if it's flushed..
    def cached_attr(name, ttl, &block)
      init_method = "init_#{name}"
      define_method(init_method, &block)

      set_attr_ttl(name, ttl)

      define_method(name) do
        cached_value(name)
      end

      define_method(name.to_s + "=") do |value|
        # Make sure the cache timestamp is set
        value_cache.synchronize do
          value_cache[name] = value
          set_expiration(name)
        end
      end
    end

    def attr_ttl(name)
      @attr_ttls[name]
    end

    def set_attr_ttl(name, value)
      @attr_ttls ||= {}
      @attr_ttls[name] = Integer(value)
    end
  end

  # Methods that get added to instances.
  module InstanceMethods
    private

    def cached_value(name)
      value_cache.synchronize do
        if value_cache[name].nil? or expired_by_ttl?(name)
          value_cache[name] = send("init_#{name}")
          set_expiration(name)
        end
        value_cache[name]
      end
    end

    def expired_by_ttl?(name)
      @attr_expirations[name] < Time.now
    end

    def set_expiration(name)
      @attr_expirations ||= {}
      @attr_expirations[name] = Time.now + self.class.attr_ttl(name)
    end

    def value_cache
      @value_cache ||= {}.extend(MonitorMixin)
    end
  end
end

Version data entries

75 entries across 75 versions & 4 rubygems

Version Path
puppet-parse-0.1.4 lib/vendor/puppet/util/cacher.rb
puppet-parse-0.1.3 lib/vendor/puppet/util/cacher.rb
puppet-parse-0.1.2 lib/vendor/puppet/util/cacher.rb
puppet-parse-0.1.1 lib/vendor/puppet/util/cacher.rb
puppet-2.7.26 lib/puppet/util/cacher.rb
puppet-2.7.25 lib/puppet/util/cacher.rb
puppet-2.7.24 lib/puppet/util/cacher.rb
puppet-3.3.2 lib/puppet/util/cacher.rb
puppet-3.3.1 lib/puppet/util/cacher.rb
puppet-3.3.1.rc3 lib/puppet/util/cacher.rb
puppet-3.3.1.rc2 lib/puppet/util/cacher.rb
puppet-3.3.1.rc1 lib/puppet/util/cacher.rb
puppet-3.3.0 lib/puppet/util/cacher.rb
puppet-3.3.0.rc3 lib/puppet/util/cacher.rb
puppet-3.3.0.rc2 lib/puppet/util/cacher.rb
puppet-3.2.4 lib/puppet/util/cacher.rb
puppet-2.7.23 lib/puppet/util/cacher.rb
puppet-3.2.3 lib/puppet/util/cacher.rb
puppet-3.2.3.rc1 lib/puppet/util/cacher.rb
puppet-3.2.2 lib/puppet/util/cacher.rb