runtime.Gosched()

让出CPU时间片,重新等待安排任务

  1. package main
  2. import (
  3. "fmt"
  4. "runtime"
  5. )
  6. func main() {
  7. go func(s string) {
  8. for i := 0; i < 2; i++ {
  9. fmt.Println(s)
  10. }
  11. }("world")
  12. // 主协程
  13. for i := 0; i < 2; i++ {
  14. // 切一下,再次分配任务
  15. runtime.Gosched()
  16. fmt.Println("hello")
  17. }
  18. }

runtime.Goexit()

退出当前协程

  1. package main
  2. import (
  3. "fmt"
  4. "runtime"
  5. )
  6. func main() {
  7. go func() {
  8. defer fmt.Println("A.defer")
  9. func() {
  10. defer fmt.Println("B.defer")
  11. // 结束协程
  12. runtime.Goexit()
  13. defer fmt.Println("C.defer")
  14. fmt.Println("B")
  15. }()
  16. fmt.Println("A")
  17. }()
  18. for {
  19. }
  20. }

runtime.GOMAXPROCS

Go运行时的调度器使用GOMAXPROCS参数来确定需要使用多少个OS线程来同时执行Go代码。默认值是机器上的CPU核心数。例如在一个8核心的机器上,调度器会把Go代码同时调度到8个OS线程上(GOMAXPROCS是m:n调度中的n)。
Go语言中可以通过runtime.GOMAXPROCS()函数设置当前程序并发时占用的CPU逻辑核心数。
Go1.5版本之前,默认使用的是单核心执行。Go1.5版本之后,默认使用全部的CPU逻辑核心数。

可以通过将任务分配到不同的CPU逻辑核心上实现并行的效果,这里举个例子:

  1. func a() {
  2. for i := 1; i < 10; i++ {
  3. fmt.Println("A:", i)
  4. }
  5. }
  6. func b() {
  7. for i := 1; i < 10; i++ {
  8. fmt.Println("B:", i)
  9. }
  10. }
  11. func main() {
  12. runtime.GOMAXPROCS(1)
  13. go a()
  14. go b()
  15. time.Sleep(time.Second)
  16. }

两个任务只有一个逻辑核心,此时是做完一个任务再做另一个任务。 将逻辑核心数设为2,此时两个任务并行执行,代码如下。

  1. func a() {
  2. for i := 1; i < 10; i++ {
  3. fmt.Println("A:", i)
  4. }
  5. }
  6. func b() {
  7. for i := 1; i < 10; i++ {
  8. fmt.Println("B:", i)
  9. }
  10. }
  11. func main() {
  12. runtime.GOMAXPROCS(2)
  13. go a()
  14. go b()
  15. time.Sleep(time.Second)
  16. }

Go语言中的操作系统线程和goroutine的关系:

  • 一个操作系统线程对应用户态多个goroutine。
  • go程序可以同时使用多个操作系统线程。
  • goroutine和OS线程是多对多的关系,即m:n。