Rails4.2迁移中如何将Country对象存入内存避免重复DB查询？

嘿，这个N+1查询的问题在数据量较大时确实会拖慢迁移进度，我来给你几个针对性的优化点，把所有数据库查询都降到最少：

核心问题分析 #

你遇到的重复查询，主要来自三个地方：

1. 循环中每次调用user.country都会触发单条查询
2. 访问country.default_area和country.domain时，没有预加载关联对象，导致额外查询
3. 每次循环都调用Area.by_country(country).pluck(:id)，重复查询区域数据

优化后的完整脚本 #

# 预加载默认国家的关联对象（default_area、domain），确保仅查询一次
default_country = Country.includes(:default_area, :domain).find_or_create_by(code: 'it', name: 'Italy')

# 提前缓存所有国家对应的区域ID列表，避免循环中重复查询数据库
country_areas_map = Country.all.each_with_object({}) do |country, map|
  map[country.id] = Area.by_country(country).pluck(:id)
end

# 预加载用户关联的Country，彻底消除N+1查询
User.includes(:country).find_each do |user|
  country = user.country || default_country
  params_to_update = {}

  # 仅当用户无country_id时才赋值
  params_to_update[:country_id] = country.id unless user.country_id

  # 从内存缓存中取区域ID列表，无需访问数据库
  user_country_areas_ids = country_areas_map[country.id]
  # 检查用户当前区域是否属于所属国家，否则更新为默认区域
  unless user_country_areas_ids.include?(user.area_id)
    params_to_update[:area_id] = country.default_area.id
  end

  # 直接从内存中的domain对象获取默认locale
  params_to_update[:locale] = country.domain.default_locale

  # 仅当有字段需要更新时执行操作，避免空更新
  user.update_columns(params_to_update) unless params_to_update.empty?
end

关键优化点说明 #

• 预加载用户的Country关联：User.includes(:country).find_each会一次性加载所有用户的Country数据到内存，替代原来每次循环单独查询user.country的行为。
• 预加载默认国家的关联对象：通过includes(:default_area, :domain)，把默认国家需要用到的关联对象一次性加载完成，后续访问country.default_area和country.domain时直接读取内存数据。
• 提前缓存区域ID映射：一次性查询所有国家对应的区域ID列表并存储到哈希中，循环时直接从哈希取值，避免重复执行Area.by_country查询。
• 避免空更新：新增unless params_to_update.empty?判断，没有字段需要更新时跳过update_columns操作，减少不必要的数据库交互。

这样修改后，数据库查询次数会从“用户数×N”级降到个位数，在大型数据库中能节省大量时间。

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