C# LINQ SelectMany по умолчанию

Question

Я ищу элегантное решение для объединения детской коллекции в коллекцию в одну большую коллекцию. Моя проблема в том, что некоторые дочерние коллекции могут быть пустыми.

EG:

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]); 

И будет работать, но я делаю определенную операцию на очень немногих детских коллекций на родителей, и она становится немного unweilding.

Я хотел бы создать метод расширения, который проверяет, является ли коллекция нулевым, и если это делает то, что делает опция 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); 

Answer 2

Если ParentCollection ваш собственный класс, который вы также должны быть в состоянии конструктор по умолчанию для класса такого as:

public ParentCollection{ 
    public ParentCollection() { 
     ChildCollection = new List<ChildCollection>(); 
    } 
} 

Это должно предотвратить исключение null ref и дать вам пустой список, если в нем ничего нет. По крайней мере, это работает с EF-моделями.

Answer 3

Ваша «опция 2» - это то, что я хотел бы сделать, с незначительной настройкой: используйте Enumerable.Empty() вместо создания пустого массива, чтобы уменьшить количество новых объектов, которые вы создаете.

Я использую тривиальный метод расширения Touch() - имя утилиты * NIX - сохранить поток синтаксиса LINQ и уменьшить ввод:

public static IEnumerable<T> Touch<T>(this IEnumerable<T> items) => 
    items ?? Enumerable.Empty<T>(); 

И будет использовать его как:

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

Answer 4

Вы можете избежать накладных расходов с расширением LINQ, используя SelectManyReference Source

public static IEnumerable<TResult> SelectManyNotNull<TSource, TResult>(
     this IEnumerable<TSource> source, Func<TSource, IEnumerable<TResult>> selector) 
{ 
    foreach (TSource element in source) 
     if (selector(element) != null) 
      foreach (TResult subElement in selector(element)) 
       yield return subElement; 
}