d89a7967b5
This reverts commit 9f5b1e1ed08df9dbedded0a6b7798d919d43b9a6. Tests have been refactored so they pass in isolation.
46 lines
1.2 KiB
Ruby
46 lines
1.2 KiB
Ruby
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] = [] }
|
|
|
|
def self.direct_descendants(klass)
|
|
@@direct_descendants[klass]
|
|
end
|
|
|
|
def self.descendants(klass)
|
|
@@direct_descendants[klass].inject([]) do |descendants, _klass|
|
|
descendants << _klass
|
|
descendants.concat _klass.descendants
|
|
end
|
|
end
|
|
|
|
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
|
|
end
|
|
else
|
|
@@direct_descendants.clear
|
|
end
|
|
end
|
|
|
|
def inherited(base)
|
|
self.direct_descendants << base
|
|
super
|
|
end
|
|
|
|
def direct_descendants
|
|
DescendantsTracker.direct_descendants(self)
|
|
end
|
|
|
|
def descendants
|
|
DescendantsTracker.descendants(self)
|
|
end
|
|
end
|
|
end
|