接受参数作为单个对象或对象数组

Question

我有一堆函数可以接受一个对象数组，或者一个对象被设计为一个包含一个对象的数组，我正在寻找一种更简洁的方法来实现它。基本上，我想知道如何使以下函数中的unless部分更简洁：

def foo(bar_or_bars)
  unless bar_or_bars.is_a?(Array)
    bar_or_bars = [bar_or_bars]
  end
  bar_or_bars.each { |baz| ... }
end

任何帮助将不胜感激！感谢。

Answer 1

我发现最干净的解决方案是使用内核方法Array：

Array(5) #=> [5]
Array([1, 2, 3]) #=> [1,2,3]

所以

def foo(bar_or_bars)
  bars = Array(bar_or_bars)
  bars.each { |baz| ... }

这甚至适用于将数组作为元素的嵌套数组（它们不会被展平）

Answer 2

您可以做的第一件事就是将unless逻辑写在一行中：

bars = bar_or_bars.is_a?(Array) ? bar_or_bars : [bar_or_bars]

如你所见，我在这里给它一个新名称，因为它不再是一个酒吧或酒吧，它现在绝对是一个集合。

这个和你原来的方法的问题在于，虽然你的函数可以在任何Enumerable上运行，但是你会强迫你的用户给你一个特定类型的参数，这会破坏鸭子的输入。

部分解决该问题的巧妙方法如下：

def foo(bar_or_bars)
  bars = [*bar_or_bars]
  bars.each { |baz| ... }
end

但是，我不会完全称之为可读。它实际上闻起来很像糟糕的API设计。可能你应该更好地采取这样的多个论点：

def foo(*bars)
  bars.each { |baz| ... }
end

让调用者决定是否要传递单个对象或数组：

foo("XYZ")
ary = ["abc", "def"]
foo(*ary)

Answer 3

我目前正在使用这个：

bars = [bar_or_bars].flatten