当前位置：首页>后端>正文

golang grpc stream 附件 golang stack

后端2024-04-22 11:14:38

go在线程的基础上实现了用户态更加轻量级的写成，线程栈为了防止stack overflow，默认大小一般是2MB，而在go中，协程栈在初始化时是2KB

go中的栈是可以扩容的，在64位操作系统上最大为1GB

1. newstack()函数

在函数序言阶段如果判断出需要扩容，则会跳转调用运行时morestack_noctxt函数，函数调用链为：

morestack_noctxt() -> morestack() -> newstack()

核心代码位于 newstack() 函数中，newstack()函数不仅会处理扩容，还会处理协程的抢占

下面看一下newstack()函数的核心实现：

func newstack() {
    oldsize := gp.stack.hi - gp.stack.lo

    // 两倍于原来大小
    newsize := oldsize * 2

    // 需要的栈太大，直接溢出
    if newsize > maxstacksize {
        throw( "stack overflow" )
    }

    // goroutine必须是正在执行过程中才会调用newstack
    // 所以这个状态一定是Grunning或者Gscanrunning
    casgstatus(gp, _Grunning, _Gcopystack)

    // gp的处于Gcopystack状态，当我们对栈进行复制时并发GC不会扫描此栈
    // 栈的复制
    copystack(gp, newsize)
    casgstatus(gp, _Gcopystack, _Grunning)

    // 继续执行
    gogo(&gp.sched)
}

什么是gp？

gp就是当前协程的结构体：

type g  struct {
    stack stack
    stackguard0 uintptr
    stackguard1 uintptr
    ... 
}

type stack  struct {
    lo uintptr  // 8 bytes
    hi uintptr
}

gp.stack.hi - gp.stack.lo就是在计算当前协程栈的大小

newstack()函数首先通过栈底地址与栈顶地址计算出旧栈的大小，并计算新栈的大小，新栈大小为旧栈的两倍大。在64为操作系统中，如果栈大小超过了1GB(maxstacksize)则直接报错stack overflow