Знайдіть усіх нащадків класу в Рубі

Я легко переходжу ієрархію класів у Ruby:

String.ancestors     # [String, Enumerable, Comparable, Object, Kernel]
Enumerable.ancestors # [Enumerable]
Comparable.ancestors # [Comparable]
Object.ancestors     # [Object, Kernel]
Kernel.ancestors     # [Kernel]

Чи є якийсь спосіб також зійти з ієрархії? Я хотів би це зробити

Animal.descendants      # [Dog, Cat, Human, ...]
Dog.descendants         # [Labrador, GreatDane, Airedale, ...]
Enumerable.descendants  # [String, Array, ...]

але, схоже, немає descendantsспособу.

(Це питання виникає, тому що я хочу знайти всі моделі в додатку Rails, що виходять з базового класу, і перерахувати їх; у мене є контролер, який може працювати з будь-якою такою моделлю, і я хотів би мати можливість додавати нові моделі без необхідності змінювати контролер.)

Ось приклад:

class Parent
  def self.descendants
    ObjectSpace.each_object(Class).select { |klass| klass < self }
  end
end

class Child < Parent
end

class GrandChild < Child
end

puts Parent.descendants
puts Child.descendants

ставить Parent.descendants дає вам:

GrandChild
Child

ставить Child.descendants дає вам:

GrandChild

Якщо ви використовуєте Rails> = 3, у вас є два варіанти. Використовуйте, .descendantsякщо ви хочете більше, ніж на рівні рівнів дітей, або використовуйте .subclassesдля дітей першого класу.

Приклад:

class Animal
end

class Mammal < Animal
end

class Dog < Mammal
end

class Fish < Animal
end

Animal.subclasses #=> [Mammal, Fish] 
Animal.descendants  #=> [Dog, Mammal, Fish]

Ruby 1.9 (або 1.8.7) з чудовими прикованими ітераторами:

#!/usr/bin/env ruby1.9

class Class
  def descendants
    ObjectSpace.each_object(::Class).select {|klass| klass < self }
  end
end

Рубін до 1,8,7:

#!/usr/bin/env ruby

class Class
  def descendants
    result = []
    ObjectSpace.each_object(::Class) {|klass| result << klass if klass < self }
    result
  end
end

Використовуйте його так:

#!/usr/bin/env ruby

p Animal.descendants

Перезазначте метод класу, названий успадкованим . Цей метод буде переданий підкласу, коли він створений, який ви можете відстежувати.

Як варіант (оновлено для ruby ​​1.9+):

ObjectSpace.each_object(YourRootClass.singleton_class)

Сумісний спосіб Ruby 1.8:

ObjectSpace.each_object(class<<YourRootClass;self;end)

Зауважте, що це не працює для модулів. Також у відповідь буде включено YourRootClass. Ви можете використовувати Array # - або інший спосіб її видалення.

Хоча використовуючи роботи ObjectSpace, метод спадкового класу, здається, краще підходить тут успадкованій (підклас) Ruby документації

Об'єктний простір - це, по суті, спосіб отримати доступ до будь-чого і всього, що зараз використовує виділену пам’ять, тому ітерація над кожним його елементом, щоб перевірити, чи це підклас класу Animal, не є ідеальним.

У наведеному нижче коді метод успадкованого класу Animal реалізує зворотний виклик, який додасть будь-який новостворений підклас до масиву його нащадків.

class Animal
  def self.inherited(subclass)
    @descendants = []
    @descendants << subclass
  end

  def self.descendants
    puts @descendants 
  end
end

Я знаю, що ви запитуєте, як це зробити у спадок, але ви можете досягти цього безпосередньо в Ruby, розділяючи клас ( Classабо Module)

module DarthVader
  module DarkForce
  end

  BlowUpDeathStar = Class.new(StandardError)

  class Luck
  end

  class Lea
  end
end

DarthVader.constants  # => [:DarkForce, :BlowUpDeathStar, :Luck, :Lea]

DarthVader
  .constants
  .map { |class_symbol| DarthVader.const_get(class_symbol) }
  .select { |c| !c.ancestors.include?(StandardError) && c.class != Module }
  # => [DarthVader::Luck, DarthVader::Lea]

Це набагато швидше, ніж порівняння з кожним класом, ObjectSpaceяк пропонують інші рішення.

Якщо вам це серйозно потрібно в спадщину, ви можете зробити щось подібне:

class DarthVader
  def self.descendants
    DarthVader
      .constants
      .map { |class_symbol| DarthVader.const_get(class_symbol) }
  end

  class Luck < DarthVader
    # ...
  end

  class Lea < DarthVader
    # ...
  end

  def force
    'May the Force be with you'
  end
end

орієнтири тут: http://www.eq8.eu/blogs/13-ruby-ancestors-descendants-and-other-annoying-relatives

оновлення

врешті-решт все, що вам потрібно зробити, це це

class DarthVader
  def self.inherited(klass)
    @descendants ||= []
    @descendants << klass
  end

  def self.descendants
    @descendants || []
  end
end

class Foo < DarthVader
end

DarthVader.descendants #=> [Foo]

дякую @saturnflyer за пропозицію

(Rails <= 3.0) Або ви можете використовувати ActiveSupport :: DescendantsTracker, щоб зробити діло. З джерела:

Цей модуль забезпечує внутрішню реалізацію для відстеження нащадків, що швидше, ніж ітерація через ObjectSpace.

Оскільки це модуляризовано, ви можете просто «вишнево» вибрати саме цей модуль для додатка Ruby.

Рубін Фасет має нащадків класу #

require 'facets/class/descendants'

Він також підтримує генераційний параметр відстані.

Проста версія, яка дає масив усіх нащадків класу:

def descendants(klass)
  all_classes = klass.subclasses
  (all_classes + all_classes.map { |c| descendants(c) }.reject(&:empty?)).flatten
end

Можна require 'active_support/core_ext'і використовувати descendantsметод. Ознайомтеся з документом і надішліть йому знімок в IRB або довідайтесь. Можна використовувати без рейок.

Rails надає метод підкласів для кожного об'єкта, але він недостатньо задокументований, і я не знаю, де це визначено. Він повертає масив назв класу у вигляді рядків.

Використання дорогоцінного каміння descendants_tracker може допомогти. Наступний приклад скопійовано з документа дорогоцінного каміння:

class Foo
  extend DescendantsTracker
end

class Bar < Foo
end

Foo.descendants # => [Bar]

Цей дорогоцінний камінь використовується популярним самоцвітом virtus , тому я думаю, що це досить солідно.

Цей метод поверне багатовимірний хеш усіх нащадків Об'єкта.

def descendants_mapper(klass)
  klass.subclasses.reduce({}){ |memo, subclass|
    memo[subclass] = descendants_mapper(subclass); memo
  }
end

{ MasterClass => descendants_mapper(MasterClass) }

Якщо у вас є доступ до коду перед завантаженням будь-якого підкласу, ви можете використовувати успадкований метод.

Якщо ви цього не зробите (це не так, але це може бути корисно всім, хто знайшов цю публікацію), ви можете просто написати:

x = {}
ObjectSpace.each_object(Class) do |klass|
     x[klass.superclass] ||= []
     x[klass.superclass].push klass
end
x[String]

Вибачте, якщо я пропустив синтаксис, але ідея повинна бути зрозумілою (на даний момент у мене немає доступу до рубіну).