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

前端之家收集整理的这篇文章主要介绍了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)

猜你在找的Ruby相关文章