Contents

module MotionSupport
  # This module provides an internal implementation to track descendants
  # which is faster than iterating through ObjectSpace.
  module DescendantsTracker
    @@direct_descendants = {}

    class << self
      def direct_descendants(klass)
        @@direct_descendants[klass] || []
      end

      def descendants(klass)
        arr = []
        accumulate_descendants(klass, arr)
        arr
      end

      def clear
        @@direct_descendants.clear
      end

      # This is the only method that is not thread safe, but is only ever called
      # during the eager loading phase.
      def store_inherited(klass, descendant)
        (@@direct_descendants[klass] ||= []) << descendant
      end

      private
      def accumulate_descendants(klass, acc)
        if direct_descendants = @@direct_descendants[klass]
          acc.concat(direct_descendants)
          direct_descendants.each { |direct_descendant| accumulate_descendants(direct_descendant, acc) }
        end
      end
    end

    def inherited(base)
      DescendantsTracker.store_inherited(self, base)
      super
    end

    def direct_descendants
      DescendantsTracker.direct_descendants(self)
    end

    def descendants
      DescendantsTracker.descendants(self)
    end
  end
end

Version data entries

12 entries across 12 versions & 2 rubygems

Version	Path
motion-support-1.2.1	motion/descendants_tracker.rb
motion-support-1.1.1	motion/descendants_tracker.rb
motion-support-1.2.0	motion/descendants_tracker.rb
motion-support-1.1.0	motion/descendants_tracker.rb
motion-support-1.0.0	motion/descendants_tracker.rb
motion-support-0.3.0	motion/descendants_tracker.rb
motion_blender-support-0.2.8	motion/descendants_tracker.rb
motion_blender-support-0.2.7	motion/descendants_tracker.rb
motion-support-0.2.6	motion/descendants_tracker.rb
motion-support-0.2.5	motion/descendants_tracker.rb
motion-support-0.2.4	motion/descendants_tracker.rb
motion-support-0.2.3	motion/descendants_tracker.rb