使用 Linq 在树状结构中遍历

Question

我有一个具有这样结构的对象：

    public class CustomObject
    {
        public int CustomProperty{get;set;}
        public string CustomStringProperty{get;set;}
        public List<CustomObject> CustomObjects{get;set;}
    }

我知道有很多方法可以拉出任何对象、其父对象或根据 CustomProperty 上的某些条件获取对象。

我想编写一些扩展方法或 linq，以便客户端可以访问每个对象及其属性。

当前客户有列表 --> List<CustomObject>();

例如。 ClientMethod(List<CustomObject> customObjs);

Answer 1

假设您希望能够遍历列表中的项目及其所有子项。

像往常一样有这样的问题，我建议你从How to flatten tree via LINQ?获取高效的通用树前序遍历方法：

public static partial class TreeHelper
{
    public static IEnumerable<T> Expand<T>(this IEnumerable<T> source, Func<T, IEnumerable<T>> elementSelector)
    {
        var stack = new Stack<IEnumerator<T>>();
        var e = source.GetEnumerator();
        try
        {
            while (true)
            {
                while (e.MoveNext())
                {
                    var item = e.Current;
                    yield return item;
                    var elements = elementSelector(item);
                    if (elements == null) continue;
                    stack.Push(e);
                    e = elements.GetEnumerator();
                }
                if (stack.Count == 0) break;
                e.Dispose();
                e = stack.Pop();
            }
        }
        finally
        {
            e.Dispose();
            while (stack.Count != 0) stack.Pop().Dispose();
        }
    }
}

并根据您的需要使用它（如果您不喜欢，可以使用其他名称）：

partial class TreeHelper
{
    public static IEnumerable<CustomObject> Expand(this IEnumerable<CustomObject> source)
    {
        return source.Expand(item => item.CustomObjects != null && item.CustomObjects.Count > 0 ? item.CustomObjects : null);
    }
}

那么你可以简单地使用：

void ClientMethod(List<CustomObject> customObjs)
{
    foreach (var item in customObjs.Expand())
    {
        // do something
    }
}

或 LINQ 查询，如 customObjs.Expand().Where(...) 等。