rails--rails/activesupport/lib/active_support/descendants_tracker.rb

44 lines
1.1 KiB
Ruby
Raw Normal View History

2010-06-19 14:44:35 +00:00
require 'active_support/dependencies'
module ActiveSupport
# This module provides an internal implementation to track descendants
# which is faster than iterating through ObjectSpace.
module DescendantsTracker
@@direct_descendants = Hash.new { |h, k| h[k] = [] }
2010-06-19 14:44:35 +00:00
def self.direct_descendants(klass)
@@direct_descendants[klass]
end
def self.descendants(klass)
2010-07-22 17:57:42 +00:00
@@direct_descendants[klass].inject([]) do |descendants, _klass|
descendants << _klass
descendants.concat _klass.descendants
end
end
2010-06-19 14:44:35 +00:00
def self.clear
@@direct_descendants.each do |klass, descendants|
2010-06-19 14:44:35 +00:00
if ActiveSupport::Dependencies.autoloaded?(klass)
@@direct_descendants.delete(klass)
2010-06-19 14:44:35 +00:00
else
descendants.reject! { |v| ActiveSupport::Dependencies.autoloaded?(v) }
end
end
end
def inherited(base)
self.direct_descendants << base
super
end
def direct_descendants
DescendantsTracker.direct_descendants(self)
2010-06-19 14:44:35 +00:00
end
def descendants
DescendantsTracker.descendants(self)
2010-06-19 14:44:35 +00:00
end
end
2010-07-22 17:57:42 +00:00
end