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

前端之家收集整理的这篇文章主要介绍了ruby – 将参数作为单个对象或对象数组接受前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我有一堆函数可以接收一个对象的数组,或者一个单一的对象被设计为一个包含一个对象的数组,我正在寻找一个更干净的方法来实现这一点.基本上,我想知道如何使得除非下列功能的部分更简洁:
  1. def foo(bar_or_bars)
  2. unless bar_or_bars.is_a?(Array)
  3. bar_or_bars = [bar_or_bars]
  4. end
  5. bar_or_bars.each {|baz| ... }
  6. end

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

解决方法

你可以做的第一件事就是把除非逻辑写在一行中:
  1. bars = bar_or_bars.is_a?(Array) ? bar_or_bars : [bar_or_bars]

如你所见,我在这里给它一个新的名字,因为它不再是一个酒吧或酒吧,现在绝对是一个集合.

这个问题和您的原始方法是尽管您的功能可以在任何Enumerable上工作,但您将强制您的用户给您一个特定类型的参数,这会打破鸭子打字.

部分解决这个问题的一个整洁的技巧如下:

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

但是,我不会完全称之为可读.它实际上闻起来很糟糕的API设计.可能你应该更好地采取这样的多个参数:

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

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

  1. foo("XYZ")
  2. ary = ["abc","def"]
  3. foo(*ary)

猜你在找的Ruby相关文章