在 Rails 迁移中迁移数据

Question

假设您有“lineitems”，并且您曾经根据 line_item 定义“make”。

最终您意识到 Make 可能应该位于其自己的模型上，因此您创建了一个 Make 模型。

然后，您想要从 line_items 表中删除 make 列，但对于每个带有 make 的 line_item，您想要 find_or_create_by(line_item.make)。

我如何在 Rails 迁移中有效地做到这一点？我很确定我可以为每个 line_item 运行一些简单的 find_or_create_by 但我担心后备支持，所以我只是在这里发布此内容以获取任何提示/建议/正确的方向。

谢谢！

Answer 1

我想您应该在删除列之前检查 Make.count 是否等于 lineitems 中的唯一制造总数，如果不相等则引发错误。由于迁移是事务性的，如果它崩溃了，架构不会改变，迁移也不会被标记为已执行。因此，你可以这样做：

class CreateMakesAndMigrateFromLineItems < ActiveRecord::Migration
  def self.up
    create_table :makes do |t|
      t.string :name
      …
      t.timestamps
    end

    makes = LineItem.all.collect(:&make).uniq
    makes.each { |make| Make.find_or_create_by_name make }
    Make.count == makes.length ? remove_column(:line_items, :make) : raise "Boom!"

  end

  def self.down
    # You'll want to put logic here to take you back to how things were before. Just in case!
    drop_table :makes
    add_column :line_items, :make
  end
end

Answer 2

您可以在迁移中放入常规的 ruby​​ 代码，这样您就可以创建新表，在旧模型上运行一些代码，将数据移动到新模型中，然后从原始模型中删除列。这甚至是可逆的，因此您的迁移仍然可以在两个方向上进行。

因此，根据您的情况，创建 Make 表并将 make_id 添加到 lineitem。然后，对于每个订单项，find_or_create 以及 lineitem 上的 make 列，将返回的 id 设置为 lineitem 上的新 make_id。完成后，从 lineitem 表中删除旧的 make 列。