指定されたユーザーをログイン時別画面へ

Question

###実現したいこと
現在、すでにログインしているユーザーは自動でPost/indexへ飛ぶようになっています。
しかし、下記のcheck_userに該当するユーザーは、別画面へ飛ぶ設定をしたいです。
###コード

class ApplicationController < ActionController::Base
  before_action :set_current_user

  def set_current_user
    @current_user = User.find_by(id: session[:user_id])
    @check_user = User.where(id: [1, 2]) #別画面へ送りたいユーザー
  end

  def authenticate_user #session_idが無ければログイン画面へ
    if @current_user == nil
      redirect_to("/login")
    end
  end

  def forbid_login_user
    if @current_user #session_idがある場合
      if @check_user
        redirect_to("/check") #@check_userに該当すればcheck画面へ
      else
        redirect_to("/posts/index") #ログインは済みユーザーはindexへ
      end
    end
  end
end

###現状
@check_userで指定されたユーザーのみならず、全ユーザーがredirect_to("/check") に飛ばされます。
また、if @ng == @current_user　と書き換えた場合、全ユーザーがredirect_to("/posts/index") に飛ばされます。
お分かりの方、ぜひ宜しくお願い致します。

Comments

[KazuSaka]

If @check_user.find_by(id: @current_user.id)
な気がします。
@check_userは必ず、id=1,2のユーザーが存在するはずなので。
間違ってたらすいません。

Answer 1

@check_user = User.where(id: [1, 2])は必ず結果が返りますから, current_user が誰であろうとif @check_userが成立します。

if @check_user　を
if [1,2].include?(@current_user.id)

としてみてください

Comments

[tomtom1]

ご回答ありがとうございます！！
if [1,2].include?(@current_user.id)で出来ました！

[1,2]に記載idをベタ打ちではなく、今回set_current_userで@check_userを定義しているようにはできるのでしょうか？

[winterboum]

check_user の定義がはっきりしていますか？
していたら @check_user_id = User.where(定義).pluck(:id)

id が1,2であること以外ないとすると、どこかで1,2をベタ打ちすることになります。
その場合のお薦めはどこかで定数として　CheckUserId = [1,2] としておきif CheckUserId.include?(@current_user.id)
ですね

[tomtom1]

CheckUserId = [1,2] としておきif CheckUserId.include?(@current_user.id)
上記で定数を書くことが出来ました！ありがとうございます。