Как я могу превратить один объект во что-то перечисляемое в ruby

Question

У меня есть метод, который может вернуть либо один объект, либо коллекцию объектов. Я хочу иметь возможность запустить object.collect для результата этого метода, независимо от того, является ли он единственным объектом или коллекцией. Как я могу это сделать?

profiles = ProfileResource.search(params)
output = profiles.collect do | profile |
    profile.to_hash
end

Если профили - это один объект, я получаю исключение NoMethodError при попытке выполнить сбор для этого объекта.

Answer 1

Вот один лайнер:

[*ProfileResource.search(params)].collect { |profile| profile.to_hash }

Хитрость заключается в том, что сплат (*) превращает как отдельные элементы, так и перечислимые в списки аргументов (в данном случае в новый оператор массива)

Answer 2

Осторожно с подходом flatten, если search () вернул вложенные массивы, это может привести к неожиданному поведению.

profiles = ProfileResource.search(params)
profiles = [profiles] if !profiles.respond_to?(:collect)
output = profiles.collect do |profile|
    profile.to_hash
end

Answer 3

profiles = [ProfileResource.search(params)].flatten
output = profiles.collect do |profile|
    profile.to_hash
end

Answer 4

Другой способ - понять, что Enumerable требует, чтобы вы указали каждый метод.

Итак. Вы МОЖЕТЕ смешать Enumerable с вашим классом и дать ему пустышку, которая работает ....

class YourClass
  include Enumerable

  ... really important and earth shattering stuff ...

  def each
    yield(self) if block_given?
  end
end

Таким образом, если вы вернете отдельный элемент поиска в результате поиска, перечислимые методы будут работать, как и ожидалось.

Этот способ имеет то преимущество, что вся его поддержка находится внутри вашего класса, а не снаружи, где его нужно многократно дублировать.

Конечно, лучше всего изменить реализацию поиска таким образом, чтобы она возвращала массив независимо от того, сколько элементов возвращается.

Answer 5

Вы также можете использовать метод Kernel # Array.

profiles = Array(ProfileResource.search(params))
output = profiles.collect do | profile |
    profile.to_hash
end

Answer 6

Сначала вы можете проверить, отвечает ли объект на метод "collect", используя "pofiles.respond_to?"

С Программирование Ruby

obj.respond_to? ( aSymbol, includePriv = false) -> true или ложь

Возвращает true, если obj отвечает на данный метод. Частные методы включается в поиск только если необязательный второй параметр оценивается как правда.

Answer 7

profiles = ProfileResource.search(params)
output = Array(profiles).collect do |profile|
    profile.to_hash
end

Answer 8

Если коллекция является массивом, вы можете использовать эту технику

profiles = [*ProfileResource.search(params)]
output = profiles.collect do | profile |
    profile.to_hash
end

Это гарантирует, что ваши профили всегда будут массивом.

Answer 9

В методе search класса ProfileResource всегда возвращайте коллекцию объектов (обычно массив), даже если она содержит только один объект.