Rails迁移：在创建另一个表的条目时更新现有记录

Question

Rails迁移：在创建另一个表的条目时更新现有记录

sqlruby-on-railsmigrationsql-updateinsert-update

3

我正在尝试为我的Rails应用程序设置Beta邀请。我有一个邀请模型和一个用户模型。

用户有一个invitation_id。我正在验证用户必须拥有invitation_id，并且它应该是唯一的。我想编写一个迁移，以便我可以更新现有用户的invitation_id。

这是我的邀请模型：

# Table name: invitations
#
#  id              :integer         not null, primary key
#  sender_id       :integer
#  recipient_email :string(255)
#  token           :string(255)
#  sent_at         :datetime
#  created_at      :datetime
#  updated_at      :datetime

为了保证唯一性，我将不得不在邀请表中创建记录，并将相应的ID分配给用户记录。

有人能否建议相应的迁移操作？

提前感谢！

- Nirav Shah

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Chris Barretto · Accepted Answer

如果您已经完成了邀请和用户的迁移，建议您将邀请生成过程放入自定义的Rake任务中。将迁移严格限制在表操作上是一个好习惯。

/lib/tasks/distribute_invitations.rake

namespace :db do
  desc "Run all custom tasks"
  task :import_all => [:distribute_invitations, :some_other_data_import]

  desc: "Some other data import"
  task :some_other_data_import => :environment do
    puts "insert task code here"
  end

  desc: "Give existing user's invitations"
  task :distribute_invitations => :environment do
    for user in User.all
      if user.invitation_id.nil?
        invite = Invitation.create(:sender_id => <some id>, :recipient_email => <some email>, :token => <some token>, :sent_at => Time.now)
        user.update_attribute(:invitation_id, invite.id)
        puts "Updated user #{user.id} with invitation_id #{invite.id}"
      else
        puts "User already has an invitation_id"
      end
    end
  end
end

在将用户表迁移到具有邀请ID的模式后，您可以运行以下命令：

rake db:distribute_invitations

您现有的用户将通过邀请ID创建并关联到邀请。

或者您可以运行所有任务：

rake db:import_all

在这种情况下，很可能只需要将它与用户迁移一起插入:

class AddInvitationID < ActiveRecord::Migration
  def self.up
    add_column :users, :invitation_id, :integer
    for user in User.all
      if user.invitation_id.nil?
        invite = Invitation.create(:sender_id => <some id>, :recipient_email => <some email>, :token => <some token>, :sent_at => Time.now)
        user.update_attribute(:invitation_id, invite.id)
        puts "Updated user #{user.id} with invitation_id #{invite.id}"
      else
        puts "User already has an invitation_id"
      end
    end
  end

  def self.down
    remove_colum :users, :invitation_id
  end
end