1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
|
# Ruby 1.9 makes our life easier, Hash is already ordered
#
# This is an ultra efficient 1.9 freindly implementation
class LruRedux::Cache
def initialize(*args)
max_size, _ = args
raise ArgumentError.new(:max_size) if max_size < 1
@max_size = max_size
@data = {}
end
def max_size=(max_size)
max_size ||= @max_size
raise ArgumentError.new(:max_size) if max_size < 1
@max_size = max_size
@data.shift while @data.size > @max_size
end
def ttl=(_)
nil
end
def getset(key)
found = true
value = @data.delete(key){ found = false }
if found
@data[key] = value
else
result = @data[key] = yield
@data.shift if @data.length > @max_size
result
end
end
def fetch(key)
found = true
value = @data.delete(key){ found = false }
if found
@data[key] = value
else
yield if block_given?
end
end
def [](key)
found = true
value = @data.delete(key){ found = false }
if found
@data[key] = value
else
nil
end
end
def []=(key,val)
@data.delete(key)
@data[key] = val
@data.shift if @data.length > @max_size
val
end
def each
array = @data.to_a
array.reverse!.each do |pair|
yield pair
end
end
# used further up the chain, non thread safe each
alias_method :each_unsafe, :each
def to_a
array = @data.to_a
array.reverse!
end
def delete(key)
@data.delete(key)
end
alias_method :evict, :delete
def key?(key)
@data.key?(key)
end
alias_method :has_key?, :key?
def clear
@data.clear
end
def count
@data.size
end
protected
# for cache validation only, ensures all is sound
def valid?
true
end
end
|