带有默认值的 C# LINQ SelectMany

Question

我正在寻找一种优雅的解决方案，以将集合中的子集合聚合到一个大集合中。我的问题是某些子集合何时可能为空。

如:

var aggregatedChildCollection = parentCollection.SelectMany(x=> x.ChildCollection);

如果任何子集合对象为空，这将引发异常。一些替代方案是:

// option 1
var aggregatedChildCollection = parentCollection
    .Where(x=>x.ChildCollection != null)
    .SelectMany(x => x.ChildCollection);

// option 2
var aggregatedChildCollection = parentCollection
    .SelectMany(x => x.ChildCollection ?? new TypeOfChildCollection[0]);

两者都可以，但我正在对父级的相当多的子集合执行特定操作，这变得有点笨拙。

我想要的是创建一个扩展方法来检查集合是否为空，如果是则执行选项 2 的操作 - 添加一个空数组。但是我对 Func 的理解还没有达到知道如何编写此扩展方法的程度。我知道我想要的语法是这样的:

var aggregatedChildCollection = parentCollection.SelectManyIgnoringNull(x => x.ChildCollection);

是否有一种简单的扩展方法可以实现这一点？

Answer 1

您可以使用自定义扩展方法:

public static IEnumerable<TResult> SelectManyIgnoringNull<TSource, TResult>(
    this IEnumerable<TSource> source, 
    Func<TSource, IEnumerable<TResult>> selector)
{
    return source.Select(selector)
        .Where(e => e != null)
        .SelectMany(e => e);
}

然后像这样使用:

var aggregatedChildCollection = parentCollection
    .SelectManyIgnoringNull(x => x.ChildCollection);