如何将单个对象转换为ruby中的Enumerable

Question

我有一个方法可以返回单个对象或对象集合。 我希望能够在该方法的结果上运行object.collect，无论它是否是单个对象或集合。 我怎样才能做到这一点？

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

如果配置文件是单个对象，当我尝试在该对象上执行collect时，我会收到NoMethodError异常。

Answer 1

小心使用展平方法，如果search（）返回嵌套数组，则可能会导致意外行为。

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

Answer 2

这是一个班轮：

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

诀窍是splat（*）将单个元素和枚举变为参数列表（在本例中为新数组运算符）

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

在ProfileResource类的search方法中，始终返回一组对象（通常是一个Array），即使它只包含一个对象。

Answer 6

如果集合是数组，则可以使用此技术

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

这样可以保证您的配置文件始终是一个数组。

Answer 7

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

Answer 8

您可以首先使用“pofiles.respond_to？”检查对象是否响应“collect”方法。

来自编程Ruby

obj.respond_to？（aSymbol，includePriv = false） - > true或false

如果obj响应给定方法，则返回true。 仅当可选的第二个参数的计算结果为true时，才会在搜索中包含私有方法。

Answer 9

您也可以使用Kernel＃Array方法。

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