如何对现有记录（历史数据）进行 Rails 迁移？

Question

我有一个模型 (Model_A) 有两列（Column_A 和 Column_B）

我的模型里面有下面的方法，两个根据column_A生成column_B的值

Class Model_A < ActiveRecord::Base
  before_save :set_column_B

  def set_column_B
      self.column_B = get_value_from_column_A
  end

  def get_value_from_column_A
    # manipulate data from A to get value
    column_A.join(,) 
  end

end

我知道我必须创建一个新的 column_B 作为迁移的第一步，但是如何更改数据库中所有预先存在的记录以在 column_B 中具有正确的值？ 我应该为此编写脚本或 Rake 任务，还是有办法通过迁移来完成？

Answer 1

如果您有很多记录，则循环遍历所有记录是个坏主意。假设您有 200 万条记录，如果您采用 ActiveRecord 方式，更新列然后执行回调和验证将需要很长时间。

如果你可以使用 SQL 本身来完成，你可以使用 SQL 方式更新所有记录

类似于 Spree 的这种迁移

class UpdateNameFieldsOnSpreeCreditCards < ActiveRecord::Migration
  def up
    if ActiveRecord::Base.connection.adapter_name.downcase.include? "mysql"
      execute "UPDATE spree_credit_cards SET name = CONCAT(first_name, ' ', last_name)"
    else
      execute "UPDATE spree_credit_cards SET name = first_name || ' ' || last_name"
    end
  end

  def down
    execute "UPDATE spree_credit_cards SET name = NULL"
  end
end

Answer 2

在迁移中，您可以执行以下操作：

 class AddColummBToModelA < ActiveRecord::Migration
  def up
    unless column_exists? :budgets, :column_b
      add_column :budgets, :column_b ......
      ModelA.find_each {|x| x.save! }
    end

  end

  def down
   remove_column :budgets, :column_b if column_exists? :budgets, :column_b
  end
end