ruby-on-rails - ActiveRecord - PostgreSQL 中的 Calculate() 方法有多快？

Question

关于 ruby​​ on rails 中的 ActiveRecord，我有一个相当菜鸟的问题。我正在 Postgresql 数据库上开发一个应用程序，它需要尽快处理来自多个平台的大量数据。我正在经历尝试优化速度的过程。

我有两个函数，我想知道理论上哪一个会更快。

示例 #1

def spend_branded(date_range)
    total_branded_spend = 0.0
    platform_list.each do |platform|
      platform.where(date: date_range).each do |platform_performance|
        total_branded_spend += platform_performance.spend["branded"].to_f
      end
    end

    total_branded_spend
 end

对比

示例#2

def spend_branded(date_range)
    total_branded_spend = 0.0
    platform_list.each do |platform|
      total_branded_spend += (platform.where(date: date_range).sum(:branded_spend)).to_f
    end

    total_branded_spend
  end

如您所见，在第一个示例中，选择的记录通过 .where() 方法检索，然后使用手动求和的所需字段进行迭代。然而，在第二个示例中，我使用 .sum() 方法在数据库级别进行求和。

我想知道是否有人知道一般来说哪种方法更快。我怀疑第二种方法更快，但速度快了多少度？

非常感谢您花时间阅读这个问题。

编辑:

正如@lacostenycoder 指出的那样，我应该澄清什么是platform_list。它引用一个数组，其中包含 1 到 3 个 ActiveRecord 集合，在 date_range 中每天包含 1 条记录。在使用他的回答中提供的方法进行基准测试后，我发现第二种方法稍微快一些。

       user     system      total        real
spend_branded  0.000000   0.000000   0.000000 (  0.003632)
spend_branded_sum  0.000000   0.000000   0.000000 (  0.002612)

(已处理 102 条记录)

Answer 1

下面介绍了如何对您的方法进行基准测试以进行比较。打开 Rails 控制台 rails c，然后将其粘贴到您的控制台中。

def spend_branded(date_range)
  total_branded_spend = 0.0
  platform_list.each do |platform|
    platform.where(date: date_range).each do |platform_performance|
      total_branded_spend += platform_performance.spend["branded"].to_f
    end
  end
  total_branded_spend
 end

def spend_branded_sum(date_range)
  total_branded_spend = 0.0
  platform_list.each do |platform|
    total_branded_spend += (platform.where(date: date_range).sum(:branded_spend)).to_f
  end
  total_branded_spend
end

require 'benchmark'
Benchmark.bm do |x|
  x.report(:spend_branded) { spend_branded(date_range) }
  x.report(:spend_branded_sum) { spend_branded_sum(date_range) }
end

当然，我们希望第二种方式更快。如果您展示更多关于模型关系以及如何定义 platform_list 的信息，我们可能会提供更多帮助。

您可能还想查看 PgHero gem这有助于识别慢速查询以及在何处添加索引以获得更好的性能。一般来说，如果正确完成，在数据库级别进行适当的计算将比对大型 Rub​​y 对象集进行迭代快几个数量级。

您也可以尝试将您的第一个版本重构为:

def spend_branded(date_range)
  platform_list.map do |platform|
    platform.where(date: date_range)
      .pluck(:spend).map{|h| h['branded'].to_f}.sum
   end.sum
 end

第二个版本

def spend_branded_sum(date_range)
  platform_list.map do |platform|
    platform.where(date: date_range).sum(:branded_spend).to_f
  end.sum
end