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