Linq递归父子

Question

我有这样的数据库数据：

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 级的结果（如树）。

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

提前致谢

Answer 1

对于 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();