您好,欢迎来到12图资源库!分享精神,快乐你我!我们只是素材的搬运工!!
  • 首 页
  • 当前位置:首页 > 开发 > WEB开发 >
    如何在Go的函数中失掉调用者函数名?
    时间:2018-11-05 21:05 来源:网络整理 作者:网络 浏览:收藏 挑错 推荐 打印

    有时分在Go的函数调用的进程中,我们需求知道函数被谁调用,比如打印日志信息等。例如下面的函数,我们希望在日志中打印出调用者的名字。

    如何在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)