Question

我有一个用ActiveRecord查询的sql数据库。（ActiveRecord的版本是4.2.5.1）

我的查询如下：

User.
  select('COUNT (DISTINCT user_id) as user_count').
  select("date_trunc('month', created_at) as month").
  select('city').
  where('created_at > ?', 1.year.ago.utc.to_s(:db)).
  group('month, city').
  to_a.map do |row|
  {
    user_count: row.user_count,
    month: row.month,
    city: row.city
  }

我得到的结果如下：

[
    [  0] {
        :user_count => 165,
             :month => 2015-09-01 00:00:00 UTC,
              :city => "Paris"
    },
    [  1] {
        :user_count => 33,
             :month => 2015-09-01 00:00:00 UTC,
              :city => "Berlin"
    },
    ...
]

这很好。

但是，使用此查询：

User.
  select('COUNT (DISTINCT user_id) as user_count').
  select("date_trunc('month', created_at) as month").
  select('city').
  where('created_at > ?', 1.year.ago.utc.to_s(:db)).
  group('month, city').
  to_a

我得到的结果如下：

[
    [  0] #<User:0x007facebac3560> {
           :user_id => nil,    # wtf is this
              :city => "Paris"
                               # where is the month?
    },
    [  1] #<User:0x007facebac33a8> {
           :user_id => nil,
              :city => "Berlin"
    },
    ...
]

哪个不好。

为什么即使选择了我想要的行，我也必须手动映射结果？

理想情况下，我想要的是：

User.
  select(...).
  where(...).
  group(...).
  to_array_of_hashes_with_the_keys_i_selected

请注意，如果我的查询是自定义查询，则to_a执行我想要的操作，例如：

ActiveRecord::Base.connection.execute(<<~SQL
  SELECT COUNT (DISTINCT user_id) as user_count, date_trunc('month', created_at) as month,  city
  FROM users
  WHERE created_at > '2015-09-01 13:28:40'
  GROUP BY month, city
SQL
).to_a # this is fine

Answer 1

从How to convert activerecord results into a array of hashes，我可以做到：

User.
  select(...).
  where(...).
  group(...).
  as_json

我得到了

[
    [  0] {
           "user_id" => nil,
              "city" => "Paris",
        "user_count" => 165,
             "month" => 2015-09-01 00:00:00 UTC
    },
    ...

不完全是我选择的，但足够接近。

.as_json.map{|e|e.except('user_id')}给了我想要的东西

使用ActiveRecord获取与所选值对应的哈希数组

1 个答案: