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

我有一堆函数可以接收一个对象的数组,或者一个单一的对象被设计为一个包含一个对象的数组,我正在寻找一个更干净的方法来实现这一点.基本上,我想知道如何使得除非下列功能的部分更简洁：

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

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

解决方法

你可以做的第一件事就是把除非逻辑写在一行中：

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)