调用Parallel.For()时如何避免闭包并消除匿名分配？

核心问题拆解 #

原代码中Parallel.For的匿名委托捕获了point、scale和ConcurrentBag<MyElement>，不仅产生闭包导致额外堆分配，ConcurrentBag的线程安全开销还抵消了并行带来的性能收益。我们需要在不依赖闭包的前提下，将必要参数传递给迭代逻辑，同时优化线程竞争问题。

方案1：使用带本地状态的Parallel.For重载 #

利用Parallel.For支持本地状态的重载，让每个线程维护独立的局部集合，最后合并结果，彻底避免全局ConcurrentBag的竞争。同时由于testPoint和scale是不可变/值类型，编译器不会生成闭包类，避免额外分配：

internal MyElement[] HitTest((double x, double y) point, double scale)
{
    var testPoint = new Point { X = point.x, Y = point.y };
    var finalResult = new List<MyElement>();

    Parallel.For(0, UIElements.Length,
        // 每个线程初始化本地结果集合
        () => new List<MyElement>(),
        // 迭代逻辑：参数明确传递，无闭包
        (index, loopState, localList) =>
        {
            var element = UIElements[index];
            if (element is IHitTestable ht && ht.HitTest(ref testPoint, scale))
            {
                localList.Add(element);
            }
            return localList;
        },
        // 合并线程本地结果到全局集合
        localList =>
        {
            lock (finalResult)
            {
                finalResult.AddRange(localList);
            }
        });

    return finalResult.ToArray();
}

方案2：自定义参数结构体+静态方法委托 #

如果要完全消除任何隐式闭包或上下文分配，可以自定义包含所有必要参数的结构体，再用静态方法作为Parallel.For的委托，确保编译时参数类型完全确定：

// 封装所有测试所需参数的结构体
private struct HitTestContext
{
    public Point TestPoint;
    public double Scale;
    public MyElement[] Elements;
    public List<MyElement> LocalResult;
}

// 静态迭代方法，无闭包、无匿名分配
private static void ExecuteHitTest(int index, ParallelLoopState loopState, object contextObj)
{
    var context = (HitTestContext)contextObj;
    var element = context.Elements[index];
    if (element is IHitTestable ht && ht.HitTest(ref context.TestPoint, context.Scale))
    {
        context.LocalResult.Add(element);
    }
}

internal MyElement[] HitTest((double x, double y) point, double scale)
{
    var testPoint = new Point { X = point.x, Y = point.y };
    var finalResult = new List<MyElement>();

    Parallel.For(0, UIElements.Length,
        () => new List<MyElement>(),
        (index, loopState, localList) =>
        {
            var context = new HitTestContext
            {
                TestPoint = testPoint,
                Scale = scale,
                Elements = UIElements,
                LocalResult = localList
            };
            ExecuteHitTest(index, loopState, context);
            return localList;
        },
        localList =>
        {
            lock (finalResult)
            {
                finalResult.AddRange(localList);
            }
        });

    return finalResult.ToArray();
}

结合性能优化规划的额外建议 #

• 优先做视口剔除：在并行测试前，先通过元素Bounds与视口矩形的交集判断，过滤出需要测试的元素，直接减少并行处理的总量，从根源降低竞争和计算开销。
• SIMD批量预处理：对视口内的元素，将边界数据打包为SIMD兼容的结构，批量判断元素是否可能被命中，进一步提升单线程处理效率。
• 延迟引入空间划分结构：只有当视口剔除+SIMD+并行的组合仍无法满足性能要求时，再实现四叉树等空间索引结构，避免过早增加代码复杂度。

内容的提问来源于stack exchange，提问作者NWoodsman