1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activesupport/lib/active_support/descendants_tracker.rb

46 lines
1.2 KiB
Ruby
Raw Normal View History

2010-06-19 10:44:35 -04:00
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 10:44:35 -04:00
def self.direct_descendants(klass)
@@direct_descendants[klass]
end
def self.descendants(klass)
2010-07-22 13:57:42 -04:00
@@direct_descendants[klass].inject([]) do |descendants, _klass|
descendants << _klass
descendants.concat _klass.descendants
end
end
2010-06-19 10:44:35 -04:00
def self.clear
if defined? ActiveSupport::Dependencies
@@direct_descendants.each do |klass, descendants|
if ActiveSupport::Dependencies.autoloaded?(klass)
@@direct_descendants.delete(klass)
else
descendants.reject! { |v| ActiveSupport::Dependencies.autoloaded?(v) }
end
2010-06-19 10:44:35 -04:00
end
else
@@direct_descendants.clear
2010-06-19 10:44:35 -04:00
end
end
def inherited(base)
self.direct_descendants << base
super
end
def direct_descendants
DescendantsTracker.direct_descendants(self)
2010-06-19 10:44:35 -04:00
end
def descendants
DescendantsTracker.descendants(self)
2010-06-19 10:44:35 -04:00
end
end
2010-07-22 13:57:42 -04:00
end