Rails中如何用双主键外键预加载has_many关联避免N+1查询？

为什么带lambda的关联会导致N+1查询 #

你使用的带lambda的has_many关联，在预加载时Rails无法批量替换每个Timesheet的date参数，只能逐个查询对应条件的time_entries，因此出现了N+1查询问题。

解决预加载问题的几种方法 #

方法1：手动预加载并关联（无需修改模型结构） #

先批量查询目标Timesheet，再一次性拉取所有匹配的TimeEntry，最后手动关联到对应的Timesheet对象，彻底避免N+1：

# 1. 批量获取目标Timesheet
timesheets = Timesheet.where(id: [1,2,3]).to_a

# 2. 一次性查询所有符合条件的TimeEntry，按[contract_id, date]分组
time_entries_group = TimeEntry.where(
  contract_id: timesheets.map(&:contract_id).uniq,
  date: timesheets.map(&:date).uniq
).group_by { |te| [te.contract_id, te.date] }

# 3. 手动将TimeEntry关联到对应的Timesheet
timesheets.each do |sheet|
  sheet.association(:time_entries).target = time_entries_group[[sheet.contract_id, sheet.date]] || []
end

方法2：优化关联条件，用eager_load生成JOIN查询 #

修改Timesheet的关联定义，通过JOIN条件匹配date，让eager_load能生成单条JOIN查询：

class Timesheet < ApplicationRecord
  belongs_to :contract
  has_many :time_entries, ->(sheet) { where(date: sheet.date) },
           primary_key: :contract_id,
           foreign_key: :contract_id

  # 新增用于预加载的关联（也可直接替换原关联）
  has_many :time_entries_preloadable, -> { joins('JOIN timesheets ON time_entries.contract_id = timesheets.contract_id AND time_entries.date = timesheets.date') },
           through: :contract,
           source: :time_entries
end

使用eager_load预加载：

Timesheet.where(id: [1,2,3]).eager_load(:time_entries_preloadable).to_a

这条语句会生成包含JOIN的SQL，一次性查询所有Timesheet和对应的TimeEntry。

方法3：用第三方gem支持复合主键（可选） #

如果想通过复合键约束关联，可以使用composite_primary_keys gem，它让Rails支持复合主键与外键。安装后可这样定义：

# TimeEntry模型
class TimeEntry < ApplicationRecord
  self.primary_keys = :contract_id, :date
  belongs_to :contract
end

# Timesheet模型
class Timesheet < ApplicationRecord
  belongs_to :contract
  has_many :time_entries, primary_key: [:contract_id, :date], foreign_key: [:contract_id, :date]
end

这样关联会自动匹配contract_id和date，且预加载可正常工作。但需注意第三方gem的项目兼容性。

关于双主键/外键的说明 #

Rails原生不支持复合主键和复合外键，官方推荐通过关联条件或业务逻辑实现类似约束。如果一定要用复合键，composite_primary_keys是最成熟的解决方案，但需要额外引入依赖。

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