Rails预加载关联子集后，如何避免N+1并仅返回该子集，保留无关联父模型

在Rails的Active Record中，当父模型（如Resource）和子模型（如Occupancy）是一对多关联时，若预加载了子模型的指定子集（比如报表周期内的Occupancy数据），调用parent.children（如resource.occupancies）仍然会触发N+1查询并返回所有关联的子模型，而非预加载的子集。同时业务需求要求保留那些在报表周期内没有对应Occupancy数据的Resource，用父模型的预设值填充展示。

方法1：自定义实例方法+定向预加载 #

默认的关联方法会查询全量子模型，所以可以在父模型里写一个实例方法，优先读取预加载的子集，避免重复查询：

# app/models/resource.rb
class Resource < ApplicationRecord
  has_many :occupancies

  # 获取指定日期范围内的预加载占用数据，无预加载时才查询数据库
  def report_occupancies(start_date, end_date)
    if loaded_associations.key?(:occupancies)
      occupancies.select { |occ| occ.date.between?(start_date, end_date) }
    else
      occupancies.where(date: start_date..end_date)
    end
  end
end

查询时预加载指定范围的子模型，同时保留所有Resource：

start_date = Date.parse('2024-01-01')
end_date = Date.parse('2024-01-31')

resources = Resource.all.preload(:occupancies) do |scope|
  scope.where(date: start_date..end_date)
end

展示报表时调用自定义方法：

resources.each do |resource|
  occs = resource.report_occupancies(start_date, end_date)
  if occs.present?
    occs.each do |occ|
      puts "#{resource.name} - #{occ.date}: #{occ.value}"
    end
  else
    puts "#{resource.name} - 日期范围内无数据，使用预设值: #{resource.default_occupancy}"
  end
end

方法2：定义带条件的专属关联 #

直接在父模型里定义一个用于报表的关联，预加载时指定条件，避免和默认的全量关联混淆：

# app/models/resource.rb
class Resource < ApplicationRecord
  has_many :occupancies
  # 空关联，用于预加载指定条件的子集
  has_many :report_occupancies, class_name: 'Occupancy'
end

查询时针对这个专属关联做定向预加载：

start_date = Date.parse('2024-01-01')
end_date = Date.parse('2024-01-31')

resources = Resource.all.preload(:report_occupancies) do |scope|
  scope.where(date: start_date..end_date)
end

展示时直接调用report_occupancies，不会触发N+1，且无数据的Resource返回空数组：

resources.each do |resource|
  if resource.report_occupancies.present?
    resource.report_occupancies.each do |occ|
      puts "#{resource.name} - #{occ.date}: #{occ.value}"
    end
  else
    puts "#{resource.name} - 日期范围内无数据，使用预设值: #{resource.default_occupancy}"
  end
end

方法3：数据库层面聚合处理（适合统计场景） #

如果需要按日期生成完整报表，可以用left_joins配合SQL函数直接在数据库层面完成数据合并，避免Ruby层面的循环处理：

start_date = Date.parse('2024-01-01')
end_date = Date.parse('2024-01-31')

# 一次性查询所有资源在指定日期范围的占用数据，无数据时用预设值填充
resource_stats = Resource.left_joins(:occupancies)
                         .where(occupancies: { date: start_date..end_date })
                         .group('resources.id', 'occupancies.date')
                         .select(
                           'resources.id',
                           'resources.name',
                           'resources.default_occupancy',
                           'occupancies.date',
                           'COALESCE(occupancies.value, resources.default_occupancy) AS occupancy_value'
                         )

后续可以将结果按资源分组，补全缺失日期的数据：

stats_by_resource = resource_stats.group_by { |stat| stat.id }
all_dates = (start_date..end_date).to_a

resources.each do |resource|
  resource_stats = stats_by_resource[resource.id] || []
  all_dates.each do |date|
    stat = resource_stats.find { |s| s.date == date }
    value = stat ? stat.occupancy_value : resource.default_occupancy
    puts "#{resource.name} - #{date}: #{value}"
  end
end

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