ActiveRecordでscopeをpreloadしたい

この記事は公開されてから1年以上経過しており、最新の内容に追従できていない可能性があります。

ひとつのUserが複数のItemを持っているという構造で、Item.nameを出したい。

class User < ActiveRecord::Base
  has_many :items
end

class Item < ActiveRecord::Base
end

@users = User.where(foo_bar_condition: 1)

@users.each do |user|
  user.items.each do |item|
    item.name
  end
end

このままではUserの数だけItemに対するクエリが出てしまい、パフォーマンスがあまりよろしくない。これはpreload(またはincludes)を使うことで対策ができる。

class User < ActiveRecord::Base
  has_many :items
end

class Item < ActiveRecord::Base
end

@users = User.where(foo_bar_condition: 1).preload(:items)

@users.each do |user|
  user.items.each do |item|
    item.name
  end
end

このとき、Userと同時にItemを WHERE item.user_id IN (?) のようなクエリで取得してくれる。

ここに加えて、Itemをis_activeで絞りたい。便利に使えるようscopeを付けたとする。

class User < ActiveRecord::Base
  has_many :items
end

class Item < ActiveRecord::Base
  scope :active, -> { where(is_active: true) }
end

@users = User.where(foo_bar_condition: 1).preload(:items)

@users.each do |user|
  user.items.active.each do |item|
    item.name
  end
end

するとN+1…N+2になってしまう。Userへのクエリ、Itemをuser_id IN (?)するクエリ、UserごとのItemへのクエリ。

scopeで記述する部分はActiveRecordのクエリになっていて、preloadする時点では知る由がないため、絞り込むことができない。これを解決する方法はいくつか方法あるけど、アソシエーションを設定するのが一番お手軽だった。

class User < ActiveRecord::Base
  has_many :items
  has_many :items_active, -> { active }, class_name: 'Item'
end

class Item < ActiveRecord::Base
  scope :active, -> { where(is_active: true) }
end

@users = User.where(foo_bar_condition: 1).preload(:items_active)

@users.each do |user|
  user.items_active.each do |item|
    item.name
  end
end

アソシエーションの定義で、対象のモデルのscopeを利用できる、というもの。モデルとは異なる名前にした場合はclass_nameの値でどこにマッピングするかを決められる。

ActiveRecord::Associations::ClassMethods, has_many label options

scopeのpreloadどうやるんだろうな〜と思ってActiveRecordの説明みていたら、scopeで作ったものを見ていそうなものがあったので、なるほど〜とやったら動いた。

他にも ActiveRecord::Associations::Preloader を使うという方法があるみたい。

ActiveRecordでScoped preloading - Qiita

これは埋め込み先をわかっていないとできないので、複雑に入りくんだデータをガッとpreloadするのは大変。かつ、使う箇所で scopeの情報(=activeであるitem)が失われるので、もしかしたら理解が難しくなるかもしれない。

@users = User.where(foo_bar_condition: 1)

ActiveRecord::Associations::Preloader.new.preload(
  @users,
  :items,
  Item.active
)

@users.each do |user|
  user.items.each do |item|
    item.name
  end
end

※これはRails6までで、Rails7では記述の仕方が違う。

Remove deprecated private API in ActiveRecord::Associations::Preloader · rails/rails@e3b9779

ActiveRecord::Associations::Preloader.new(
  records: @users,
  associations: :items,
  scope: Item.active
)

なのだけれど、手元で試すとクエリのログがいくつも流れてしまうのでちょっと違うような気がする。どうやるんだろう。。

preloadじゃなくてeager_loadとmergeを使う方法もある。mergeは別のActiveRecordのクエリをくっつけることができる。

ActiveRecord::SpawnMethods#merge

eager_loadの場合、クエリを分けずにJOINしてまとめてデータを取ることになるので、たくさんくっつけるとクエリが大変になる。その上、条件次第、具体的にはeager_loadした先のフィールドで絞るので結果が違うケースが出てくる。これはuser.nameも出すとわかる。

@users = User.where(foo_bar_condition: 1).eager_load(:items).merge(Item.active)
# User LEFT OUTER JOIN Item ON User.id = Item.user_id WHERE User.foo_bar_condition = 1 AND Item.is_active = true みたいなクエリになる。

@users.each do |user|
  user.name # Item.active を持っていないUserが出てこない
  user.items.each do |item|
    item.name
  end
end

サイト案内

運営してるひと: @sters9

最近は Go, Ruby, Rails, Kubernetes, GCP, Datadog あたりをしていますがもっといろいろやりたい!

サイト案内

開発環境の紹介

プライバシーポリシー

tools.gomiba.co

サイト内検索

アーカイブ

2024/09 (3) 2024/07 (1) 2024/06 (3) 2024/05 (1) 2024/04 (7) 2024/03 (4) 2024/01 (3)

2023/12 (1) 2023/11 (3) 2023/10 (1) 2023/09 (1) 2023/08 (2) 2023/05 (4) 2023/04 (4) 2023/03 (4) 2023/02 (2) 2023/01 (1)

2022/12 (2) 2022/11 (4) 2022/10 (3) 2022/09 (2) 2022/08 (4) 2022/07 (5) 2022/06 (4) 2022/05 (9) 2022/04 (8) 2022/03 (10) 2022/02 (21) 2022/01 (8)

2021/12 (11) 2021/11 (1) 2021/10 (4) 2021/09 (2) 2021/08 (1) 2021/07 (2) 2021/06 (5) 2021/05 (10) 2021/04 (1) 2021/03 (8) 2021/02 (12) 2021/01 (8)

2020/05 (2) 2020/04 (2) 2020/02 (2) 2020/01 (1)

2019/12 (3) 2019/11 (2) 2019/10 (5) 2019/09 (3) 2019/07 (6) 2019/06 (4) 2019/04 (3) 2019/01 (2)

2018/12 (6) 2018/10 (4) 2018/09 (6) 2018/08 (7) 2018/07 (16) 2018/06 (7) 2018/05 (7) 2018/04 (5) 2018/03 (3) 2018/02 (10) 2018/01 (6)

2017/12 (8) 2017/11 (6) 2017/10 (10) 2017/09 (12) 2017/08 (12) 2017/07 (3) 2017/06 (1) 2017/01 (4)

2016/12 (5) 2016/10 (3) 2016/09 (1) 2016/07 (2) 2016/06 (1) 2016/04 (1) 2016/02 (1) 2016/01 (2)

2015/12 (1) 2015/10 (1) 2015/09 (3) 2015/06 (1) 2015/01 (1)

2014/08 (2) 2014/07 (3) 2014/05 (1) 2014/01 (7)

2013/12 (2) 2013/11 (4) 2013/10 (1) 2013/09 (1) 2013/08 (3) 2013/07 (4) 2013/06 (5) 2013/05 (2) 2013/04 (7) 2013/03 (1)