36 lines
916 B
Ruby
36 lines
916 B
Ruby
|
require 'active_support/dependencies'
|
||
|
|
||
|
module ActiveSupport
|
||
|
# This module provides an internal implementation to track descendants
|
||
|
# which is faster than iterating through ObjectSpace.
|
||
|
module DescendantsTracker
|
||
|
mattr_accessor :descendants
|
||
|
@@descendants = Hash.new { |h, k| h[k] = [] }
|
||
|
|
||
|
def self.clear
|
||
|
@@descendants.each do |klass, descendants|
|
||
|
if ActiveSupport::Dependencies.autoloaded?(klass)
|
||
|
@@descendants.delete(klass)
|
||
|
else
|
||
|
descendants.reject! { |v| ActiveSupport::Dependencies.autoloaded?(v) }
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def inherited(base)
|
||
|
self.direct_descendants << base
|
||
|
super
|
||
|
end
|
||
|
|
||
|
def direct_descendants
|
||
|
@@descendants[self]
|
||
|
end
|
||
|
|
||
|
def descendants
|
||
|
@@descendants[self].inject([]) do |descendants, klass|
|
||
|
descendants << klass
|
||
|
descendants.concat klass.descendants
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|