Linq递归亲子

时间:2017-01-12 08:49:52

标签: c# linq recursion

我有来自数据库的数据:

id | description              | parent_id
-----------------------------------------
1  | Record 1                 | null
2  | Record 2                 | 1
3  | Record 3                 | null
4  | Record 4                 | 2
5  | Record 5                 | 3

我想使用Linq处理,它应该是这样的结果:

Record 1
- Record 2
  - Record 4
Record 3
- Record 5

我得到了解决方案:Recursive LINQ query: select item and all children with subchildren

但不幸的是结果只有两个级别,因为我希望结果处于多级> 2级(像树)。

还有其他最佳解决方案吗?或如何修改该解决方案

提前致谢

1 个答案:

答案 0 :(得分:8)

对于linq-to-objects,您可以在IEnumerable<T>上定义自己的扩展方法,递归获取所有子项。

public static class EnumerableExtensions
{
    public static IEnumerable<T> SelectRecursive<T>(this IEnumerable<T> source, Func<T, IEnumerable<T>> selector)
    {
        foreach (var parent in source)
        {
            yield return parent;

            var children = selector(parent);
            foreach (var child in SelectRecursive(children, selector))
                yield return child;
        }
    }
}

用法:

var lookup = col.ToLookup(x => x.Parent_Id);
var res = lookup[null].SelectRecursive(x => lookup[x.Id]).ToList();