go源码分析(五) 获取函数名和调用者的函数名

时间:2021-07-21 07:34:35

参考资料

实现代码保存在我的github

// input flag 1:FunName 2:CallerFunName
func GetFuncName(flag int) string {
    if flag != 1 && flag != 2 {
        flag = 1
    }
    pc, _, _, _ := Caller(flag)
    return FuncForPC(pc).Name()
}

测试代码

func main() {
    test()
}
func test{
    fmt.Println(runtime.GetFuncName(1), runtime.GetFuncName(2), runtime.GetFuncName(3))
}

输出

  main.test main.main main.test

1 为函数名,2 为 调用者名,3 未识别的调用,设置成1

其实0 就是函数本身 GetFuncName,为一个函数调用栈