Go并行度验证：GOMAXPROCS效果测试方法

理解Go并发模型与GOMAXPROCS

在Go语言中，GOMAXPROCS是一个环境变量或通过runtime.GOMAXPROCS函数设置的参数，它控制了Go运行时可以同时用于执行Go代码（goroutine）的操作系统线程数量。简而言之，它决定了Go调度器能够并行运行goroutine的“最大工人”数量。

• runtime.GOMAXPROCS(n int): 这个函数用于设置或获取GOMAXPROCS的值。
  • 当传入一个大于0的整数n时，它会将GOMAXPROCS设置为n，并返回之前的值。
  • 当传入0时，它会返回当前的GOMAXPROCS值，而不改变其设置。
• 默认行为: 从Go 1.5版本开始，GOMAXPROCS的默认值被设置为机器上的逻辑CPU核心数（即runtime.NumCPU()的返回值）。这意味着Go程序默认会尝试充分利用所有可用的CPU核心来并行执行任务。

获取系统CPU核心数：runtime.NumCPU()

runtime.NumCPU()函数返回当前机器上可用的逻辑CPU核心数。这通常包括物理核心和通过超线程技术模拟出的逻辑核心。这个值代表了操作系统能够提供的最大并行处理能力。

确定程序实际并行度

一个Go程序在任何给定时刻能够实际并行执行的最大CPU核心数，受限于两个因素的最小值：

1. GOMAXPROCS的设置值: 这是Go运行时自身对并行度的上限。
2. runtime.NumCPU()的返回值: 这是系统硬件实际提供的逻辑CPU核心数。

因此，程序实际的最大并行度是min(runtime.GOMAXPROCS(0), runtime.NumCPU())。

实现并行度验证函数

为了方便地验证Go程序当前配置下的实际最大并行度，我们可以编写一个简单的函数：

package main

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

var wg sync.WaitGroup

// MaxParallelism 返回Go程序当前配置下能够使用的最大并行CPU核心数。
// 它取 runtime.GOMAXPROCS(0) 和 runtime.NumCPU() 中的最小值。
func MaxParallelism() int {
    maxProcs := runtime.GOMAXPROCS(0) // 获取当前 GOMAXPROCS 设置，不改变
    numCPU := runtime.NumCPU()        // 获取系统逻辑CPU核心数
    if maxProcs < numCPU {
        return maxProcs
    }
    return numCPU
}

// doTasks 模拟一个CPU密集型任务
func doTasks(id int) {
    fmt.Printf("Goroutine %d: 开始执行任务...\n", id)
    start := time.Now()
    // 模拟长时间的CPU密集型计算，避免编译器优化
    for ji := 1; ji < 100000000; ji++ {
        for io := 1; io < 10; io++ {
            _ = ji * io
        }
    }
    duration := time.Since(start)
    fmt.Printf("Goroutine %d: 任务完成，耗时 %s\n", id, duration)
    wg.Done()
}

func main() {
    fmt.Printf("系统逻辑CPU核心数: %d\n", runtime.NumCPU())

    // 1. 初始状态下的并行度 (Go 1.5+ 默认为 runtime.NumCPU())
    fmt.Printf("\n--- 初始状态 ---\n")
    fmt.Printf("当前 GOMAXPROCS 设置: %d\n", runtime.GOMAXPROCS(0))
    fmt.Printf("程序实际最大并行度: %d\n", MaxParallelism())

    // 2. 示例: GOMAXPROCS 设为 1
    fmt.Printf("\n--- 示例