Module: MCollective::Cache

Defined in:
lib/mcollective/cache.rb

Overview

A class to manage a number of named caches. Each cache can have a unique timeout for keys in it and each cache has an independent Mutex protecting access to it.

This cache is setup early in the process of loading the mcollective libraries, before any threads are created etc making it suitable as a cross thread cache or just a store for Mutexes you need to use across threads like in an Agent or something.

# sets up a new cache, noop if it already exist
Cache.setup(:ddl, 600)
=> true

# writes an item to the :ddl cache, this item will
# be valid on the cache for 600 seconds
Cache.write(:ddl, :something, "value")
=> "value"

# reads from the cache, read failures due to non existing
# data or expired data will raise an exception
Cache.read(:ddl, :something)
=> "value"

# time left till expiry, raises if nothing is found
Cache.ttl(:ddl, :something)
=> 500

# forcibly evict something from the cache
Cache.invalidate!(:ddl, :something)
=> "value"

# deletes an entire named cache and its mutexes
Cache.delete!(:ddl)
=> true

# you can also just use this cache as a global mutex store
Cache.setup(:mylock)

Cache.synchronize(:mylock) do
  do_something()
end

Class Method Summary collapse

Class Method Details

.delete!(cache_name) ⇒ Object



74
75
76
77
78
79
80
81
82
83
# File 'lib/mcollective/cache.rb', line 74

def self.delete!(cache_name)
  @locks_mutex.synchronize do
    raise("No cache called '%s'" % cache_name) unless @cache_locks.include?(cache_name)

    @cache_locks.delete(cache_name)
    @cache.delete(cache_name)
  end

  true
end

.has_cache?(cache_name) ⇒ Boolean

Returns:

  • (Boolean)


68
69
70
71
72
# File 'lib/mcollective/cache.rb', line 68

def self.has_cache?(cache_name)
  @locks_mutex.synchronize do
    @cache.include?(cache_name)
  end
end

.invalidate!(cache_name, key) ⇒ Object



135
136
137
138
139
140
141
142
143
# File 'lib/mcollective/cache.rb', line 135

def self.invalidate!(cache_name, key)
  raise("No cache called '%s'" % cache_name) unless @cache.include?(cache_name)

  @cache_locks[cache_name].synchronize do
    return false unless @cache[cache_name].include?(key)

    @cache[cache_name].delete(key)
  end
end

.read(cache_name, key) ⇒ Object



97
98
99
100
101
102
103
104
105
106
107
108
109
110
# File 'lib/mcollective/cache.rb', line 97

def self.read(cache_name, key)
  raise("No cache called '%s'" % cache_name) unless @cache.include?(cache_name)

  unless ttl(cache_name, key) > 0
    Log.debug("Cache expired on '%s' key '%s'" % [cache_name, key])
    raise("Cache for item '%s' on cache '%s' has expired" % [key, cache_name])
  end

  Log.debug("Cache hit on '%s' key '%s'" % [cache_name, key])

  @cache_locks[cache_name].synchronize do
    @cache[cache_name][key][:value]
  end
end

.setup(cache_name, ttl = 300) ⇒ Object



54
55
56
57
58
59
60
61
62
63
64
65
66
# File 'lib/mcollective/cache.rb', line 54

def self.setup(cache_name, ttl=300)
  @locks_mutex.synchronize do
    break if @cache_locks.include?(cache_name)

    @cache_locks[cache_name] = Mutex.new

    @cache_locks[cache_name].synchronize do
      @cache[cache_name] = {:max_age => Float(ttl)}
    end
  end

  true
end

.synchronize(cache_name) ⇒ Object



125
126
127
128
129
130
131
132
133
# File 'lib/mcollective/cache.rb', line 125

def self.synchronize(cache_name)
  raise("No cache called '%s'" % cache_name) unless @cache.include?(cache_name)

  raise ("No block supplied to synchronize") unless block_given?

  @cache_locks[cache_name].synchronize do
    yield
  end
end

.ttl(cache_name, key) ⇒ Object



112
113
114
115
116
117
118
119
120
121
122
123
# File 'lib/mcollective/cache.rb', line 112

def self.ttl(cache_name, key)
  raise("No cache called '%s'" % cache_name) unless @cache.include?(cache_name)

  @cache_locks[cache_name].synchronize do
    unless @cache[cache_name].include?(key)
      Log.debug("Cache miss on '%s' key '%s'" % [cache_name, key])
      raise("No item called '%s' for cache '%s'" % [key, cache_name])
    end

    @cache[cache_name][:max_age] - (Time.now - @cache[cache_name][key][:cache_create_time])
  end
end

.write(cache_name, key, value) ⇒ Object



85
86
87
88
89
90
91
92
93
94
95
# File 'lib/mcollective/cache.rb', line 85

def self.write(cache_name, key, value)
  raise("No cache called '%s'" % cache_name) unless @cache.include?(cache_name)

  @cache_locks[cache_name].synchronize do
    @cache[cache_name][key] ||= {}
    @cache[cache_name][key][:cache_create_time] = Time.now
    @cache[cache_name][key][:value] = value
  end

  value
end