Метод расширения Linq, как найти дочерний элемент в коллекции - PullRequest
Новая
       20

Метод расширения Linq, как найти дочерний элемент в коллекции

11 голосов
/ 27 августа 2010

Я уже знаком с Linq, но плохо разбираюсь в методах расширения, надеюсь, кто-нибудь мне поможет.

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

class Product
  prop name
  prop type
  prop id
  prop List<Product> children

И у меня есть список продуктов Список продуктов.

Можно ли как-нибудь искать товар в этой коллекции по id с методом расширения?Другими словами, мне нужен один элемент где-то в иерархии.

Ответы [ 6 ]

17 голосов
/ 27 августа 2010

Вот обобщенное решение, которое обойдет короткое замыкание иерархии, когда совпадение найдено. 

public static class MyExtensions
{
    public static T FirstOrDefaultFromMany<T>(
        this IEnumerable<T> source, Func<T, IEnumerable<T>> childrenSelector,
        Predicate<T> condition)
    {
        // return default if no items
        if(source == null || !source.Any()) return default(T);

        // return result if found and stop traversing hierarchy
        var attempt = source.FirstOrDefault(t => condition(t));
        if(!Equals(attempt,default(T))) return attempt;

        // recursively call this function on lower levels of the
        // hierarchy until a match is found or the hierarchy is exhausted
        return source.SelectMany(childrenSelector)
            .FirstOrDefaultFromMany(childrenSelector, condition);
    }
}

Чтобы использовать его в вашем случае: 

var matchingProduct = products.FirstOrDefaultFromMany(p => p.children, p => p.Id == 27);
8 голосов
/ 27 августа 2010

Вы можете сгладить свою древовидную структуру, используя этот метод расширения: 

static IEnumerable<Product> Flatten(this IEnumerable<Product> source)
{
    return source.Concat(source.SelectMany(p => p.Children.Flatten()));
}

Использование: 

var product42 = products.Flatten().Single(p => p.Id == 42);

Обратите внимание, что это, вероятно, не очень быстро. Если вам неоднократно нужно найти товар по идентификатору, создайте словарь: 

var dict = products.Flatten().ToDictionary(p => p.Id);

var product42 = dict[42];
1 голос
/ 11 января 2014

Я просто рефакторинг решения dtb, чтобы сделать его более общим.Попробуйте этот метод расширения: 

public static IEnumerable<T> Flatten<T, R>(this IEnumerable<T> source, Func<T, R> recursion) where R : IEnumerable<T>
{
    return source.SelectMany(x => (recursion(x) != null && recursion(x).Any()) ? recursion(x).Flatten(recursion) : null)
                 .Where(x => x != null);
}

И вы можете использовать его так: 

productList.Flatten(x => x.Children).Where(x => x.ID == id);
0 голосов
/ 10 июля 2013

Альтернативное решение, использующее выход для оптимизации необходимых перечислений. 

public static IEnumerable<T> SelectManyRecursive<T>(
    this IEnumerable<T> source,
    Func<T, IEnumerable<T>> childrenSelector)
{
    if (source == null)
        throw new ArgumentNullException("source");

    foreach (var i in source)
    {
        yield return i;
        var children = childrenSelector(i);
        if (children != null)
        {
            foreach (var child in SelectManyRecursive(children, childrenSelector))
            {
                yield return child;
            }
        }
    }
}

Тогда вы можете найти соответствие, вызвав что-то вроде FirstOrDefault: 

    var match = People.SelectManyRecursive(c => c.Children)
                      .FirstOrDefault(x => x.Id == 5);
0 голосов
/ 27 августа 2010 
static IEnumerable<Product> FindProductById(this IEnumerable<Product> source, int id) 
{
    return source.FirstOrDefault(product => product.Id = id) ?? source.SelectMany(product => product.Children).FindProductById(id);
}
0 голосов
/ 27 августа 2010

Если вы хотите «повторить» и найти дочернего элемента в списке продуктов: 

List<Product>
    Product
       Child
       Child
       Child
       Child
    Product
       Child
       Child *find this one
       Child

Вы можете использовать существующий метод расширения SelectMany.SelectMany может использоваться для «выравнивания» двухуровневой иерархии.

Вот отличное объяснение SelectMany: http://team.interknowlogy.com/blogs/danhanan/archive/2008/10/10/use-linq-s-selectmany-method-to-quot-flatten-quot-collections.aspx

Ваш синтаксис должен выглядеть следующим образом: 

List<Product> p = GetProducts(); //Get a list of products
var child = from c in p.SelectMany(p => p.Children).Where(c => c.Id == yourID);
...