Rails技术问题：如何为已有School批量生成对应Setting实例

嘿，这个需求在Rails项目里太常见了，我给你几个实用的方法，你可以根据自己的场景来选：

适合临时一次性处理，打开终端跑rails c，然后执行这段代码：

School.find_each do |school|
  # 先判断有没有对应的Setting，避免重复创建报错
  school.create_setting! unless school.setting.present?
end

这里用find_each而不是all.each，是因为它会分批加载数据，不会一次性把所有School都塞进内存，处理大量数据的时候更稳妥。create_setting!是has_one关联自动生成的方法，要是创建过程中有验证失败之类的问题会直接抛出异常，方便你排查。

如果你想把这个操作固化下来，避免以后换环境还要手动执行，可以写个迁移：

1. 先生成迁移文件：

rails generate migration AddSettingsForExistingSchools

2. 打开生成的迁移文件，把内容改成这样（注意把Rails版本号换成你项目用的）：

class AddSettingsForExistingSchools < ActiveRecord::Migration[7.0]
  def up
    School.find_each do |school|
      school.create_setting! unless school.setting.present?
    end
  end

  def down
    # 回滚的时候可以删除所有自动创建的Setting，要是有手动创建的需要区分的话，这里可以加逻辑
    Setting.destroy_all
  end
end

3. 最后执行rails db:migrate就搞定了。

如果以后可能还要重复执行（比如后续新增的School忘了创建Setting），写个Rake任务更方便：

1. 在lib/tasks目录下新建settings.rake文件，内容如下：

namespace :settings do
  desc "自动给所有没有Setting的School创建对应实例"
  task create_for_schools: :environment do
    created_count = 0
    School.find_each do |school|
      unless school.setting.present?
        school.create_setting!
        created_count += 1
      end
    end
    puts "搞定啦！一共创建了 #{created_count} 个Setting实例～"
  end
end

2. 执行的时候跑rails settings:create_for_schools就行，还能看到创建的数量。

如果你的Setting模型有必填字段（比如default_config之类的），记得在创建的时候传入默认值，比如：

school.create_setting!(default_config: "{}", enabled: true)

不然会因为验证不通过创建失败哦。

要是数据量特别大（比如上万条School），可以用批量插入提升速度：

# 先找出所有还没有Setting的School ID
school_ids = School.where.not(id: Setting.pluck(:school_id)).pluck(:id)
# 批量插入
Setting.insert_all(school_ids.map do |id|
  { school_id: id, created_at: Time.current, updated_at: Time.current }
end)

这个方法是单条SQL插入，比遍历创建快很多，但要记得手动处理created_at和updated_at，还有其他必填字段的默认值。

内容的提问来源于stack exchange，提问作者Tetramputechture