blob: c6bf0980749ac2594609a372825a0bdc0e8c37c1 (
plain) (
tree)
|
|
module ActiveSupport
class WeakHash
def initialize(cache = Hash.new)
@cache = cache
@key_map = {}
@rev_cache = Hash.new{|h,k| h[k] = {}}
@reclaim_value = lambda do |value_id|
if value = @rev_cache.delete(value_id)
value.each_key{|key| @cache.delete key}
end
end
end
def [](key)
value_id = @cache[key]
value_id && ObjectSpace._id2ref(value_id)
rescue RangeError
nil
end
def []=(key, value)
key2 = case key
when Fixnum, Symbol, true, false, nil
key
else
key.dup
end
@rev_cache[value.object_id][key2] = true
@cache[key2] = value.object_id
@key_map[key.object_id] = key2
ObjectSpace.define_finalizer(value, @reclaim_value)
end
def clear
@cache.clear
end
def delete(key)
@cache.delete(key)
end
end
end
ActiveSupport::WeakHash = ::Weakling::WeakHash if defined?(RUBY_ENGINE) && RUBY_ENGINE == 'jruby'
|