Go语言调度器之主动调度(20)

本文是《Go语言调度器源代码情景分析》系列的第20篇，也是第五章《主动调度》的第1小节。

---

Goroutine的主动调度是指当前正在运行的goroutine通过直接调用runtime.Gosched()函数暂时放弃运行而发生的调度。

主动调度完全是用户代码自己控制的，我们根据代码就可以预见什么地方一定会发生调度。比如下面的程序，在main goroutine中创建了一个新的我们称之为g2的goroutine去执行start函数，g2在start函数的循环中反复调用Gosched()函数放弃自己的执行权，主动把CPU让给调度器去执行调度。

package main

import (
    "runtime"
    "sync"
)

const N = 1

func main() {
    var wg sync.WaitGroup

    wg.Add(N)
    for i := 0; i < N; i++ {
        go start(&wg)
    }

    wg.Wait()
}

func start(wg *sync.WaitGroup) {
    for i := 0; i < 1000 * 1000 * 1000; i++ {
        runtime.Gosched()
    }

    wg.Done()
}

下面我们就从这个程序开始分析主动调度是如何实现的。

首先从主动调度的入口函数Gosched()开始分析。

runtime/proc.go : 262

// Gosched yields the processor, allowing other goroutines to run. It does not
// suspend the current goroutine, so execution resumes automatically.
func Gosched() {
    checkTimeouts() //amd64 linux平台空函数

    //切换到当前m的g0栈执行gosched_m函数
    mcall(gosched_m)
    //再次被调度起来则从这里开始继续运行
}

因为我们需要关注程序运行起来之后g2 goroutine的状态，所以这里用gdb配合源代码一起来进行调试和分析，首先使用b proc.go:266在Gosched函数的mcall(gosched_m)这一行设置一个断点，然后运行程序，等程序被断下来之后，反汇编一下程序当前正在执行的函数

(gdb) disass
Dump of assembler code for function main.start:
     0x000000000044fc90 <+>:mov   %fs:0xfffffffffffffff8,%rcx
     0x000000000044fc99 <+>:cmp   0x10(%rcx),%rsp
     0x000000000044fc9d <+>:jbe   0x44fcfa <main.start+>
     0x000000000044fc9f <+>:sub   $0x20,%rsp
     0x000000000044fca3 <+>:mov   %rbp,0x18(%rsp)
     0x000000000044fca8 <+>:lea   0x18(%rsp),%rbp
     0x000000000044fcad <+>:xor   %eax,%eax
     0x000000000044fcaf <+>:jmp   0x44fcd0 <main.start+>
     0x000000000044fcb1 <+>:mov   %rax,0x10(%rsp)
     0x000000000044fcb6 <+>:nop
     0x000000000044fcb7 <+>:nop
=> 0x000000000044fcb8 <+>:lea   0x241e1(%rip),%rax        # 0x473ea0
     0x000000000044fcbf <+>:mov   %rax,(%rsp)
     0x000000000044fcc3 <+>:callq 0x447380 <runtime.mcall>
     0x000000000044fcc8 <+>:mov   0x10(%rsp),%rax
     0x000000000044fccd <+>:inc   %rax
     0x000000000044fcd0 <+>:cmp   $0x3b9aca00,%rax
     0x000000000044fcd6 <+>:jl     0x44fcb1 <main.start+>
     0x000000000044fcd8 <+>:nop
    0x000000000044fcd9 <+>:mov   0x28(%rsp),%rax
     0x000000000044fcde <+>:mov   %rax,(%rsp)
     0x000000000044fce2 <+>:movq   $0xffffffffffffffff,0x8(%rsp)
     0x000000000044fceb <+>:callq 0x44f8f0 <sync.(*WaitGroup).Add>
     0x000000000044fcf0 <+>:mov   0x18(%rsp),%rbp
     0x000000000044fcf5 <+>:add   $0x20,%rsp
     0x000000000044fcf9 <+>:retq
     0x000000000044fcfa <+>:callq 0x447550 <runtime.morestack_noctxt>
     0x000000000044fcff <+>:jmp   0x44fc90 <main.start>

