ruby-on-rails - 用 Ruby 思考

Question

我目前正忙于学习 Ruby 和 Rails，并且由于我有基于 C 语言的背景，Ruby 的一些概念是新的并且有些陌生。对我来说特别具有挑战性的是适应处理常见问题的“Ruby 方式”，因此我经常发现自己用 Ruby 编写 C 代码，这不是我想要实现的目标。

想象一下有这样一个模式:

ActiveRecord::Schema.define(:version => 20111119180638) do
    create_table "bikes", :force => true do |t|
        t.string   "Brand"
        t.string   "model"
        t.text     "description"
    end
end

数据库已经包含了几辆不同的自行车。我的目标是获取数据库中代表的所有品牌的数组。

这是我的代码:

class Bike < ActiveRecord::Base
    def Bike.collect_brands
        temp_brands = Bike.find_by_sql("select distinct brand from bikes")
        brands = Array.new
        temp_brands.each do |item|
          brands.push(item.brand)
        end
        brands
    end
end

Ruby 大师会如何编写代码来实现这一目标？

Answer 1

tl;dr:您的整个方法可以替换为 Bike.uniq.pluck(:brand)。

---

此功能已经存在(请参阅我的回答的末尾)，但首先，让我们单步执行您的代码并使其更加地道:

首先，也是最重要的，每一级缩进使用两个空格，而不是四个、八个，不是制表符。使用两个空格。这不是个人喜好，这是 Ruby 社区内非常的约定，如果您打算参与，这几乎是必需的。

接下来，几乎从来没有有充分的理由在 Ruby 中使用这种模式:

 brands = Array.new
 temp_brands.each do |item|
   brands.push(item.brand)
 end

当您想通过对输入数组中的每个值应用一些代码将一个数组转换为另一个数组(实际上，一个 Enumerable 转换为另一个 Enumerable)时，请使用 map或 collect (这是同义词):

brands = temp_brands.map { |item| item.brand }

接下来可以利用symbol#to_proc使上面的代码更清楚一点:

brands = temp_brands.map &:brand 

对于外行来说这看起来很奇怪，但是一旦您习惯了使用 map 和 &:field，它就会更清楚。稍微有点经验就会明白这行代码的意图很明显:它将 brand 方法应用于数组中的每个元素，它完全等同于前面的 { |item| item.brand 版本。

现在，您的整个方法可以变成一个非常简单的单行代码:

def Bike.collect_brands
  Bike.find_by_sql("select distinct brand from bikes").map &:brand
end

内联 select/distinct SQL 有点难看，特别是因为 ActiveRecord 已经让我们 select specific fields使用 select，并使用 uniq 使结果不同:

def Bike.collect_brands
  Bike.select(:brand).uniq.map &:brand
end

作为最后的迭代，我们可以使用 pluck而不是 map 只从我们感兴趣的结果中提取字段。但是，因为 pluck 实际上修改了生成的 SQL 以仅包含被提取的字段，我们可以省略 select(:brand) 部分，我们的代码归结为包含两个链接方法的非常短的单行:

def Bike.collect_brands
  Bike.uniq.pluck(:brand)
end

请注意，顺序很重要，因为 pluck 总是返回一个数组，而不是为附加方法链接准备好的 ActiveRecord 关系。 Bike.pluck(:brand).uniq 将从每条记录中选择品牌(select brand from bikes)然后，在 Ruby 中，减少数组到唯一项。可能是一项非常昂贵的操作。

就是这样，Bike.uniq.pluck(:brand)。作为一名 C 程序员，您会发现许多您习惯于使用小循环完成的重复性任务实际上已经通过语言本身或支持库为您解决了。一旦您学会了编写惯用的 Ruby 和 Rails 代码，您不会编写的代码量可能会非常惊人。