Rails ActiveRecordでgroup_by countによる集計結果をrelationとして取得する

12875 ワード
Rails Rails テキストリンク

先に結論だけ

countメソッドを利用するとhashが返却されるため
selectメソッド内でmysqlのcount処理を記述

articles =
  Article
    .joins(:comments)
    .select('articles.id, articles.category_id, count(comments.id) as comments_count')
    .group(:id)
    .order('comments_count desc')

はじめに

RailsのActiveRecordを通してmysqlでcountした結果を取得しようと思いました。
その際、countメソッドを利用すると返却値がhashとなってしまったため
ActiveRecord::relationで取得する方法を模索します。

前提

ブログサイトをイメージして、以下のようなデータモデルを想定します。

データモデル

各テーブルの説明

記事(articles)

ブログ記事を表します。

カテゴリ(categories)

記事に付与されるカテゴリです。
この投稿記事だとRailsのようなものです。
モデルをシンプルにするために、記事はカテゴリを一つだけ持つことにします。

コメント(comments)

記事に付与されるコメントです。
一つの記事に対して複数のコメントがひもづきます。

データ例

categories

id name
1 音楽
2 プログラミング
3 旅行

articles

id category_id title body
1 1 6月に言ったライブ ...
2 2 rubyについて思うこと ...
3 1 6月に聞いた新譜一覧 ...
4 2 sidekiqを触ってみた ...
5 2 enumerableを理解する ...
6 3 夏休みフランス旅行 ...

comments

id article_id body user_id
1 1 いいね 1
2 2 なるほど 2
3 1 面白い 4
4 4 深いなぁ 5
5 3 うける 9
6 3 ナイス 10
7 1 勉強になる 8

今回取得したいデータ

概要

ActiveRecordを通じて 各記事ごとのコメント数ランキング を取得します。
また、その際に

  • 記事タイトル(articles.title)
  • カテゴリ名(categories.name)

も取得します。

必要データ

id category_name title comment_count
1 音楽 6月に言ったライブ 3
3 音楽 6月に聞いた新譜一覧 2
2 プログラミング rubyについて思うこと 1
4 プログラミング sidekiqを触ってみた 1

こんなイメージです。

まずは: count・sizeメソッドを使う

ActiveRecordを通じてcount文を発行するメソッドに countやsizeがあります。(どちらを使うべきかは他の記事で多々挙げられているので、そちらを参考のこと)
今回はcountメソッドを用いて集計をしてみます。

counts = 
  Article
    .joins(:comments)
    .group(:id)
    .order('count_all desc')
    .count

発行されるクエリは以下の通りです。

SELECT  
  COUNT(*) AS count_all, 
  `articles`.`id` AS articles_id 
FROM `articles` 
INNER JOIN `comments` 
  ON `comments`.`article_id` = `articles`.`id` 
GROUP BY `articles`.`id` 
ORDER BY count_all desc

返却されるデータは以下の通りです。

{ 1=>3, 3=>2, 2=>1, 4=>1 }

ActiveRecord::relationではなくhashが返却されてしまったため、
記事タイトルおよびカテゴリ名を取得するには、
記事リレーションを取得するクエリを再発行することが必要になります。

articles = Article.find(counts.map{|id, count| id})

つぎに: selectメソッドを使う

あまり綺麗な方法ではないですが、
countメソッドを使う代わりにselectメソッドを用いて
明示的にselect文内で集約処理を実現します。

articles =
  Article
    .joins(:comments)
    .select('articles.id, articles.category_id, count(comments.id) as comments_count')
    .group(:id)
    .order('comments_count desc')

発行されるクエリは以下の通りです。

SELECT  
  articles.id, 
  articles.category_id, 
  count(comments.id) as comments_count 
FROM `articles` 
INNER JOIN `comments` 
  ON `comments`.`article_id` = `articles`.`id` 
GROUP BY `articles`.`id` 
ORDER BY comments_count desc

返却されるデータは以下の通りです。

[
  #<Article: id: 1, category_id: 1>,
  #<Article: id: 3, category_id: 1>,
  #<Article: id: 2, category_id: 2>,
  #<Article: id: 4, category_id: 2>
]

Articleモデルのrelationが返ってきました。

コメント数を取得する場合は

articles[0].comments_count
=> 3

カテゴリ名を取得する場合は

articles[0].category.name
=> "音楽"

relationとして取得することができるようになりました。

あなたの猫が太っていることを証明する方法
[リンクリスト](Link List)の2つのソートリストをマージ