Как я могу загрузить все дочерние записи рекурсивно? - PullRequest
Новая
       37

Как я могу загрузить все дочерние записи рекурсивно?

0 голосов
/ 13 октября 2019

У меня есть класс проекта: 

public class Project
{
    public int Id { get; set; }
    public int? ParentId { get; set; }
    public List<Project> ChildProjects { get; set; }
    // more properties
}

Это моя попытка загрузить всех потомков любого данного проекта: 

private async Task<List<Project>> LoadDescendantsOf(Project project)
{
    project.ChildProjects = await db.Projects
        .Where(p => p.ParentId == project.Id)
        .ToListAsync();
    foreach (Project childProject in project.ChildProjects)
    {
        yield return childProject.ChildProjects = 
            await LoadDescendantsOf(childProject);
    }
}

... но он не работает. Я получаю сообщение об ошибке:

Тело 'ProjectsController.LoadDescendantsOf (Project)' не может быть блоком итератора, поскольку «Task>» не является типом интерфейса итератора

Я попытался сделать метод синхронным, но это то же самое сообщение об ошибке, только без "Task" -part.

Как я могу заставить его работать?

1 Ответ

1 голос
/ 14 октября 2019

Вы можете написать простое расширение для этого: 

    public static IEnumerable<T> Traverse<T>(this T e, Func<T, IEnumerable<T>> childrenProvider)
    {
        return TraverseMany(new[] { e }, childrenProvider);
    }

    public static IEnumerable<T> TraverseMany<T>(this IEnumerable<T> collection, Func<T, IEnumerable<T>> childrenProvider)
    {
        var stack = new Stack<T>();
        foreach(var c in collection)
        {
            stack.Push(c);
        }
        while (stack.Count > 0)
        {
            var i = stack.Pop();
            yield return i;
            var children = childrenProvider(i);
            if (children != null)
            {
                foreach (var c in children)
                {
                    stack.Push(c);
                }
            }
        }
    }

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

var allProjectIds = p.Traverse(x => x.ChildProjects).Select(x => x.Id).ToList();

Если вы хотите загрузить подпроекты, я бы порекомендовал написать рекурсивную процедуру SQL на курсорах для этого,но это также подойдет для небольших данных: 

var allProjectIds = p
    .Traverse(x => 
    {
       x.ChildProjects = db.Projects
                           .Where(p => p.ParentId == project.Id)
                           .ToList();
       return x.ChildProjects;
    })
    .Select(x => x.Id)
    .ToList();
...