如何在 Rails 迁移中将列（包含内容）移动到另一个表？

Question

我需要将一些列从一个现有表移动到另一个。我如何使用 Rails 迁移来做到这一点？

class AddPropertyToUser < ActiveRecord::Migration
  def self.up
    add_column :users, :someprop, :string
    remove_column :profiles, :someprop
  end

  def self.down
    add_column :profiles, :someprop, :string
    remove_column :users, :someprop
  end
end

上面只是创建了新的列，但是值是空的...

我想避免登录数据库手动更新表。

如果有以编程方式移动列值的方法，性能特征是什么？它会逐行更新，还是有办法批量更新？

Answer 1

我会将此作为三个迁移或三部分迁移。第一部分是添加列，第二部分是复制数据，第三部分是删除列。

听起来中间步骤就是你要问的，你可以在 ruby​​ 中通过循环所有用户并设置属性来做到这一点，如下所示：

Users.each do |user|
   user.someprop = user.profile.some_prop
   user.save
end 

我不喜欢这种方式，因为它非常慢。我建议像这样执行原始 sql：

execute "UPDATE users u, profiles p SET u.someprop=p.someprop WHERE u.id=p.user_id"

这些都假设您的个人资料/用户关联，如果我假设错误，您可以调整。

Answer 2

我最终使用了这个迁移（经过测试，它可以工作，并且成功回滚）：

class AddPropertyToUser < ActiveRecord::Migration
  def self.up
    add_column :users, :someprop, :string
    execute "UPDATE users u, profiles p SET u.someprop = p.someprop WHERE u.id = p.user_id"
    remove_column :profiles, :someprop
  end

  def self.down
    add_column :profiles, :someprop, :string
    execute "UPDATE profiles p, users u SET p.someprop = u.someprop WHERE p.user_id = u.id"
    remove_column :users, :someprop
  end
end

我喜欢它，因为它避免了对大型数据库的逐行更新。

Answer 3

您可以使用 update_all 和/或 find_each 避免硬编码的数据库特定 sql 语句

Answer 4

对我（postgreSQL 9.1）来说，RAW SQL 不起作用。我已经改了：

" UPDATE users u
  SET someprop = (SELECT p.someprop
                  FROM profiles p
                  WHERE u.id = p.user_id );"

Answer 5

该语法不适用于更高版本的 Postgres。对于 @Eero 对 Postges 9.4.5 的更新答案，请执行以下操作：

class AddPropertyToUser < ActiveRecord::Migration
  def self.up
    add_column :users, :someprop, :string
    execute "UPDATE users u SET someprop = (SELECT p.someprop FROM profiles p WHERE u.id = p.user_id);"
    remove_column :profiles, :someprop
  end

  def self.down
    add_column :profiles, :someprop, :string
    execute "UPDATE profiles p SET someprop = (SELECT u.someprop FROM users u WHERE p.user_id = u.id);"
    remove_column :users, :someprop
  end
end

Answer 6

以下UPDATE 语法适用于最近的 Postgres 版本并避免子查询：

class MoveSomePropertyToUser < ActiveRecord::Migration
  def self.up
    add_column :users, :some_property, :string
    execute "UPDATE users u SET some_property = p.some_property FROM profiles p WHERE u.id = p.user_id;"
    remove_column :profiles, :some_property
  end

  def self.down
    add_column :profiles, :some_property, :string
    execute "UPDATE profiles p SET some_property = u.some_property FROM users u WHERE p.user_id = u.id;"
    remove_column :users, :some_property
  end
end

Answer 7

这是我在项目中所做的：-

class MoveColumnDataToUsersTable < ActiveRecord::Migration[5.1]
  def up
    add_column :users, :someprop, :string
    User.find_each do |u|
        Profile.create!(user_id: u.id, someprop: someprop)
    end
    remove_column :profiles, :someprop
  end

  def down
    add_column :profiles, :someprop, :someprop_data_type
    Profile.find_each do |p|
      User.find_by(id: p.user_id).update_columns(someprop: p.someprop)   
    end
    Profile.destroy_all
  end
end