[Rails] N+1問題と子要素条件検索の両立ができない

Question

前提・実現したいこと

Ruby On Rails 5で社内のみで使う業務管理システムを作成中。
プロジェクトごとにタグが複数ついていて、タグと案件番号と顧客会社名によるプロジェクト検索機能を作りたい。
######テーブルは下記(質問に関係ない情報は省略)
1←多(矢印の向きがbelongs_to)
projects → clients
projects ← project_tags → tags(Active Hash)

[projects]
id(PK) integer
code string (「案件番号」と呼ぶ、「302-5」みたいな数字とハイフンの文字列)
client_id(FK) integer

[clients]
id(PK) integer
company_name integer (株式会社○○など)

project_tags
id(PK) integer
project_id(FK) integer
tag_id(FK) integer

[tags](Active Hash)11個ある
id(PK)
name

######検索機能の詳細
案件番号は前方一致、顧客会社名は部分一致、タグは1,2,3を選択した場合タグIDとして1,2,3のいずれかを含む案件がヒット

発生している問題の例

赤は検索フォームで選択されたtag_id
Project:1(青)とProject: 2(緑)のみヒット

eager_loadでJOINする
↓

projects.id	project_tags.tag_id
1	1
1	2
1	3
1	4
1	5
2	2
2	3
2	4
3	4
3	5
3	6

↓
tag_id: [1,2,3]で絞り込み
↓

projects.id	project_tags.tag_id
1	1
1	2
1	3
2	2
2	3

→ Project:1は4番・5番、Project: 2は4番のタグが消えているためビューで全てタグが表示されない

該当のソースコード

※注意：なぜindexではなくallなのか
メモリの問題もあるので文字通りallを取り出すわけではない。
しかし resourcesのindexを使うと
検索前のURLが/projects
検索後のURLは/projects/search....(←クエリ文字列)
検索フォームの自動補完機能をjQueryで実装するときautocompleteアクションへの相対パスを書くわけだが、検索前と検索後でURLの階層が異なるとそれが使えなくなるため、
検索前のURLを/projects/allとして階層をそろえている。

######views/projects/all.html.erb (検索フォーム部分)

<%= form_with url: search_projects_path, method: :get,  min: 1, local: true do |f| %>
  <%= f.label :code, '案件番号' %>
  <%= f.search_field :code, value: @code, id: 'code_autocomplete' %>
      
  <%= f.label :client, '顧客' %>
  <%= f.search_field :client, value: @company_name, id: 'client_autocomplete' %>
      
  <%= f.collection_check_boxes(:tag_ids, Tag.all, :id, :name, include_hidden: false) do |tag| %>
    <%= tag.label do %>
      <%= tag.check_box %>
        <%= tag.text %>
      <% end %>
    <% end %>    
  <%= f.submit '検索' %>
<% end %>

models/project.rb

  def self.search(code, client_company_name, tag_ids)
    if tag_ids
      Project.eager_load(:project_tags, :client)
             .where('projects.code LIKE ? AND clients.company_name LIKE ?',"#{code}%", "%#{client_company_name}%")
             .where(project_tags: {tag_id: tag_ids})
    else # タグ選択がない場合は検索条件に入れない
      Project.eager_load(:project_tags, :client)
             .where('projects.code LIKE ? AND clients.company_name LIKE ?', "#{code}%", "%#{client_company_name}%")
    end
  end

######controllers/projects_controller.rb

class ProjectsController  < ApplicationController
  def all
    @projects = Project.limit(100).preload(:client, :project_tags)
  end

  def search
    @code         = params[:code]
    @company_name = params[:company_name]
    @projects     = Project.limit(100).eager_load(:client, :project_tags).search(@code, @company_name, params[:tag_ids])
    render action: :all
  end
end

######views/projects/all.html.erb(検索結果表示部分)

<% @projects&.each do |pj| %>      
   <% pj.project_tags&.each do |pt| %>
     <%= pt.tag.name %>
   <% end %>
<% end %>

試したこと

######・案件を絞ったあとでタグは別々にロードする(N+1問題が残る)
一応ヒットした案件のタグはすべて表示される

<% @projects&.each do |pj| %>      
   <% ProjectTag.where(project_id: pj.id)&.each do |pt| %>
     <%= pt.tag.name %>
   <% end %>
<% end %>

補足情報（FW/ツールのバージョンなど）

FW: AWSのcloud9
rails: 5.2.3
DB: sqlite3
OS: Windows10

Answer 1

クエリ一発で解決するのは複雑そうなので、

• 該当するProjectのIDsを返すクエリ
• 上記クエリ結果から、該当するProjectの全タグをeager_loadするクエリ

の2回に分けるのはどうでしょうか？

2回目のクエリについては、サブセット条件を使えばINで効率よく検索してくれそうです。

Comments

[Zray]

なるほど！！回答ありがとうございます！！「条件に合致するIDsを作ってから」という発想にとても納得しました！そうなると顧客条件に合致するClientもIDsを配列に格納してしまえば完全にProjectテーブルのカラムでの絞り込みにできるのでeager_loadではなくpreloadで済みますね。美しい！！
project_ids = ProjectTag.where(tag_id: params[:tag_ids]).group(:project_id)
　　　　　　　　.select(:project_id)
client_ids = Client.where("company_name LIKE ?", "%#{@company_name}%").select(:id)
@projects = Project.where("code LIKE ?", "#{params[:code]}%")
　　　　　　　　　　　　.where(id: project_ids, client_id: client_ids)
　　　　　　　　　　　　.preload(:project_tags, :client)