Rails：迁移生产数据库，需要为每个现有用户创建新数据

Question

我的 Ruby on Rails 应用程序中有一个生产数据库，其中包含用户和其他内容。现在，在实现新功能时，我需要为每个现有用户创建一些新数据。有什么聪明的方法可以做到吗？也许是 Rake 脚本？

更多详细信息：

我有一个用户表。现在我添加了书签表。这个想法是，默认情况下每个用户都有 5 个带有预填充数据的书签。对于新用户来说很容易做到，但现有用户呢？在用户登录时创建数据是一种选择，但感觉有点脏。

谢谢！

Answer 1

我喜欢使用 rails runner 来完成这样的一次性脚本，它允许您轻松运行加载了 Rails env 的脚本。 。

如果您有很多用户，我建议使用 activerecord-import 来加快速度。

我们可以在脚本目录中创建一个名为 make_bookmarks.rb 的脚本，如下所示：

require 'activerecord-import'

bookmarks = []
User.all.each do |user|
    ["value1","value2","value3","value4","value5"].each do |value|
        bookmarks << user.bookmarks.new(:col => value)
    end
end

# import the bookmarks all with bulk sql = much faster
Bookmark.import bookmarks

然后运行它 rails runner script/make_bookmarks.rb

Answer 2

一种是在迁移中包含数据转换（假设您的 has_many 关系使用连接模型 user_bookmark.rb）：

class AddBookmarksTable < ActiveRecord::Migration
  def self.up
    create_table :bookmarks, :force => true do |t|
      t.string  :some_col
      t.timestamps
    end
    create_table :user_bookmarks, :force => true do |t|
      t.integer  :bookmark_id, user_id
      t.timestamps
    end
    # create the 5 bookmarks you want to seed as values for existing users
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    # add values to the join table for your model
    User.all.each{|u| Bookmark.all.each{|b| UserBookmark.create(:user_id => u.id, :bookmark_id => b.id)}}
  end

  def self.down
    drop_table :bookmarks
  end
end

在您的 user.rb 中，您应该有

has_many :bookmarks

和 bookmark.rb

belongs_to :user

另一种方法（首选）是仅使用您的迁移来创建表，因为种子数据并不真正属于迁移，并且有一个自定义的 rake 任务可以在 lib/tasks 中为您完成工作。

lib/tasks/add_bookmarks_to_existing_users.rake

namespace :db do
  desc "Add bookmarks to existing users"
  task :add_bookmarks_to_existing_users => :environment do
    # create seed book marks (under impression none exist right now and associations have been set up in user and bookmark models)
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    Bookmark.create(:some_col => 'value')
    User.all.each{|u| u.bookmarks << Bookmark.all}
  end
end

之后你可以运行：

rake db:add_bookmarks_to_existing_users