有时分在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等函数更详细的跟踪函数的调用堆栈。
1、func Caller(skip int) (pc uintptr, file string, line int, ok bool)
Caller可以前往函数调用栈的某一层的顺序计数器、文件信息、行号。
0 代表以后函数,也是调用runtime.Caller的函数。1 代表上一层调用者,以此类推。
2、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
3、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()
(责任编辑:admin)