Phương pháp mở rộng LINQ, cách tìm trẻ em trong bộ sưu tập đệ quy

Question

Tôi đã quen với LINQ nhưng có ít sự hiểu biết về các phương pháp mở rộng Tôi hy vọng ai đó có thể giúp tôi.

Vì vậy, tôi có thứ bậc mã bộ sưu giả nghĩa này:

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

Và tôi có một danh sách các sản phẩm Danh sách sản phẩm.

Có cách nào để tôi có thể tìm kiếm sản phẩm trong bộ sưu tập này theo số id bằng phương pháp tiện ích không? Nói cách khác, tôi cần một mục ở đâu đó trong hệ thống phân cấp.

Answer 1

Dưới đây là một giải pháp chung chung mà sẽ ngắn mạch traversal của hệ thống phân cấp một khi trận đấu được tìm thấy.

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

Để sử dụng nó trong trường hợp của bạn:

var matchingProduct = products.FirstOrDefaultFromMany(p => p.children, p => p.Id == 27); 

Answer 2

Bạn có thể san bằng cấu trúc cây của bạn sử dụng phương pháp mở rộng này:

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

Cách sử dụng:

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

Lưu ý rằng điều này có lẽ không phải là rất nhanh. Nếu bạn liên tục cần phải tìm một sản phẩm của id, tạo ra một từ điển:

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

var product42 = dict[42]; 

Answer 3

Nếu bạn muốn "sub-lặp" và tìm thấy một đứa trẻ trong một danh sách các sản phẩm:

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

Bạn có thể sử dụng phương thức mở rộng SelectMany hiện có. SelectMany có thể được sử dụng để "làm phẳng" cấu trúc phân cấp hai cấp.

Dưới đây là một lời giải thích tuyệt vời của SelectMany: http://team.interknowlogy.com/blogs/danhanan/archive/2008/10/10/use-linq-s-selectmany-method-to-quot-flatten-quot-collections.aspx

cú pháp bạn muốn như thế này:

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

Answer 4

static IEnumerable<Product> FindProductById(this IEnumerable<Product> source, int id) 
{ 
    return source.FirstOrDefault(product => product.Id = id) ?? source.SelectMany(product => product.Children).FindProductById(id); 
} 

Answer 5

Một giải pháp thay thế bằng cách sử dụng năng suất để tối ưu hóa các kiểu liệt kê cần thiết.

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; 
      } 
     } 
    } 
} 

Sau đó, bạn có thể tìm thấy một trận đấu bằng cách gọi một cái gì đó giống như FirstOrDefault:

var match = People.SelectManyRecursive(c => c.Children) 
         .FirstOrDefault(x => x.Id == 5); 

Answer 6

Tôi chỉ refactoring giải pháp DTB của để làm cho nó chung chung hơn. Hãy thử phương pháp mở rộng này:

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

Và bạn có thể sử dụng nó như thế này:

productList.Flatten(x => x.Children).Where(x => x.ID == id);