Twitterみたいなユーザ同士をフォローする仕組みを実装するにあたり、ActiveRecordでN:Nの自己結合を実装するのにちょっとハマったので書いておきます。
前提
- usersテーブルがある
- User同士のフォロー関係を登録するfollowsテーブルがある
具体的なmigrationは以下。
20130825021654_create_users.rb
class CreateUsers < ActiveRecord::Migration
def change
create_table :users do |t|
t.string :name, null: false
t.string :email
# :
# :
t.timestamps
end
end
end
20140303134930_create_follows.rb
class CreateFollows < ActiveRecord::Migration
def change
create_table :follows do |t|
t.integer :from_user_id, null: false
t.integer :to_user_id, null: false
t.timestamps
t.index :from_user_id
t.index :to_user_id
t.index [:from_user_id, :to_user_id], unique: true
end
end
end
やりたいこと
下記のようなインターフェイスで実装したいとします。
User.find(1).following # フォローしているUserの配列が返る
User.find(1).followed # フォローされているUserの配列が返る
ハマりポイント
- リレーションがN:N
- Railsのデフォルトでは
user_idを結合に使うが、今回は同じテーブルを結合するので使えない
実装
まずUserからFollow、FollowからUserに1:Nのリレーションをはります。
user.rb
has_many :follows, foreign_key: :from_user_id
follow.rb
belongs_to :from_user, class_name: User, foreign_key: :from_user_id
こうすると、 User.find(1).follows でFollowが、 Follow.find(1).from_user でフォロー元のUserというリレーションができます。次に、これら1:Nのリレーションを結合する設定をUserに追加します。
user.rb
has_many :follows, foreign_key: :from_user_id
has_many :following, through: :follows, source: :to_user # ← 追加
これで、 User.find(1).following でフォロー先のUserの配列が取得できるようになりました。followed側も同様に実装できます。
最終形
最終的に下記のようになりました。条件の異なる2つの has_many :follows が必要になるのでリネームしました。
user.rb
has_many :follows_from, class_name: Follow, foreign_key: :from_user_id, dependent: :destroy
has_many :follows_to, class_name: Follow, foreign_key: :to_user_id, dependent: :destroy
has_many :following, through: :follows_from, source: :to_user
has_many :followed, through: :follows_to, source: :from_user
follow.rb
belongs_to :from_user, class_name: User, foreign_key: :from_user_id
belongs_to :to_user, class_name: User, foreign_key: :to_user_id
おまけ
下記のようなインターフェイスを追加したくなりました。
user.rb
user.following?(User.find(1))
user.followed_by?(User.find(1))
最初は下記のように any? にブロックを渡して評価していました。
user.rb
def following?(user)
self.following.any? { |u| u.id == user.id }
end
しかしこれでは普通にSELECT文が走って全件インスタンス化しているようだったので下記のようにしてみたところ、 SELECT COUNT(*) が走るようになったのでこちらのほうがパフォーマンスがよいでしょう。
user.rb
# 指定ユーザをフォローしているかどうかを返す
def following?(user)
self.following.where(id: user.id).any?
end
# 指定ユーザにフォローされているかどうかを返す
def followed_by?(user)
self.followed.where(id: user.id).any?
end