Go 并发编程 - runtime 协程调度(三)

Go Runtime

Go runtime 可以形象的理解为 Go 程序运行时的环境,类似于 JVM。不同于 JVM 的是,Go 的 runtime 与业务程序直接打包在一块,是一个可执行文件,直接运行在操作系统上,效率很高。

runtime 包含了一些 Go 的一些非常核心的功能:协程调度、垃圾回收、内存分配等。本文将着重介绍协程调度(GMP 模型)。

协程调度

协程调度是指 Go 如何管理和执行协程,Go 的协程调度基于 GMP 模型。即:

  • G (Goroutine):即 Go 的协程,包含了栈信息,代码指针,状态等;
  • M (Machine):代表一个工作线程,由操作系统直接分配;
  • P (Processor):处理器(Go 定义的一个概念,不是指 CPU),包含了协程运行的所需资源,如本地队列、全局队列、计数器等。

GMP 三者的关系:

  • P 的个数取决于设置的 runtime.GOMAXPROCS,默认是物理 CPU 的逻辑核心数量,比如四核八线程的 CPU,P 的数量就是 8;
  • M 的数量一般是多于 P 的,M 要想被 CPU 执行,必须先获取 P。没有获取 P 的 M,则处于休眠状态;
  • G 可以理解为代码本体,G 必须要被 P 调度进入 M,才可以被 CPU 执行;
  • P 包含了一个 LRQ (Local Run Queue)本地运行队列,保存着等待执行的协程(G)队列。没有被分配到 P 的 G,会被保存到 GRQ (Global Run Queue) 全局队列中,处于休眠状态。

假如主机是单逻辑 CPU 的,那么 GMP 是这样的:

红色部分表示休眠或者挂起状态,黄色代表等待执行,绿色表示正在运行。系统初始化了两个线程,但我们只有一个处理器(P), M1 没有获取到 P,所以只能休眠。M0 当前获取到 P ,正在处理 G0, LRQ 里面目前有三个 G 在排队等待被 M 运行,GRQ 里面保存着 G4、G5、G6,表示它们还没有分配到队列中。

P 这个时候会分别对 LRQ 进行周期队列轮转 和 GRO 周期性检查:

  • 队列轮转:LRQ 中的 G 被 P 调度到 M 中执行,每个 G 执行一段时间后,就会保存其上下文并放入队列尾部,然后取出队列头部的 G 进入 M 执行。
  • 周期性检查:P 会检查 GRQ 中是否有 G 正在等待运行,并将其放入 M 中执行,防止协程被饿死。
  • 在队列轮转中,如果当前正在运行的 G 遇到了系统调用,那么系统就会挂起当前 M0,释放 P,M1 就会绑定释放的 P,来继续执行其他协程。

假设 G0 遇到了系统调用:

等到 M1 中所有的协程执行完或者 M1 处理某个协程也遇到了了系统调用,就会重新释放 P 给其他空闲的 M。而另外一边 G0 的系统调用结束后,就会将 M0 线程从挂起状态变成休眠状态,并将 G0 放入 GRQ,等待被 P 重新调入 LRQ 中轮转执行。

如果我们的主机具备多个逻辑 CPU,创建了多个 P,那么就会变成多个线程并行执行:

多线程同时处理时,很有可能多个 LRQ 是不均衡的。假如上图的 M0 已经执行完了,其他线程还处于繁忙状态,M0 所绑定的 P 就会去检查 GQR,GQR 中也没有 G,那么它就会去偷取其他 LRQ 一部分的 G 来执行,一般每次会偷取一半。

runtime 包

runtime.GOMAXPROCS

runtime.GOMAXPROCS() 可以用来设置 P 的数量,一般设置为和逻辑 CPU 数量相等的值:

复制代码
fmt.Println(runtime.NumCPU())
runtime.GOMAXPROCS(runtime.NumCPU()) // 使用所有的逻辑 CPU

// 结果
我的主机 CPU 是16核24线程,所以会使用24个 P

runtime.Gosched

runtime.Gosched() 用于让出当前协程的运行时间片,也就是当 P 遇到它时,会先安排其他协程先执行:

复制代码
func main() {
	runtime.GOMAXPROCS(1)
	go func() {
		for i := 0; i < 5; i++ {
			fmt.Println("go")
		}
	}()

	runtime.Gosched()
	fmt.Println("hello")
}


// 结果
输入结果不是固定的,有可能是
go
go
go
go
go
hello
也有可能是
go
go
go
go
hello
go
也有可能是
hello

输出第一种情况容易理解,主协程让出了时间片,理所应当先打印 Go,但是如果子协程还没有来得及被调度或者打印,就会出现其他情况。

runtime.Goexit

runtime.Goexit() 会结束当前的协程,但是 defer 语句会正常执行。此语法不能在主函数中使用,会引发 panic:

复制代码
func main() {
	runtime.GOMAXPROCS(1)
	go func() {
		defer fmt.Println("defer不受影响")
		fmt.Println("我被执行了")
		runtime.Goexit()
		fmt.Println("我被跳过了")
	}()

	time.Sleep(1 * time.Second)
}

// 结果
我被执行了
defer不受影响

本系列文章:

  1. Go 并发编程 - Goroutine 基础 (一)
  2. Go 并发编程 - 并发安全(二)
  3. Go 并发编程 - runtime 协程调度(三)
相关推荐
He1955019 小时前
Go初级之十:错误处理与程序健壮性
开发语言·python·golang
不会吃萝卜的兔子10 小时前
go webrtc - 1 go基本概念
开发语言·golang·webrtc
岁忧14 小时前
(LeetCode 面试经典 150 题) 200. 岛屿数量(深度优先搜索dfs || 广度优先搜索bfs)
java·c++·leetcode·面试·go·深度优先
小红帽2.015 小时前
从零构建一款开源在线客服系统:我的Go语言实战之旅
开发语言·golang·开源
程序员爱钓鱼16 小时前
Go语言实战案例- 命令行参数解析器
后端·google·go
007php00717 小时前
Go语言面试:传值与传引用的区别及选择指南
java·开发语言·后端·算法·面试·golang·xcode
q5673152320 小时前
手把手教你用Go打造带可视化的网络爬虫
开发语言·爬虫·信息可视化·golang
戎码江湖20 小时前
使用CI/CD部署后端项目(gin)
ci/cd·golang·gin·后端自动部署项目·自动化部署项目
二哈不在线1 天前
代码随想录二刷之“贪心算法”~GO
算法·贪心算法·golang
君万1 天前
【LeetCode每日一题】94. 二叉树的中序遍历 104. 二叉树的最大深度
算法·leetcode·golang