runtime 調度器是個非常有用的東西,關于 runtime 包幾個方法:
- Gosched:讓當前線程讓出 cpu 以讓其它線程運行,它不會掛起當前線程,因此當前線程未來會繼續執行
- NumCPU:返回當前系統的 CPU 核數量
- GOMAXPROCS:設置最大的可同時使用的 CPU 核數
- Goexit:退出當前 goroutine(但是defer語句會照常執行)
- NumGoroutine:返回正在執行和排隊的任務總數
- GOOS:目標操作系統
NumCPU
package main
import (
"fmt"
"runtime"
)
func main() {
fmt.Println("cpus:", runtime.NumCPU())
fmt.Println("goroot:", runtime.GOROOT())
fmt.Println("archive:", runtime.GOOS)
}
運行結果:

GOMAXPROCS
Golang 默認所有任務都運行在一個 cpu 核里,如果要在 goroutine 中使用多核,可以使用 runtime.GOMAXPROCS 函數修改,當參數小于 1 時使用默認值。
package main
import (
"fmt"
"runtime"
)
func init() {
runtime.GOMAXPROCS(1)
}
func main() {
// 任務邏輯...
}
Gosched
這個函數的作用是讓當前 goroutine 讓出 CPU,當一個 goroutine 發生阻塞,Go 會自動地把與該 goroutine 處于同一系統線程的其他 goroutine 轉移到另一個系統線程上去,以使這些 goroutine 不阻塞
package main
import (
"fmt"
"runtime"
)
func init() {
runtime.GOMAXPROCS(1) //使用單核
}
func main() {
exit := make(chan int)
go func() {
defer close(exit)
go func() {
fmt.Println("b")
}()
}()
for i := 0; i 4; i++ {
fmt.Println("a:", i)
if i == 1 {
runtime.Gosched() //切換任務
}
}
-exit
}
結果:

使用多核測試:
package main
import (
"fmt"
"runtime"
)
func init() {
runtime.GOMAXPROCS(4) //使用多核
}
func main() {
exit := make(chan int)
go func() {
defer close(exit)
go func() {
fmt.Println("b")
}()
}()
for i := 0; i 4; i++ {
fmt.Println("a:", i)
if i == 1 {
runtime.Gosched() //切換任務
}
}
-exit
}
結果:

根據你機器來設定運行時的核數,但是運行結果不一定與上面相同,或者在 main 函數的最后加上 select{} 讓程序阻塞,則結果如下:

多核比較適合那種 CPU 密集型程序,如果是 IO 密集型使用多核會增加 CPU 切換的成本。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
您可能感興趣的文章:- 如何判斷Golang接口是否實現的操作
- 淺談golang中的&^位清空操作
- Golang之defer 延遲調用操作
- 解決golang sync.Wait()不執行的問題
- golang執行命令操作 exec.Command
- golang等待觸發事件的實例
- 對Golang中的runtime.Caller使用說明