EF Core中如何批量加载导航属性？求工具/方案推荐

场景与现有方案弊端 #

我使用EF Core 5，现有如下Code First数据库模型：

Company {
  string Name,
  int PrimaryAddressId,
  Address PrimaryAddress,
  Location[] Locations
}

Address {
  string City,
  string StreetAddress
}

Location {
  int CompanyId,
  int AddressId,
  Address Address
}

需要从数据库加载约50个Company实体及其所有关联数据，现有两种加载方式均有弊端：

• 懒加载代理：引发N+1查询问题，逐个查询关联实体，产生大量请求
• 贪婪加载（Include）：生成包含多JOIN的巨型单查询，冗余数据多、性能差，且需手动枚举所有关联路径，容易出错

理想的批量加载模式 #

期望实现类似以下手动批量加载的逻辑（性能优，利用IN查询和索引），但需避免样板代码，实现通用化：

var companies = context.Company.Take(50).ToList();
var companyIds = companies.Select(e => e.Id).ToHashSet();

// 加载PrimaryAddress
var addressIds = companies.Select(e => e.PrimaryAddressId).ToHashSet();
var addressesById = context.Address.Where(e => addressIds.Contains(e.Id)).ToDictionary(e => e.Id);
companies.ForEach(e => e.PrimaryAddress = addressesById[e.PrimaryAddressId]);

// 加载Locations及关联的Address
var locationsByCompanyId = context.Location.Where(e => companyIds.Contains(e.CompanyId)).ToLookup(e => e.CompanyId);
companies.ForEach(e => e.Locations.AddRange(locationsByCompanyId[e.Id]));

// 批量加载Location关联的Address
var locationAddressIds = locationsByCompanyId.SelectMany(l => l).Select(l => l.AddressId).ToHashSet();
var locationAddressesById = context.Address.Where(e => locationAddressIds.Contains(e.Id)).ToDictionary(e => e.Id);
foreach (var location in locationsByCompanyId.SelectMany(l => l))
{
    location.Address = locationAddressesById[location.AddressId];
}

通用化实现思路 #

1. 利用EF Core元数据自动识别关联关系 #

通过DbContext.Model访问实体元数据，自动提取关键信息：

• 实体主键：通过IEntityType.FindPrimaryKey()获取
• 导航属性对应外键：通过INavigation.ForeignKey获取关联外键字段
• 目标实体类型：通过INavigation.TargetEntityType定位关联的实体类型

2. 封装泛型批量加载方法 #

构建通用扩展方法（如LoadRelatedEntities<T>），核心步骤：

1. 执行主查询，获取实体列表
2. 遍历指定导航属性，分类型处理：
   • 引用类型导航（如Company.PrimaryAddress）：提取主实体的外键值去重，批量加载目标实体并按主键分组，再逐个赋值给主实体的导航属性
   • 集合类型导航（如Company.Locations）：提取主实体主键值，批量加载关联实体并按外键分组，将分组结果添加到主实体的集合属性中
3. 支持嵌套导航：递归调用方法，批量加载多层关联的实体

3. 标记导航属性为已加载 #

利用EF Core的ChangeTracker标记导航属性状态，避免后续触发懒加载：

context.Entry(company).Reference(c => c.PrimaryAddress).IsLoaded = true;
context.Entry(company).Collection(c => c.Locations).IsLoaded = true;

4. 性能优化细节 #

• 对提取的外键值去重，减少IN查询的参数数量
• 并行执行独立的批量查询（注意DbContext线程安全，可使用异步并行或多实例）
• 用ToLookup或ToDictionary实现O(1)的实体查找，提升赋值效率

替代方案：自定义工具类 #

无需依赖第三方库，基于EF Core元数据能力封装自定义工具类即可完成通用化，核心逻辑就是上述批量加载+元数据解析的组合。

内容的提问来源于stack exchange，提问作者Zoltán Tamási