Go语言循环并行化咨询：粒子模拟程序多核提速实现方案

你之前使用goroutine得到错误结果，大概率是踩到了循环变量捕获、共享变量并发读写这两个Go并行场景的常见坑，正确并行实现方案如下：

前置修改 #

首先导入同步依赖包：

import (
    "sync"
    // 原有其他导入保持不变
)

并行化后的循环代码 #

将你标注的需要并行的循环替换为以下实现：

//// 并行化粒子循环
var wg sync.WaitGroup
wg.Add(n_particles + 1)

// 单独创建新角度数组，避免并发读写冲突
newAngles := angles

for j := 0; j <= n_particles; j++ {
    // 必须将循环变量作为参数传入goroutine，避免变量复用导致的取值错误
    go func(idx int) {
        defer wg.Done()
        sumanglesx := 0.0
        sumanglesy := 0.0
        ix, iy := int64(math.Floor(x[idx])), int64(math.Floor(y[idx]))

        for dx := -1; dx <= 1; dx++ {
            for dy := -1; dy <= 1; dy++ {
                adjacentParticles := adjacencyIndex[intpos{ix + int64(dx), iy + int64(dy)}]
                for _, k := range adjacentParticles {
                    dist := ((x[k] - x[idx]) * (x[k] - x[idx])) + ((y[k] - y[idx]) * (y[k] - y[idx]))
                    if dist < radius {
                        sy, sx := math.Sincos(angles[k])
                        if k <= idx {
                            sumanglesx += sx
                            sumanglesy += sy
                        } else {
                            sumanglesx += alpha * sx
                            sumanglesy += alpha * sy
                        }
                    }
                }
            }
        }
        // 每个goroutine仅写入自己负责的下标位置，无并发冲突无需加锁
        r_angles := math.Atan2(sumanglesy, sumanglesx)
        // 你原有随机噪声逻辑可直接在这里补充，随机源加上idx避免不同goroutine生成相同随机序列
        // bsource := rand.NewSource(time.Now().UnixNano() + int64(idx))
        // bandomizer := rand.New(bsource)
        newAngles[idx] = r_angles
    }(j)
}
// 等待所有粒子计算完成
wg.Wait()
// 统一更新全局角度数组
angles = newAngles

关键优化说明 #

• 读写分离：当前步计算仅读取全局的x、y、angles数组，计算结果全部写入独立的newAngles数组，所有goroutine写入位置互不重叠，无锁开销
• 随机数优化：每个goroutine单独创建随机源，避免全局rand的锁开销，性能更高
• 若粒子数规模进一步扩大，可固定启动和CPU核心数相等的worker协程，通过通道分发计算任务，进一步减少协程创建销毁的开销

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