先给大家看一张图片：
相信这种情况大家都不陌生，这个是go程序发生panic错误时的报错信息，（小编最近经常遇到），但是有没有想过在不报错的情况下，怎么获取程序的调用方式呢？
有时候在Go的函数调用的过程中，我们需要知道函数被谁调用，比如打印日志信息等。例如下面的函数，我们希望在日志中打印出调用者的名字。
func Foo() {
fmt.Println("谁在调用我？")
bar()
}
func Bar() {
fmt.Println("谁又在调用我？")
}
首先打印函数本身的名称
最简单的方式就是硬编码。 因为在编译之前，我们肯定知道打印的时候所在哪个函数，但是更好的方式是编写一个通用的函数，比如下面的例子：
package main
import (
"fmt"
"runtime"
)
func main() {
Foo()
}
func Foo() {
fmt. Printf ("我是 %s, 谁在调用我?\n", printMyName())
Bar()
}
func Bar() {
fmt.Printf("我是 %s, 谁又在调用我?\n", printMyName())
}
func printMyName() string {
pc, _, _, _ := runtime.Caller(1)
return runtime.FuncForPC(pc).Name()
}
结果输出：
我是 main.Foo, 谁在调用我? 我是 main.Bar, 谁又在调用我?
可以看到函数在被调用的时候，printMyName把函数本身的名字打印出来了，注意这里Caller的参数是1, 因为我们将业务代码封装成了一个函数。
首先打印函数调用者的名称
将上面的代码修改一下，增加一个新的printCallerName的函数，可以打印调用者的名称。
func main() {
Foo()
}
func Foo() {
fmt.Printf("我是 %s, %s 在调用我!\n", printMyName(), printCallerName())
Bar()
}
func Bar() {
fmt.Printf("我是 %s, %s 又在调用我!\n", printMyName(), printCallerName())
}
func printMyName() string {
pc, _, _, _ := runtime.Caller(1)
return runtime.FuncForPC(pc).Name()
}
func printCallerName() string {
pc, _, _, _ := runtime.Caller(2)
return runtime.FuncForPC(pc).Name()
}
相关函数介绍
你可以通过runtime.Caller、runtime.Callers、runtime.FuncForPC等函数更详细的跟踪函数的调用堆栈。
0x1： func Caller(skip int) (pc uintptr, file string, line int, ok bool)
Caller可以返回函数调用栈的某一层的程序计数器、文件信息、行号。
0 代表当前函数，也是调用runtime.Caller的函数。1 代表上一层调用者，以此类推。
0x2：func Callers(skip int, pc []uintptr) int
Callers用来返回调用站的程序计数器, 放到一个uintptr中。
0 代表 Callers 本身，这和上面的Caller的参数的意义不一样，历史原因造成的。 1 才对应 这上面的 0。
比如在上面的例子中增加一个trace函数，被函数Bar调用。
……
func Bar() {
fmt.Printf("我是 %s, %s 又在调用我!\n", printMyName(), printCallerName())
trace()
}
func trace() {
pc := make([]uintptr, 10) // at least 1 entry needed
n := runtime.Callers(0, pc)
for i := 0; i < n; i++ {
f := runtime.FuncForPC(pc[i])
file, line := f.FileLine(pc[i])
fmt.Printf("%s:%d %s\n", file, line, f.Name())
}
}
输出结果可以看到这个goroutine的整个栈都打印出来了：
/usr/local/go/src/runtime/extern.go:218 runtime.Callers /Users/yuepan/go/src/git.intra.weibo.com/platform/tool/g/main.go:34 main.trace /Users/yuepan/go/src/git.intra.weibo.com/platform/tool/g/main.go:20 main.Bar /Users/yuepan/go/src/git.intra.weibo.com/platform/tool/g/main.go:15 main.Foo /Users/yuepan/go/src/git.intra.weibo.com/platform/tool/g/main.go:10 main.main /usr/local/go/src/runtime/proc.go:210 runtime.main /usr/local/go/src/runtime/asm_amd64.s:1334 runtime.goexit
0x3：func CallersFrames(callers []uintptr) *Frames
上面的Callers只是或者栈的程序计数器，如果想获得整个栈的信息，可以使用CallersFrames函数，省去遍历调用FuncForPC。
上面的trace函数可以更改为下面的方式：
func trace2() {
pc := make([]uintptr, 10) // at least 1 entry needed
n := runtime.Callers(0, pc)
frames := runtime.CallersFrames(pc[:n])
for {
frame, more := frames.Next()
fmt.Printf("%s:%d %s\n", frame.File, frame.Line, frame.Function)
if !more {
break
}
}
}
0x4：获取程序堆栈
在程序panic的时候，一般会自动把堆栈打出来，如果你想在程序中获取堆栈信息，可以通过debug.PrintStack()打印出来。比如你在程序中遇到一个Error,但是不期望程序panic,只是想把堆栈信息打印出来以便跟踪调试，你可以使用debug.PrintStack()。小例子：
package main
import (
"runtime/debug"
"fmt"
)
func test1() {
fmt.Printf("%s", debug.Stack())
debug.PrintStack()
}
func main() {
test1()
}
抑或，你自己读取堆栈信息，自己处理和打印：
func DumpStacks() {
buf := make([]byte, 16384)
buf = buf[:runtime.Stack(buf, true)]
fmt.Printf("=== BEGIN goroutine stack dump ===\n%s\n=== END goroutine stack dump ===", buf)
}
如果你觉得小编总结的不错，可以关注小编的公众号哦^_^
Golang小白一起学
