Go 并行程序性能优化：深入分析与实践

聖光之護

发布时间：2025-08-28 19:33:00

840人浏览过

来源于php中文网

原创

go 并行程序性能优化：深入分析与实践

本文针对 Go 语言并行程序中出现的性能瓶颈问题，以一个大整数分解的例子入手，深入分析了 big.Int 类型在并行计算中的性能问题根源，并提供了优化建议。文章重点讨论了内存分配对并行性能的影响，并指出了程序中潜在的并发安全问题，旨在帮助读者更好地理解和优化 Go 并行程序。

性能瓶颈分析：big.Int 与内存分配

在 Go 语言中，big.Int 类型用于处理任意精度的整数。然而，在高并发场景下，频繁使用 big.Int 可能会导致性能问题。问题在于，big.Int 的许多方法（如 Mod、Add 等）需要在堆上分配内存来存储计算结果。由于 Go 语言的堆是全局共享的，所有 goroutine 都在竞争同一块内存区域，导致内存分配操作串行化，从而限制了并行程序的性能。

例如，在提供的代码中，factorize 函数中频繁使用 m.Mod(n, i) 进行取模运算。每次调用 Mod 都会分配新的 big.Int 对象来存储余数，这在高并发场景下会造成严重的性能瓶颈。

优化方案

避免不必要的 big.Int 使用: 如果可以，尽量使用内置的整数类型（如 int64）代替 big.Int。在示例代码中，如果被分解的数和可能的因子都在 int64 的范围内，则可以使用 int64 类型进行计算，从而避免内存分配开销。

package main

import (
    "fmt"
    "runtime"
    "sync"
)

func factorize(n int64, start int64, step int64, c chan int64) {
    for i := start; ; i += step {
        if n%i == 0 {
            c <- i
            return // 找到一个因子后退出
        }
    }
}

func main() {
    numCPUs := runtime.NumCPU()
    runtime.GOMAXPROCS(numCPUs)

    n := int64(28808539627864609)
    numProcesses := numCPUs // 使用 CPU 核心数作为进程数
    c := make(chan int64)

    var wg sync.WaitGroup
    wg.Add(numProcesses)

    for i := 0; i < numProcesses; i++ {
        go func(i int) {
            defer wg.Done()
            factorize(n, int64(2+i), int64(numProcesses), c)
        }(i)
    }

    go func() {
        wg.Wait()
        close(c) // 关闭 channel
    }()

    factor := <-c // 从 channel 接收结果
    fmt.Println("Factor:", factor)
}

注意事项: 上述代码使用了 sync.WaitGroup 来确保所有 goroutine 都完成后再关闭 channel，并使用了 return 语句在找到一个因子后退出 goroutine，避免了后续的并发安全问题。

减少内存分配: 如果必须使用 big.Int，尽量复用已有的 big.Int 对象，避免频繁创建新的对象。例如，可以将 m 声明在循环外部，然后在每次迭代中更新其值。
```
func factorize(n *big.Int, start int, step int, c chan *big.Int) {
    var m big.Int
    i := big.NewInt(int64(start))
    s := big.NewInt(int64(step))
    z := big.NewInt(0)

    for {
        m.Mod(n, i) // 复用 m
        if m.Cmp(z) == 0{
            c <- big.NewInt(0).Set(i) // 发送 i 的拷贝
            return // 找到一个因子后退出
        }
        i.Add(i, s)
    }
}
```
注意事项: 由于 big.Int 是指针类型，直接发送 i 的指针会导致并发安全问题。因此，需要创建一个新的 big.Int 对象，并将 i 的值复制到新的对象中，再发送新的对象的指针。

知我AI
一款多端AI知识助理，通过一键生成播客/视频/文档/网页文章摘要、思维导图，提高个人知识获取效率；自动存储知识，通过与知识库聊天，提高知识利用效率。

下载
使用更高效的算法: 在 big.Int 的方法中，不同的方法性能差异很大。例如，QuoRem 方法同时计算商和余数，通常比单独使用 Mod 和 Quo 方法更高效。
调整 GOMAXPROCS: runtime.GOMAXPROCS 设置了 Go 程序可以同时使用的 CPU 核心数。将其设置为 CPU 核心数可以提高并行程序的性能。

并发安全问题

原代码存在一个并发安全问题：当一个 goroutine 找到一个因子并通过 channel 发送结果时，它发送的是指向局部变量 i 的指针。但是，该 goroutine 并没有退出循环，而是继续递增 i 的值。当主 goroutine 从 channel 接收到指针时，i 的值可能已经被修改，导致结果错误。

为了解决这个问题，需要在找到一个因子后退出 goroutine。此外，由于 big.Int 是指针类型，发送指针可能会导致数据竞争。因此，应该发送 i 的拷贝，而不是 i 的指针。