Go语言获取CPU核心数

后端

2024-02-19 17:07:56

Go语言中的CPU核心数

在Go语言中，CPU核心数是指计算机中可以同时执行指令的处理器内核的数量。获取CPU核心数可以帮助我们更好地了解计算机的性能，并根据核心数来优化Go程序的并发性能。

获取CPU核心数的方法

Go语言提供了两种获取CPU核心数的方法：

runtime.NumCPU()：此函数返回当前可用的CPU核心数。
GOMAXPROCS：此变量表示Go程序可以同时执行的最大goroutine数量，默认值为CPU核心数。

runtime.NumCPU()

runtime.NumCPU()函数返回当前可用的CPU核心数。此函数在Go程序启动时自动调用，并将结果存储在runtime.NumCPU变量中。因此，我们可以直接使用runtime.NumCPU变量来获取CPU核心数。

package main

import (
	"fmt"
	"runtime"
)

func main() {
	// 获取CPU核心数
	numCPU := runtime.NumCPU()

	// 打印CPU核心数
	fmt.Println("Number of CPU cores:", numCPU)
}

输出：

Number of CPU cores: 8

GOMAXPROCS

GOMAXPROCS变量表示Go程序可以同时执行的最大goroutine数量，默认值为CPU核心数。我们可以通过设置GOMAXPROCS变量来限制Go程序同时执行的goroutine数量。

package main

import (
	"fmt"
	"runtime"
)

func main() {
	// 获取CPU核心数
	numCPU := runtime.NumCPU()

	// 设置GOMAXPROCS为CPU核心数的一半
	runtime.GOMAXPROCS(numCPU / 2)

	// 打印GOMAXPROCS的值
	fmt.Println("GOMAXPROCS:", runtime.GOMAXPROCS())
}

输出：

GOMAXPROCS: 4

区别和使用场景

runtime.NumCPU()函数返回当前可用的CPU核心数，而GOMAXPROCS变量则表示Go程序可以同时执行的最大goroutine数量。GOMAXPROCS变量的默认值为runtime.NumCPU()，但我们可以通过设置GOMAXPROCS变量来限制Go程序同时执行的goroutine数量。

在大多数情况下，我们不需要修改GOMAXPROCS变量。但是，在某些场景下，我们可能需要限制Go程序同时执行的goroutine数量，例如：