ruby-on-rails - 从原始 sql 查询返回具有多个键/值对的哈希

Question

我正在构建一个使用原始 SQL 查询我的数据库的 Ruby on Rails 应用程序。

为了简单起见，我在表1中有以下记录:

<date:"2020-06-09 23:10:00", pid: 1, price: 5, tax: 0.65, shipping: 10>
<date:"2020-06-09 13:00:00", pid: 1, price: 10, tax: 1.3, shipping: 12>
<date:"2020-06-10 10:00:00", pid: 2, price: 20, tax: 2.6, shipping: 11>
<date:"2020-06-10 11:00:00", pid: 3, price: 5, tax: 0.65, shipping: 13>

我的查询应该返回一个散列，其中每个键代表一天，值将是唯一 pid 的计数和其余值的总和。所以在这种情况下的响应是:

{
   2020-06-09: {
      pid_count: 1,
      price: 15,
      tax: 1.95,
      shipping: 22
   },
   2020-06-10: {
      pid_count: 2,
      price: 25,
      tax: 3.25,
      shipping: 24
   }
}

我的查询是:

query = <<-SQL
  SELECT TO_CHAR(TO_TIMESTAMP(table1.date), 'YYYY-MM-DD'), COUNT(DISTINCT table1.pid) as pid_count,
  sum(table1.price) as price, sum(table1.tax) as tax, sum(table1.shipping) as shipping from table1
  group by TO_CHAR(TO_TIMESTAMP(table1.date), 'YYYY-MM-DD')
SQL
results = ActiveRecord::Base.connection.execute(query).values

计算是正确的，但问题是这没有将它们正确地分组到散列中。在 results 上调用 .to_h 会给出错误 ArgumentError(错误的数组长度为 0(预期为 2，为 5))

Answer 1

你不能调用 to_h 的原因是因为该方法需要结构:

[
  [key, value],
  [key, value],
  # ...
]

而你的结构是:

[
  [value_1, value_2, value_3, value_4, value_5],
  [value_1, value_2, value_3, value_4, value_5],
  # ...
]

要将它正确地转换成您想要的结构，您必须通过提供一个 block 来告诉 to_h 如何做到这一点。

column_names = %i[pid_count price tax shipping]
hash = results.to_h { |date, *values| [date, column_names.zip(values).to_h] }

这告诉 to_h 调用使用日期作为键，然后将列名称与其他值压缩并将它们转换为散列并将它们用作值。

以上可能会解决问题，但在大多数情况下，无需使用 execute 开始。假设 table1 有一个模型，我建议使用模型方法，而不是像 group 和 select/pluck。对于更复杂的查询，可以为它们提供 SQL 或 Arel。