Rails中用RequestBulk表ID替换EntityRequest表from_id的Rake任务问题

我有4张数据表及对应模型：Activity、Census、RequestBulk、EntityRequest，相关逻辑如下：

• RequestBulk用于镜像Activity和Census的所有记录，两表独有字段存入RequestBulk的metadata字段，request_type字段标记对应原表类型（'Activity'或'Census'）
• 创建Activity/Census记录时，会同步生成EntityRequest，其from_id字段存储对应Activity/Census的ID
• 需要编写Rake任务，将EntityRequest的from_id替换为对应的RequestBulk记录ID

之前尝试的代码会把所有EntityRequest的from_id更新为最后一条RequestBulk的ID，完全不符合预期：

RequestBulk.all.each do |rb|
  if rb.request_type == 'BulkActivity' # 此处存在笔误，应为'Activity'
    EntityRequest.update(from_id: rb.id)
  end
end

问题核心：EntityRequest.update(from_id: rb.id)会更新所有EntityRequest记录，循环过程中每次都会覆盖之前的更新，最终所有记录的from_id都会变成最后遍历到的RequestBulk的ID，完全没有建立对应关系。

方案1：基于字段匹配批量更新（高效通用） #

假设RequestBulk表有original_id字段存储对应原Activity/Census的ID（镜像表必备关联字段），用批量更新精准匹配：

namespace :entity_requests do
  desc "将EntityRequest的from_id替换为对应RequestBulk的ID"
  task migrate_from_id: :environment do
    # 处理Activity类型的RequestBulk
    RequestBulk.where(request_type: 'Activity').find_in_batches do |batch|
      batch.each do |rb|
        # 定位原Activity对应的EntityRequest，批量更新from_id
        EntityRequest.where(parentable_type: 'Activity', from_id: rb.original_id).update_all(from_id: rb.id)
      end
    end

    # 处理Census类型的RequestBulk
    RequestBulk.where(request_type: 'Census').find_in_batches do |batch|
      batch.each do |rb|
        EntityRequest.where(parentable_type: 'Census', from_id: rb.original_id).update_all(from_id: rb.id)
      end
    end

    puts "迁移完成！"
  end
end

说明：

• find_in_batches避免一次性加载所有RequestBulk记录，防止内存溢出
• 通过parentable_type（对应原模型类型）和原from_id（原记录ID）精准定位目标EntityRequest
• update_all直接生成SQL批量更新，比单条update性能提升显著

方案2：利用模型关联实现（贴合现有配置） #

如果给Activity/Census模型添加与RequestBulk的反向关联（比如在Activity模型中添加：has_one :request_bulk, -> { where(request_type: 'Activity') }, foreign_key: :original_id），可借助现有has_many :entity_requests, as: :parentable关联实现：

namespace :entity_requests do
  desc "将EntityRequest的from_id替换为对应RequestBulk的ID"
  task migrate_from_id: :environment do
    Activity.find_in_batches do |batch|
      batch.each do |activity|
        next unless activity.request_bulk.present?
        # 直接通过关联获取对应EntityRequest并更新
        activity.entity_requests.update_all(from_id: activity.request_bulk.id)
      end
    end

    Census.find_in_batches do |batch|
      batch.each do |census|
        next unless census.request_bulk.present?
        census.entity_requests.update_all(from_id: census.request_bulk.id)
      end
    end

    puts "迁移完成！"
  end
end

说明：

• 利用现有关联关系，代码更贴合业务模型设计
• 先检查对应RequestBulk是否存在，避免空指针异常
• 同样用批量操作保证性能

1. 执行前必须备份数据库，数据迁移操作不可逆
2. 先在测试环境验证结果，确认无误后再到生产环境执行
3. 如果RequestBulk存储原记录ID的字段不是original_id，替换为你实际使用的字段名（比如activity_id/census_id）
4. 原错误代码中的'BulkActivity'是笔误，需改为实际的'Activity'

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