Rails中用has_and_belongs_to_many实现好友模型及查询特定用户好友

看起来你要做的是用户之间的双向好友关系对吧？用has_and_belongs_to_many来实现完全可行，不过这里因为是User模型的自关联多对多（毕竟好友本质也是用户），所以需要稍微调整关联配置。下面是具体实现步骤和获取好友的方法：

1. 配置User模型关联 #

修改你的app/models/user.rb，添加自关联的多对多配置：

class User < ApplicationRecord
  # 自关联多对多：用户拥有多个好友，好友也属于用户
  has_and_belongs_to_many :friends,
                          class_name: "User",  # 关联的目标模型还是User
                          join_table: "friendships",  # 指定连接表名称
                          foreign_key: "user_id",  # 当前用户在连接表中的外键
                          association_foreign_key: "friend_id"  # 好友用户在连接表中的外键
end

2. 确保连接表迁移文件正确 #

你的20180309142447_create_friendships_table.rb迁移文件需要创建无主键的连接表（HABTM模式不需要主键），同时添加索引避免重复关系并优化查询：

class CreateFriendshipsTable < ActiveRecord::Migration[5.1]
  def change
    # 创建无主键的连接表，强制外键非空
    create_table :friendships, id: false do |t|
      t.integer :user_id, null: false
      t.integer :friend_id, null: false
    end

    # 添加复合唯一索引，防止同一个好友关系被重复创建
    add_index :friendships, [:user_id, :friend_id], unique: true
    # 添加单字段索引，优化反向查询好友列表的性能
    add_index :friendships, :friend_id
  end
end

执行迁移使配置生效：

rails db:migrate

3. 获取特定用户的所有好友 #

当关联配置完成后，获取特定用户的好友就非常简单了。假设你有一个用户实例@user，直接调用：

@user.friends

这会返回一个ActiveRecord集合，包含该用户的所有好友用户实例，你可以像操作普通User集合一样进行筛选、排序等操作。

额外提示：处理双向好友关系 #

默认情况下，执行@user.friends << @another_user只会在friendships表中添加一条单向记录，这意味着@another_user.friends不会显示@user。如果需要双向好友关系，可以在User模型中添加一个辅助方法：

class User < ApplicationRecord
  # ... 之前的关联代码 ...

  def add_friend(user)
    friends << user
    user.friends << self
    save
  end
end

调用@user.add_friend(@another_user)就会同时创建两条关联记录，保证双方的好友列表都能看到彼此。

内容的提问来源于stack exchange，提问作者Net Sim