本文参与了思否技术征文,欢迎正在阅读的你也加入。

光看标题,大家可能不太理解我说的是啥。

我们平时创建一个协程,跑一段逻辑,代码大概长这样。

package main

import (
    "fmt"
    "time"
)
func Foo() {
    fmt.Println("打印1")
    defer fmt.Println("打印2")
    fmt.Println("打印3")
}

func main() {
    go  Foo()
    fmt.Println("打印4")
    time.Sleep(1000*time.Second)
}

// 这段代码,正常运行会有下面的结果
打印4
打印1
打印3
打印2
defer
Foo()
打印4
打印1
打印2

也不卖关子了,我这边直接说答案。

runtime.Goexit()defer
package main

import (
    "fmt"
    "runtime"
    "time"
)
func Foo() {
    fmt.Println("打印1")
    defer fmt.Println("打印2")
    runtime.Goexit() // 加入这行
    fmt.Println("打印3")
}

func main() {
    go  Foo()
    fmt.Println("打印4")
    time.Sleep(1000*time.Second)
}


// 输出结果
打印4
打印1
打印2

可以看到打印3这一行没出现了,协程确实提前结束了。

其实面试题到这里就讲完了,这一波自问自答可还行?

但这不是今天的重点,我们需要搞搞清楚内部的逻辑。

runtime.Goexit()是什么?

看一下内部实现。

func Goexit() {
    // 以下函数省略一些逻辑...
    gp := getg() 
    for {
    // 获取defer并执行
        d := gp._defer
        reflectcall(nil, unsafe.Pointer(d.fn), deferArgs(d), uint32(d.siz), uint32(d.siz))
    }
    goexit1()
}

func goexit1() {
    mcall(goexit0)
}
runtime.Goexit()defer
goexit1goexit0
goexit0
// goexit continuation on g0.
func goexit0(gp *g) {
  // 获取当前的 goroutine
    _g_ := getg()
    // 将当前goroutine的状态置为 _Gdead
    casgstatus(gp, _Grunning, _Gdead)
  // 全局协程数减一
    if isSystemGoroutine(gp, false) {
        atomic.Xadd(&sched.ngsys, -1)
    }
  
  // 省略各种清空逻辑...

  // 把g从m上摘下来。
  dropg()


    // 把这个g放回到p的本地协程队列里,放不下放全局协程队列。
    gfput(_g_.m.p.ptr(), gp)

  // 重新调度,拿下一个可运行的协程出来跑
    schedule()
}

这段代码,信息密度比较大。

很多名词可能让人一脸懵。

MGgoroutinePG和M之间GM
G

暂时不能运行的G,P上会有个本地队列去存放这些这些G,P的本地队列存不下的话,还有个全局队列,干的事情也类似。

goexit0_Gdead

所以简单总结一下,只要执行 goexit 这个函数,当前协程就会退出,同时还能调度下一个可执行的协程出来跑。

runtime.Goexit()

goexit的用途

goexit

有个小细节,不知道大家平时debug的时候有没有关注过。

为了说明问题,这里先给出一段代码。

package main

import (
    "fmt"
    "time"
)
func Foo() {
    fmt.Println("打印1")
}

func main() {
    go  Foo()
    fmt.Println("打印3")
    time.Sleep(1000*time.Second)
}
gogoroutineFoo()sleep
Foo()debug
runtime.goexit()

如果大家平时有注意观察,会发现,其实所有的堆栈底部,都是从这个函数开始的。我们继续跟跟代码。

goexit是什么?

debugruntimegoexit1()
// The top-most function running on a goroutine
// returns to goexit+PCQuantum.
TEXT runtime·goexit(SB),NOSPLIT,$0-0
    BYTE    $0x90    // NOP
    CALL    runtime·goexit1(SB)    // does not return
    // traceback from goexit1 must hit code range of goexit
    BYTE    $0x90    // NOP
pruntime/proc.go
// 省略部分代码
func goexit1() {
    mcall(goexit0)
}
runtime.Goexit()goexit0

为什么每个堆栈底部都是这个方法?

我们首先需要知道的是,函数栈的执行过程,是先进后出。

假设我们有以下代码

func main() {
    B()
}

func B() {
    A()
}

func A() {

}

上面的代码是main运行B函数,B函数再运行A函数,代码执行时就跟下面的动图那样。

main()goexitgoexit
goexit


goexit
runtime/proc.gonewproc1
func newproc1(fn *funcval, argp unsafe.Pointer, narg int32, callergp *g, callerpc uintptr) {
    // 获取当前g
  _g_ := getg()
    // 获取当前g所在的p
    _p_ := _g_.m.p.ptr()
  // 创建一个新 goroutine
    newg := gfget(_p_)

    // 底部插入goexit
    newg.sched.pc = funcPC(goexit) + sys.PCQuantum 
    newg.sched.g = guintptr(unsafe.Pointer(newg))
    // 把新创建的g放到p中
    runqput(_p_, newg, true)

    // ...
}

主要的逻辑是获取当前协程G所在的调度器P,然后创建一个新G,并在栈底插入一个goexit。

所以我们每次debug的时候,就都能看到函数栈底部有个goexit函数。

main函数也是个协程,栈底也是goexit?

goexit
goexit()
asm_amd64.sruntime·mainPCruntime.main
    // create a new goroutine to start program
    MOVQ    $runtime·mainPC(SB), AX        // 也就是runtime.main
    PUSHQ    AX
    PUSHQ    $0            // arg size
    CALL    runtime·newproc(SB)
runtime·newprocruntime.mainruntime.mainmain.main
// runtime/proc.go
func main() {
    // 省略大量代码
    fn := main_main // 其实就是我们的main函数入口
    fn() 
}

//go:linkname main_main main.main
func main_main()

结论是,其实main函数也是由newproc创建的,只要通过newproc创建的goroutine,栈底就会有一个goexit。

os.Exit()和runtime.Goexit()有什么区别

最后再回到开头的问题,实现一下首尾呼应。

runtime.Goexit()os.Exit()
os.Exit()runtime.Goexit()
os.Exit()
package main

import (
    "fmt"
    "os"
    "time"
)
func Foo() {
    fmt.Println("打印1")
    defer fmt.Println("打印2")
    os.Exit(0)
    fmt.Println("打印3")
}

func main() {
    go  Foo()
    fmt.Println("打印4")
    time.Sleep(1000*time.Second)
}

// 输出结果
打印4
打印1

总结

runtime.Goexit()runtime.Goexit()newprocgoroutineos.Exit()runtime.Goexit()

最后

无用的知识又增加了。

一般情况下,业务开发中,谁会没事执行这个函数呢?

但是开发中不关心,不代表面试官不关心!

下次面试官问你,如果想在goroutine执行一半就退出协程,该怎么办?你知道该怎么回答了吧?


好了,兄弟们,有没有发现这篇文章写的又水又短,真的是因为我变懒了吗?

不!

当然不!

我是为了兄弟们的身体健康考虑,保持蹲姿太久对身体不好,懂?


如果文章对你有帮助,欢迎.....

算了。

一起在知识的海洋里呛水吧

我是小白,我们下期见!

参考资料

饶大的《哪来里的 goexit?》- https://qcrao.com/2021/06/07/...