修剪树的性能不佳

Question

我做了我称之为TreePruner 的东西。其目的：给定从根级别节点列表开始的层次结构，返回一个新的层次结构，其中新的根节点是满足特定条件的最高级别节点。这是我的课。

public class BreadthFirstPruner<TResource>
{
    private IEnumerable<TResource> originalList;
    private IEnumerable<TResource> prunedList;
    private Func<TResource, ICollection<TResource>> getChildren;

    public BreadthFirstPruner(IEnumerable<TResource> list, Func<TResource, ICollection<TResource>> getChildren)
    {
        this.originalList = list;
        this.getChildren = getChildren;
    }

    public IEnumerable<TResource> GetPrunedTree(Func<TResource,bool> condition)
    {
        this.prunedList = new List<TResource>();
        this.Prune(this.originalList, condition);
        return this.prunedList;
    }

    private void Prune(IEnumerable<TResource> list, Func<TResource,bool> condition)
    {
        if (list.Count() == 0)
        {
            return;
        }

        var included = list.Where(condition);
        this.prunedList = this.prunedList.Union(included);
        var excluded = list.Except(included);
        this.Prune(excluded.SelectMany(this.getChildren), condition);
    }
}

这个类做它应该做的，但是它做的太慢了，我不知道为什么。我已经在非常小的层次结构上使用了它，其中完整的层次结构已经在内存中（所以不应该有 linq-to-sql 的惊喜）。但是，无论我多么渴望或多么懒惰，实际计算 linq 表达式结果的第一行代码最终都需要 3-4 秒才能执行。

这是当前正在使用 pruner 的代码：

Func<BusinessUnitLabel, ICollection<BusinessUnitLabel>> getChildren = l => l.Children;
var hierarchy = scope.ToList();
var pruner = new BreadthFirstPruner<BusinessUnitLabel>(hierarchy, getChildren);
Func<BusinessUnitLabel, bool> hasBusinessUnitsForUser = l =>
    l.BusinessUnits.SelectMany(bu => bu.Users.Select(u => u.IDGUID)).Contains(userId);
var labels = pruner.GetPrunedTree(hasBusinessUnitsForUser).ToList();

正如我之前所说，我在执行此操作时使用的数据集非常小。它只有几个级别，大多数级别只有一个节点。正如目前所写的那样，当我调用list.Count() 时，第一次递归 调用Prune 时会出现缓慢，因为那时正在评估层次结构的第二级(excluded.SelectMany(this.getChildren))。

但是，如果我像这样添加.ToList 调用：

var included = list.Where(condition).ToList()

那么缓慢就会在那个时候发生。

我需要怎么做才能让这件事快速进行？

更新

在有人提示我更仔细地重新评估我的状况后，我意识到hasBusinessUnitsForUser 中的那些关联并没有被急切加载。那是有问题的。

Answer 1

这些调用都是延迟执行的，结果没有被缓存/物化：

    var included = list.Where(condition);
    this.prunedList = this.prunedList.Union(included);
    var excluded = list.Except(included);

即使在这个 sn-p included 中运行两次。由于这是一种递归算法，因此可能会有更多的调用。

将ToList 调用添加到任何可能多次执行的序列。