可以看到当前正在执行的函数是main.start而不是runtime.Gosched，在整个start函数中都找不到Gosched函数的身影，原来它被编译器优化了。程序现在停在了0x000000000044fcb8 <+40>: lea 0x241e1(%rip),%rax 这一指令处，该指令下面的第二条callq指令在调用runtime.mcall，我们首先使用si 2来执行2条汇编指令让程序停在下面这条指令处：

=> 0x000000000044fcc3 <+>: callq 0x447380 <runtime.mcall>

然后使用i r rsp rbp rip记录一下CPU的rsp、rbp和rip寄存器的值备用：

(gdb) i r rsprbprip
rsp   0xc000031fb0     0xc000031fb0
rbp   0xc000031fc8     0xc000031fc8
rip    0x44fcc3             0x44fcc3 <main.start+>

继续看0x000000000044fcc3位置的callq指令，它首先会把紧挨其后的下一条指令的地址0x000000000044fcc8放入g2的栈，然后跳转到mcall函数的第一条指令开始执行。回忆一下第二章我们详细分析过的mcall函数的执行流程，结合现在这个场景，mcall将依次完成下面几件事：

1. 把上面call指令压栈的返回地址0x000000000044fcc8取出来保存在g2的sched.pc字段，把上面我们查看到的rsp(0xc000031fb0)和rbp(0xc000031fc8)分别保存在g2的sched.sp和sched.bp字段，这几个寄存器代表了g2的调度现场信息；

2. 把保存在g0的sched.sp和sched.bp字段中的值分别恢复到CPU的rsp和rbp寄存器，这样完成从g2的栈到g0的栈的切换；

3. 在g0栈执行gosched_m函数（gosched_m函数是runtime.Gosched函数调用mcall时传递给mcall的参数）。

继续看gosched_m函数

runtime/proc.go : 2623

// Gosched continuation on g0.
func gosched_m(gp *g) {
    if trace.enabled { //traceback 不关注
        traceGoSched()
    }
    goschedImpl(gp)  //我们这个场景：gp = g2
}

gosched_m函数只是简单的在调用goschedImpl：

runtime/proc.go : 2608

func goschedImpl(gp *g) {
    ......
    casgstatus(gp, _Grunning, _Grunnable)
    dropg() //设置当前m.curg = nil, gp.m = nil
    lock(&sched.lock)
    globrunqput(gp) //把gp放入sched的全局运行队列runq
    unlock(&sched.lock)

    schedule() //进入新一轮调度
}

goschedImpl函数有一个g指针类型的形参，我们这个场景传递给它的实参是g2，goschedImpl函数首先把g2的状态从_Grunning设置为_Grunnable，并通过dropg函数解除当前工作线程m和g2之间的关系（把m.curg设置成nil，把g2.m设置成nil），然后通过调用我们已经分析过的globrunqput函数把g2放入全局运行队列之中。

g2被挂入全局运行队列之后，g2以及其它一些相关部分的状态和关系如下图所示：

从上图我们可以清晰的看到，g2被挂在了sched的全局运行队列里面，该队列有一个head头指针指向队列中的第一个g对象，还有一个tail尾指针指向队列中的最后一个g对象，队列中各个g对象通过g的schedlink指针成员相互链接起在一起；g2的sched结构体成员中保存了调度所需的所有现场信息（比如栈寄存器sp和bp的值，pc指令寄存器的值等等），这样当g2下次被schedule函数调度时，gogo函数会负责把这些信息恢复到CPU的rsp, rbp和rip寄存器中，从而使g2又得以从0x44fcc8地址处开始在g2的栈中执行g2的代码。

把g2挂入全局运行队列之后，goschedImpl函数继续调用schedule()进入下一轮调度循环，至此g2通过自己主动调用Gosched()函数自愿放弃了执行权，达到了调度的目的。