Go语言中获取协程ID

简介

java同事都知道，线程会有对应的id，那么go语言中协程有id吗，其实是有的，但是不建议使用。

实在需要使用的话可以使用本文的例子获取

stack

我们先看一下runtime.Stack打印出来的栈结构，其中就包括了协程id

func main() {
	b := make([]byte, 64)
	b = b[:runtime.Stack(b, false)]
	fmt.Println(string(b))
}

goroutine 1 [running]:
main.main()
    /Users/shura/GolandProject
1
2
3
4
5
6
7
8
9

通过上面可以看出 1 就是需要的协程id

获取协程id

根据以上的结构，我们就可以截取到协程id

func main() {
	fmt.Println(GoID())
}

func GoID() uint64 {
	b := make([]byte, 64)
	b = b[:runtime.Stack(b, false)]
	b = bytes.TrimPrefix(b, []byte("goroutine "))
	b = b[:bytes.IndexByte(b, ' ')]
	n, _ := strconv.ParseUint(string(b), 10, 64)
	return n
}


输出 1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

优化

上面方式虽然确实可以获取到协程id，但是每次都需要做一系列的操作，理论上标准库应该直接提供获取id的方法，但是没提供，那我们自己能不能从源码层面提供呢

我们跟踪runtime.Stack方法进入源码可以看到协程的信息是由 gp := getg()获取到的

再进入getg()查看是一个g的结构体

type g struct {
    goid         int64
}
1
2
3

这里面就有一个关键的属性goid，但是由于是私有的我们获取不到，所以我们只能从源码层面去添加获取协程id的方法

找到该go文件的路径，例如/Users/shura/support/go/go/src/runtime,执行如下脚本

cat > /Users/shura/support/go/go/src/runtime/shura_goid.go << EOF
package runtime

func GoID() int64{
    return getg().goid
}
EOF
1
2
3
4
5
6
7

测试

func main() {
	fmt.Println(runtime.GoID())
	fmt.Println(GoID())
}

输出

1
1
1
2
3
4
5
6
7
8
9

---

欢迎关注，学习不迷路！