EF Core基于布尔属性条件Include关联对象的最佳实践（避if-else）

问题背景 #

定义了如下实体类：

public class VendorHotelRoom : BaseEntity
{
    public virtual ICollection<VendorHotelRoomName> VendorHotelRoomNames { get; set; }
    public bool CompleteVendorHotelRoomNames { get; set; }

    public Guid VendorHotelRoomDetailId { get; set; }
    public virtual VendorHotelRoomDetail VendorHotelRoomDetail { get; set; }
    public bool CompleteVendorHotelRoomDetail { get; set; }

    public virtual ICollection<VendorHotelRoomAvailability> VendorHotelRoomAvailabilities { get; set; }
    public bool CompleteVendorHotelRoomAvailabilities { get; set; }

    public virtual ICollection<VendorHotelRoomFacility> VendorHotelRoomFacilities { get; set; }
    public bool CompleteVendorHotelRoomFacilities { get; set; }

    public virtual ICollection<VendorHotelRoomDescription> VendorHotelRoomDescriptions { get; set; }
    public bool CompleteVendorHotelRoomDescriptions { get; set; }
}

需要实现的查询逻辑是：根据实体的布尔属性值，条件性地Include对应的关联对象，避免写大量if-else语句。目标逻辑如下：

var vhr = hotelContext.VendorHotelRooms.Where(c => c.Id == VendorHotelRoomId)
       //if CompleteVendorHotelRoomAvailabilities true .Include(c => c.VendorHotelRoomAvailabilities )
       //if condition(boolean) true     .Include(c => c.VendorHotelRoomDescriptions)
       //if condition(boolean) true     .Include(c => c.VendorHotelRoomDetail)
       //if condition(boolean) true     .Include(c => c.VendorHotelRoomFacilities)
       //if condition(boolean) true     .Include(c => c.VendorHotelRoomNames).FirstOrDefault();

解决方案 #

推荐两种实用方案来简化代码，避免冗余的if-else：

方案1：条件链式调用扩展方法 #

利用EF Core的IQueryable链式特性，自定义IncludeIf扩展方法实现条件Include，代码简洁直观：

首先定义扩展方法：

public static class QueryableExtensions
{
    public static IQueryable<T> IncludeIf<T, TProperty>(this IQueryable<T> query, 
        bool condition, 
        Expression<Func<T, TProperty>> navigationPropertyPath) where T : class
    {
        return condition ? query.Include(navigationPropertyPath) : query;
    }
}

然后在查询中使用：

// 先获取目标实体的布尔属性值（如果是外部传入的条件可直接用参数）
var targetRoom = hotelContext.VendorHotelRooms.AsNoTracking().FirstOrDefault(c => c.Id == VendorHotelRoomId);
if (targetRoom == null) return null;

var query = hotelContext.VendorHotelRooms.Where(c => c.Id == VendorHotelRoomId);

// 按条件链式追加Include
query = query.IncludeIf(targetRoom.CompleteVendorHotelRoomAvailabilities, c => c.VendorHotelRoomAvailabilities)
             .IncludeIf(targetRoom.CompleteVendorHotelRoomDescriptions, c => c.VendorHotelRoomDescriptions)
             .IncludeIf(targetRoom.CompleteVendorHotelRoomDetail, c => c.VendorHotelRoomDetail)
             .IncludeIf(targetRoom.CompleteVendorHotelRoomFacilities, c => c.VendorHotelRoomFacilities)
             .IncludeIf(targetRoom.CompleteVendorHotelRoomNames, c => c.VendorHotelRoomNames);

var vhr = query.FirstOrDefault();

方案2：元数据映射动态Include #

如果后续可能新增大量关联属性和对应布尔标记，可通过映射字典统一管理，遍历字典动态添加Include：

// 构建映射关系：布尔属性名称 -> 对应导航属性表达式
var includeMappings = new Dictionary<string, Expression<Func<VendorHotelRoom, object>>>
{
    { nameof(VendorHotelRoom.CompleteVendorHotelRoomAvailabilities), c => c.VendorHotelRoomAvailabilities },
    { nameof(VendorHotelRoom.CompleteVendorHotelRoomDescriptions), c => c.VendorHotelRoomDescriptions },
    { nameof(VendorHotelRoom.CompleteVendorHotelRoomDetail), c => c.VendorHotelRoomDetail },
    { nameof(VendorHotelRoom.CompleteVendorHotelRoomFacilities), c => c.VendorHotelRoomFacilities },
    { nameof(VendorHotelRoom.CompleteVendorHotelRoomNames), c => c.VendorHotelRoomNames }
};

var targetRoom = hotelContext.VendorHotelRooms.AsNoTracking().FirstOrDefault(c => c.Id == VendorHotelRoomId);
if (targetRoom == null) return null;

var query = hotelContext.VendorHotelRooms.Where(c => c.Id == VendorHotelRoomId);

// 遍历映射字典，根据布尔属性值动态Include
foreach (var mapping in includeMappings)
{
    var boolValue = (bool)typeof(VendorHotelRoom).GetProperty(mapping.Key).GetValue(targetRoom);
    if (boolValue)
    {
        query = query.Include(mapping.Value);
    }
}

var vhr = query.FirstOrDefault();

注意事项 #

• 方案1中的IncludeIf如果是基于外部传入的布尔参数（而非实体属性），可直接传入参数判断，无需提前查询实体。
• 方案2通过反射获取属性值，适合需要统一管理关联映射的场景，后续新增属性只需在字典中添加条目即可。

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