如何在 Rails 中建立相互友谊的模型

Question

我知道之前在 Stack Overflow 上已经问过这个问题，但答案并没有以我可以解释的方式为我做。我的一般方法受到本教程的启发。

我想要做的是创建一个非常简单的模型来为用户添加好友，通过一条记录在两端创建等效的好友关系。

在数据库级别，我只有一个“friendships”表，其中只有一个 user_id、一个friend_id 和一个 is_pending 布尔列。

在 user.rb 中，我将关系定义为：

has_many :friendships
has_many :friends, through: :friendships

在friendship.rb 中，我将这种关系定义为：

belongs_to :user
belongs_to :friend, :class_name => 'User'

如果我添加好友，我可以访问如下：

> a = User.first
> b = User.last
> Friendship.new(a.id, b.id)
> a.friends
=> #<User b>

这是完美的，但我想要的是也能够像这样朝另一个方向前进：

> b.friends

不幸的是，按照原样定义的关系，我得到了一个空集合。运行的 SQL 显示它正在搜索 user_id = b.id。我如何指定它还应该搜索friend_id = b.id？

Answer 1

这也可以通过单个has_many :through关联和一些查询调整来实现：

# app/models/friendship.rb

class Friendship < ApplicationRecord
  belongs_to :user
  belongs_to :friend, class_name: 'User'
end

 

# app/models/user.rb

class User < ApplicationRecord
  has_many :friendships,
    ->(user) { FriendshipsQuery.both_ways(user_id: user.id) },
    inverse_of: :user,
    dependent: :destroy

  has_many :friends,
    ->(user) { UsersQuery.friends(user_id: user.id, scope: true) },
    through: :friendships
end

 

# app/queries/friendships_query.rb

module FriendshipsQuery
  extend self

  def both_ways(user_id:)
    relation.unscope(where: :user_id)
      .where(user_id: user_id)
      .or(relation.where(friend_id: user_id))
  end

  private

  def relation
    @relation ||= Friendship.all
  end
end

 

# app/queries/users_query.rb

module UsersQuery
  extend self

  def friends(user_id:, scope: false)
    query = relation.joins(sql(scope: scope)).where.not(id: user_id)

    query.where(friendships: { user_id: user_id })
      .or(query.where(friendships: { friend_id: user_id }))
  end

  private

  def relation
    @relation ||= User.all
  end

  def sql(scope: false)
    if scope
      <<~SQL
        OR users.id = friendships.user_id
      SQL
    else
      <<~SQL
        INNER JOIN friendships
          ON users.id = friendships.friend_id
          OR users.id = friendships.user_id
      SQL
    end
  end
end

它可能不是其中最简单的，但肯定是最干燥的。它不使用任何回调、附加记录和关联，并保持关联方法完整，包括隐式关联创建：

user.friends << new_friend

通过gist.github.com

Answer 2

本文展示如何设置互惠关系：Rails 中的双向关系

它展示了如何使用after_create和after_destroy插入模拟互惠关系的附加关系。这样，您的连接表中的记录就会加倍，但您可以灵活地使用a.friendsandb.friends并看到两者正确包含彼此。

使其适用于您的模型：

class Person < ActiveRecord::Base
  has_many :friendships, :dependent => :destroy
  has_many :friends, :through => :friendships, :source => :person
end

class Friendship < ActiveRecord::Base
  belongs_to :person, :foreign_key => :friend_id
  after_create do |p|
    if !Friendship.find(:first, :conditions => { :friend_id => p.person_id })
      Friendship.create!(:person_id => p.friend_id, :friend_id => p.person_id)
    end
  end
  after_update do |p|
    reciprocal = Friendship.find(:first, :conditions => { :friend_id => p.person_id })
    reciprocal.is_pending = self.is_pending unless reciprocal.nil?
  end
  after_destroy do |p|
    reciprocal = Friendship.find(:first, :conditions => { :friend_id => p.person_id })
    reciprocal.destroy unless reciprocal.nil?
  end
end

我已经在几个项目中成功地使用了这种方法，而且非常方便！