Go 并发模型与多核 CPU 利用：深入理解 GOMAXPROCS

本文旨在澄清关于 Go 语言并发模型及其多核 CPU 利用的常见误解。许多人认为 Go 使用用户空间线程，从而限制了其在多核处理器上的性能。本文将深入探讨 Go 的线程模型，解释其如何利用操作系统线程，并通过 GOMAXPROCS() 函数充分发挥多核 CPU 的性能，并提供实际代码示例。

Go 的并发模型：OS 线程与 Goroutine

关于 Go 语言是否使用用户空间线程，以及是否能有效利用多核 CPU 的问题，常常引起讨论。 事实上，Go 并非使用用户空间线程。 它使用操作系统（OS）线程，因此可以充分利用多核处理器的能力。

理解 Go 的并发模型，需要区分两个关键概念：goroutine 和 OS 线程。

• Goroutine: Go 语言中的并发执行单元，类似于轻量级线程。Goroutine 的创建和管理由 Go 运行时负责，开销远小于传统的 OS 线程。
• OS 线程: 操作系统提供的线程，由内核进行调度和管理。

Go 运行时会将多个 goroutine 调度到少量的 OS 线程上执行。这种 M:N 调度模型，使得 Go 能够高效地处理大量并发任务，而无需为每个 goroutine 创建一个 OS 线程。

GOMAXPROCS：释放多核潜力

默认情况下，Go 程序只会使用一个 OS 线程来运行所有的 goroutine。这意味着即使你的机器拥有多个 CPU 核心，Go 程序也只能利用其中一个核心。为了充分利用多核 CPU 的性能，我们需要使用 runtime.GOMAXPROCS() 函数。

GOMAXPROCS() 函数用于设置 Go 程序可以同时使用的 OS 线程的最大数量。通过将 GOMAXPROCS() 设置为 CPU 核心数，我们可以让 Go 程序将 goroutine 分配到多个 OS 线程上并行执行，从而充分利用多核 CPU 的计算能力。

百度智能云·曦灵

百度旗下的AI数字人平台

83

查看详情

示例代码：

package main

import (
    "fmt"
    "runtime"
    "time"
)

func worker(id int) {
    for i := 0; i < 100000; i++ {
        // 模拟一些计算密集型任务
        _ = i * i
    }
    fmt.Printf("Worker %d finished\n", id)
}

func main() {
    // 设置 GOMAXPROCS 为 CPU 核心数
    runtime.GOMAXPROCS(runtime.NumCPU())
    fmt.Printf("Running with GOMAXPROCS = %d\n", runtime.GOMAXPROCS(-1)) // -1 获取当前值

    startTime := time.Now()

    // 启动多个 goroutine
    for i := 0; i < 4; i++ {
        go worker(i)
    }

    // 等待所有 goroutine 完成
    time.Sleep(5 * time.Second)
    endTime := time.Now()

    fmt.Printf("Total execution time: %v\n", endTime.Sub(startTime))
}

代码解释：

1. runtime.NumCPU() 函数返回当前机器的 CPU 核心数。
2. runtime.GOMAXPROCS(runtime.NumCPU()) 将 GOMAXPROCS 设置为 CPU 核心数，允许 Go 程序使用多个 OS 线程。
3. 启动 4 个 goroutine 执行 worker() 函数，模拟计算密集型任务。
4. time.Sleep(5 * time.Second) 确保所有 goroutine 完成执行。

运行结果对比：

• 未设置 GOMAXPROCS (默认为 1): 程序执行时间较长，因为所有 goroutine 都在同一个 OS 线程上串行执行。
• 设置 GOMAXPROCS 为 CPU 核心数: 程序执行时间明显缩短，因为 goroutine 可以并行执行在不同的 OS 线程上，充分利用了多核 CPU 的性能。

注意事项：

• GOMAXPROCS() 的设置应该在程序启动时进行，通常在 main() 函数中设置。
• 过度增加 GOMAXPROCS() 的值可能导致上下文切换开销增加，反而降低性能。通常情况下，将其设置为 CPU 核心数即可。
• 对于 I/O 密集型任务，增加 GOMAXPROCS() 的效果可能不明显，因为瓶颈在于 I/O 操作，而不是 CPU 计算。

总结

Go 语言使用 OS 线程，并通过 M:N 调度模型高效地管理 goroutine。通过合理设置 GOMAXPROCS()，我们可以充分利用多核 CPU 的性能，提升 Go 程序的并发处理能力。 理解 Go 的并发模型，并正确使用 GOMAXPROCS() 函数，是编写高性能 Go 程序的关键。

以上就是Go 并发模型与多核 CPU 利用：深入理解 GOMAXPROCS的详细内容，更多请关注php中文网其它相关文章！