Ruby on Rails：寻找适用于单个对象/哈希的pluck替代方法

在Ruby on Rails里，数组方法pluck非常实用——不管是哈希数组还是对象数组，都能用它批量提取指定字段：

哈希数组示例 #

array_of_hashes = [
  { name: "Alice", age: 25 },
  { name: "Bob", age: 30 },
  { name: "Charlie", age: 35 }
]

array_of_hashes.pluck :name
=> ["Alice", "Bob", "Charlie"]

对象数组示例 #

array_of_objects = People.last(3).to_a

array_of_objects.pluck(:name)
=> ["Alice", "Bob", "Charlie"]

现在需要一个类似pluck但针对单个元素的方法：不管元素是哈希还是对象，都能统一语法提取值——比如调用object_or_hash.foobar(:name)就能返回对应的值，不用区分是调用方法还是访问哈希键。

实际需求：重构代码消除if判断 #

原代码需要根据元素类型分支处理，很繁琐：

if answer_attributes.is_a?(Hash)
  AnswerParams.new(
    question: answer_attributes[:question],
    options: answer_attributes[:options],
    answer: answer_attributes[:answer]
  )
else
  AnswerParams.new(
    question: answer_attributes.question,
    options: answer_attributes.options,
    answer: answer_attributes.answer
  )
end

解决方案 #

方案1：写个通用工具方法

定义一个简单的工具方法，自动判断元素类型并取值：

def fetch_value(obj, key)
  obj.respond_to?(key) ? obj.send(key) : obj[key]
end

重构后的代码简洁统一：

AnswerParams.new(
  question: fetch_value(answer_attributes, :question),
  options: fetch_value(answer_attributes, :options),
  answer: fetch_value(answer_attributes, :answer)
)

方案2：给对象扩展通用方法

如果需要在多个地方复用，可以给Object类扩展一个方法（注意：全局类扩展需谨慎，避免和现有方法冲突）：

class Object
  def pluck_single(key)
    respond_to?(key) ? send(key) : self[key]
  end
end

之后就能像pluck一样直接调用：

AnswerParams.new(
  question: answer_attributes.pluck_single(:question),
  options: answer_attributes.pluck_single(:options),
  answer: answer_attributes.pluck_single(:answer)
)

方案3：利用Rails自带方法快速简化

如果不想额外定义方法，也可以用Rails的try结合哈希取值（注意：仅适合对象方法返回值不会为nil的场景，避免误取哈希值）：

AnswerParams.new(
  question: answer_attributes.try(:question) || answer_attributes[:question],
  options: answer_attributes.try(:options) || answer_attributes[:options],
  answer: answer_attributes.try(:answer) || answer_attributes[:answer]
)

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