EF Core仓储类中区分引用与集合实现显式加载关联数据的方法

我有一个包含Include功能的Repository类，目前的代码仅适用于一对一关系。但针对集合类型（如ICollection<T>或IList<T>，示例对象类型为System.Collections.Generic.IList1[[WestCore.Domain.Entities.WestLife.MhpProduct]]），需要将DbContext.Entry(model.Result).Reference(include).Load();改为DbContext.Entry(model.Result).Collection(include).Load();`。

我的现有代码如下：

public virtual async Task<TEntity> GetByIdAsync(object[] keyValues, List<Expression<Func<TEntity, object>>> includes, CancellationToken cancellationToken = default(CancellationToken))
{
    var model = DbSet.FindAsync(keyValues, cancellationToken);
    if (includes == null) return await model;
    foreach (var include in includes)
        //if (include.Body.Type.IsAssignableFrom(typeof(ICollection<>)))
        DbContext.Entry(model.Result).Reference(include).Load();
    return await model;
    //return await DbSet.FindAsync(keyValues, cancellationToken);
}

请问我可以使用什么条件来区分引用（Reference）和集合（Collection）类型，以分别调用对应的加载方法？

要区分EF Core中的引用导航属性和集合导航属性，你可以通过解析表达式返回类型结合EF Core元数据，或者直接识别集合类型特征来实现。下面提供两种可靠的方案：

方式一：利用EF Core元数据API（推荐） #

EF Core的DbContext内置了实体元数据，能精准判断导航属性的类型，避免手动类型判断的误差，同时支持异步加载优化：

public virtual async Task<TEntity> GetByIdAsync(object[] keyValues, List<Expression<Func<TEntity, object>>> includes, CancellationToken cancellationToken = default(CancellationToken))
{
    var entityTask = DbSet.FindAsync(keyValues, cancellationToken);
    var entity = await entityTask;
    
    if (includes == null || entity == null) return entity;

    // 获取当前实体的元数据
    var entityType = DbContext.Model.FindEntityType(typeof(TEntity));
    foreach (var include in includes)
    {
        // 解析导航属性的成员表达式
        var memberExpr = include.Body as MemberExpression;
        if (memberExpr == null) continue;

        // 通过元数据查找导航属性
        var navProperty = entityType.FindNavigation(memberExpr.Member.Name);
        if (navProperty == null) continue;

        // 根据导航属性类型调用对应加载方法
        if (navProperty.IsCollection())
        {
            await DbContext.Entry(entity).Collection(include).LoadAsync(cancellationToken);
        }
        else
        {
            await DbContext.Entry(entity).Reference(include).LoadAsync(cancellationToken);
        }
    }
    return entity;
}

优势说明 #

• 元数据判断完全贴合EF Core的导航属性定义，不会因为自定义集合类型、继承关系出现误判。
• 替换为异步的LoadAsync方法，和你的异步方法逻辑保持一致，避免同步阻塞问题。

方式二：手动判断集合类型特征 #

如果不想依赖EF Core元数据，也可以通过检查表达式返回类型的集合特征来区分：

public virtual async Task<TEntity> GetByIdAsync(object[] keyValues, List<Expression<Func<TEntity, object>>> includes, CancellationToken cancellationToken = default(CancellationToken))
{
    var entityTask = DbSet.FindAsync(keyValues, cancellationToken);
    var entity = await entityTask;
    
    if (includes == null || entity == null) return entity;

    foreach (var include in includes)
    {
        var returnType = GetRealReturnType(include);
        bool isCollection = IsCollectionType(returnType);

        if (isCollection)
        {
            await DbContext.Entry(entity).Collection(include).LoadAsync(cancellationToken);
        }
        else
        {
            await DbContext.Entry(entity).Reference(include).LoadAsync(cancellationToken);
        }
    }
    return entity;
}

// 辅助方法：解析表达式的真实返回类型（处理装箱导致的类型转换）
private Type GetRealReturnType<T>(Expression<Func<T, object>> expr)
{
    if (expr.Body is UnaryExpression unaryExpr && unaryExpr.NodeType == ExpressionType.Convert)
    {
        return unaryExpr.Operand.Type;
    }
    return expr.Body.Type;
}

// 辅助方法：判断是否为集合类型（排除string这类特殊的可枚举类型）
private bool IsCollectionType(Type type)
{
    if (type == typeof(string))
        return false;

    return type.IsGenericType && 
           (typeof(ICollection<>).IsAssignableFrom(type.GetGenericTypeDefinition()) || 
            typeof(IList<>).IsAssignableFrom(type.GetGenericTypeDefinition())) ||
           typeof(ICollection).IsAssignableFrom(type);
}

注意点 #

• 必须排除string类型：它虽然实现了IEnumerable<char>，但不属于集合导航属性范畴。
• 处理装箱情况：当表达式返回值为值类型/可空值类型时，EF会自动装箱为object，需要解析出真实的返回类型。

另外，原代码里直接访问model.Result会导致同步阻塞，上面的两种方案都先await获取实体再处理，修复了这个潜在问题。

内容的提问来源于stack exchange，提问作者Derviş Kayımbaşıoğlu