Ruby中接收单个/对象数组、转换后添加至数组的优雅实现方式

针对YAML解析时单个值与数组值需要统一处理的场景，这里分享几种更优雅的实现思路：

1. 语义化数组转换 + 直接合并 #

利用Ruby内置的Array()方法（比[*o]语义更清晰）将输入统一转为数组，再通过concat直接合并到内部数组，避免展开元素的开销：

class YamlEater
  def initialize
    @foos = []
  end

  def add(o)
    @foos.concat(Array(o).map { |item| FooItem.new(item) })
  end
end

Array()会智能处理输入：如果是数组则原样返回，非数组则包装为单元素数组，完美适配YAML的两种输出格式。

2. 提取通用辅助方法（多处复用场景） #

如果代码中多处需要这种“统一处理单个/数组输入”的逻辑，可以提取成可复用的模块：

module SingleOrArrayHandler
  def to_array(obj)
    Array(obj)
  end
end

class YamlEater
  include SingleOrArrayHandler

  def initialize
    @foos = []
  end

  def add(o)
    @foos.concat(to_array(o).map { |item| FooItem.new(item) })
  end
end

其他类也可以通过include快速复用该逻辑，保持代码一致性。

3. 链式调用风格（更流畅的代码结构） #

用tap方法实现链式操作，让代码逻辑更连贯：

def add(o)
  Array(o).map { |item| FooItem.new(item) }.tap { |processed_items| @foos.concat(processed_items) }
end

这种写法把“转换元素”和“添加到内部数组”的逻辑串联起来，可读性更强。

对比原有实现的优势 #

• 相比递归版本：避免了方法递归调用的开销，代码更简洁直观
• 相比push(*[*o].map...)：Array(o)比[*o]语义更明确，concat比push(*...)性能更优（无需展开数组元素）

内容的提问来源于stack exchange，提问作者Phrogz