いいね機能においてN+1問題を解決したい

Question

いつもお世話になっています。
Railsでいいね機能を実装したのですが、いいねの部分でN+1問題が解決できないので、質問致します。

現在は

• Userモデル
• Tweetモデル
• Likeモデル

の３つのモデルを作っており、User→Tweet→Likeのように関連付けをしています。

そして、tweets_controller.rbは以下の様に書いています。

//tweets_controller.rb
  def index
    @tweets = Tweet.all.includes(:user)
  end

そして、tweets/index内に以下の記述をしていて、

<% @tweets.each do |tweet| %>
省略
<%= render partial:'likes/like',locals:{tweet:tweet} %>
<% end %>

likes/_like.html.erb内には、以下の様に書いています。

<% if user_signed_in? %>
  <% if tweet.likes.find_by(user_id:current_user.id) %> //この行がN+1問題の原因？
    <%= button_to [like, tweet.likes.find_by(user_id:current_user.id)], method: :delete do %>
    いいね済
    <% end %>
  <% else %>
    <%= button_to tweet_likes_path(tweet)  do %>
     いいね
    <% end %>
  <% end %>
<% else %>
ログインするといいねができます。
<% end %>

ログを見ると、おそらく２行目の「<% if tweet.likes.find_by(user_id:current_user.id) %>」がN+1問題を引き起こしていると思うのですが、何か解決策はあるのでしょうか？

よろしくお願い致します。

Answer 1

find_byによるN+1問題の解決法が分からないので、妥協案として以下の方法を採用しました。

（https://qiita.com/strviola/items/5e509b2102556eb7c082の記事がヒントになりそうですが、まだ理解ができていないです）

まず、いいねの部分テンプレートである、likes/_like.html.erbを以下の様に訂正しました。

<% if user_signed_in? %>

    <%= button_to tweet_likes_path(tweet)  do %>
     いいね
    <% end %>
<% else %>
ログインするといいねができます。
<% end %>

find_byメソッドを廃止して、「いいねをしていない場合はいいねボタン（create）を表示して、いいねをしている場合は、いいね済み（destroy）と表示させるのを止めました。

そして、likes_contorollerに以下を追加しました。

def create 
@tweet = Tweet.find(params[:tweet_id])
return if @tweet.likes.find_by(user_id:current_user.id)//この一行を追加
like.create(省略
end

これで一人一回いいねができる、と言う実装をしました。

Answer 2

おそらく、LikeをロードしていないためN+1問題が起きているのではと思います。
UserをロードするタイミングでLikeも一緒にロードしてあげれば解決するかと。

    @tweets = Tweet.all.includes([:user, :like])

Comments

[退会済みユーザー]

ご回答ありがとうございます。
試しにやってみましたが、N+１問題を解決できませんでした。
やはり、find_byが呼ばれる毎に、クエリが発生している気がします。