如何用lib/dev:fake在Followship表生成500条user_id≠following_id的不重复随机记录？

我来帮你搞定这个需求！要生成满足「user_id≠following_id」且完全无重复的Followship记录，核心要把控两个关键点：排除自我关注和确保(user_id, following_id)组合唯一。下面是修复后的完整实现方案，附详细解释：

修复后的Rake任务代码 #

把这段代码放到你的lib/tasks/dev.rake文件中：

namespace :dev do
  desc "Generate 500 unique Followship records where user_id != following_id"
  task fake: :environment do
    # 一次性拉取所有用户ID，减少数据库查询次数
    user_ids = User.pluck(:id)
    
    # 先检查有没有用户，避免无意义的循环
    if user_ids.empty?
      puts "Error: No users found in the database! Please create some users first."
      exit 1
    end

    followships = []
    unique_pairs = Set.new

    # 先导入已存在的Followship配对，避免生成重复记录（可选但推荐）
    existing_pairs = Followship.pluck(:user_id, :following_id).to_set
    unique_pairs.merge(existing_pairs)

    # 生成足够的有效配对
    while unique_pairs.size < 500
      user_id = user_ids.sample
      following_id = user_ids.sample

      # 跳过自我关注和已存在的配对
      next if user_id == following_id || unique_pairs.include?([user_id, following_id])

      unique_pairs.add([user_id, following_id])
      followships << Followship.new(
        user_id: user_id,
        following_id: following_id
        # 如果有其他必填字段，在这里补充随机值，比如 created_at: Time.now
      )
    end

    # 批量插入，大幅提升效率
    Followship.insert_all(followships.map(&:attributes))

    puts "✅ Successfully created #{followships.size} unique Followship records!"
  end
end

关键优化点说明 #

1. 用Set管理唯一配对
   Ruby的Set会自动去重，我们把每一组(user_id, following_id)存入Set，就能确保不会生成重复的关注关系。

2. 提前拉取所有用户ID
   一次性从数据库取出所有用户ID存入数组，避免在循环中反复查询数据库，节省性能开销。

3. 跳过无效配对
   在生成随机ID后直接判断：如果是自我关注（user_id=following_id）或者该配对已经存在，就跳过这次循环，继续生成下一组。

4. 批量插入数据
   用insert_all一次性插入所有记录，比循环调用save高效太多，尤其适合生成大量数据的场景。

5. 兼容已有数据（可选）
   代码中加入了导入已有Followship配对的逻辑，这样新生成的记录不会和数据库里已有的重复，如果你不需要这个功能，可以删掉对应的两行代码。

注意事项 #

• 确保数据库中有足够多的用户：如果用户数量太少（比如只有3个用户），最多能生成3×2=6条有效记录，这时候循环会一直跑下去。所以代码里加了用户检查，避免这种情况。
• 如果Followship模型有其他必填字段（比如created_at、updated_at之外的字段），记得在Followship.new里补充对应的随机值。

运行任务的命令还是：rails dev:fake，执行完成后就能得到恰好500条符合要求的记录啦！

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