Как я могу получить первое значение IEnumerable, если длина равна точно одному, или по умолчанию в противном случае?

Question

Мне хотелось бы что-то вроде IEnumerable.FirstOrDefault, но я хотел бы вернуть значение по умолчанию, если length != 1. Что-то вроде v.Count() == 1 ? v[0] : default. Но я бы предпочел сделать это, не превратив все перечислимое в массив. Каков наилучший способ получить это поведение?

Answer 1

Это хорошая идея, чтобы избежать повторения перечислимого более одного раза, поскольку не все перечисляемые повторяют одни и те же значения.

Чтобы гарантировать, что ваш перечислимый возвращает только 1 элемент, наиболее эффективно перенести первые два элемента из перечислимого и преобразовать результат в массив. Если массив имеет два элемента, вы знаете, что перечисляемый имеет более одного элемента без повторения всего перечислимого. Можно перечислить бесконечно долго.

Так вот как это сделать:

public static T OneOnlyThenFirstOrDefault<T>(this IEnumerable<T> source) 
{ 
    var beginning = source.Take(2).ToArray(); 
    return beginning.Length == 1 ? beginning[0] : default(T); 
} 

Answer 2

Вы пробовали написать метод пользовательского расширения, что-то вроде:

public static TSource MyFirstOrDefault<TSource>(this IEnumerable<TSource> source) 
{ 
    return source.Count() == 1 ? source.First() : default(TSource); 
} 

но вы должны иметь в виду, что методы Count() и First() необходимо для конкретизации внутреннюю коллекцию элементов для того, чтобы получить count (или первый элемент).

Надеюсь, это вам поможет.

Answer 3

реализация Microsoft по Enumerable.FirstOrDefault работает непосредственно с основным переписчиком. Также могут быть ваши:

public static T FirstIfOnlyOne<T>(this IEnumerable<T> source, T defaultValue = default(T)) 
{ 
    // (Null check and IList<T> optimization omitted...) 

    using (IEnumerator<T> enumerator = source.GetEnumerator()) 
    { 
     if (enumerator.MoveNext()) // Is there at least one item? 
     { 
      T item = enumerator.Current; // Save it. 
      if (!enumerator.MoveNext()) // Is that it? 
       return item; 
     } 
    } 
    return defaultValue; 
} 

Возможно, это наиболее эффективная реализация.