rubyでオブジェクトの配列から条件一致するもののみを抽出したい

Question

基本的な質問で申し訳ないんですが下記でactive recordで抽出したオブジェクトの配列で条件に合うもののみを再度抽出したいんですがどうですればいいでしょうか？

@plans = Plan.find(plan_ids)
# 上記でidの配列で複数のplanオブジェクトを@plansに入れています
account_user_plans = @plans.select(user_id: account_user.id)
これを特定のuser_idのもののみを抽出したいので上記で記述

selectではないと思うんですがここの適切な書き方ってあるでしょうか？

特定のuser_idの@plansをオブジェクトで抽出したいです。

Answer 1

@plans = Plan.where(id: plan_ids)

account_user_plans = Plan.where(user_id: account_user.id, id: plan_ids)
# account_user_plans = @plans.where(user_id: account_user.id)

もし user has plans, plan belongs_to user となっているなら、

account_user_plans = acount_user.plans.where(id: plan_ids)

Answer 2

いや、、、
find は id の配列を与えることも出来ます。
ただ、plans_by_find = Plan.find(ids) と plans_by_where=Plan.where(id: ids) では結果がやや異なります。
plans_by_find は インスタンスの配列ですが、plans_by_where は リレーションです。

ので、user_id: account_user.id なものを抜き取る方法が異なります。
plans_by_find.find{|plan| plan.user_id == account_user.id }
plans_by_where.find_by(user_id: account_user.id)

select で行うと 中身が一つの配列が帰ってくるので 更に first とかが必要です。
後者も　find_by でなく whereでやれますが、その場合も配列（のようなもの、リレーション）が帰ってくるので firstが必要です。

Answer 3

大前提の知識

findは引数にidを取り、1件のレコードだけを取得します。
配列＝複数レコードを取得する事はできません。
また、該当するレコードがない場合、例外が発生し処理が停止します。

したがって、複数のレコードを取得したい場合findは使えません。
複数レコードを取得したいなら、whereを使います。

それを踏まえて

例えば、複数のplan_idに対応するレコードの検索はこうなります。

plan_ids = [1,2,3]

# SELECT * FROM plans WHERE id IN (1, 2, 3)
@plans = Plan.where(id: plan_ids)

さらにuser_id = account.user_idで絞りたいならAND条件なので、where()を重ねます。

plan_ids = [1,2,3]

# SELECT * FROM plans WHERE id IN (1, 2, 3) AND user_id = 1
# ※account_user.id = 1の場合
@plans = Plan.where(id: plan_ids).where(user_id: account_user.id)

Comments

[you88]

ありがとうございます！