Go 函数中获取调用者的函数名和文件名及行号

131 阅读 0 评论 87 点赞

我是靠谱客的博主舒服滑板，最近开发中收集的这篇文章主要介绍Go 函数中获取调用者的函数名和文件名及行号，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

背景

我们在应用程序的代码中添加业务日志的时候，不论是什么级别的日志，除了我们主动传给 Logger 让它记录的信息外，这行日志是由哪个函数打印的、所在的位置也是非常重要的信息，不然排查问题的时候很有可能就犹如大海捞针。

对于在记录日志时记录调用 Logger 方法的调用者的函数名、行号这些信息。有的日志库支持，比如 Zap：

func main() {
  logger, _ := zap.NewProduction(zap.AddCaller())
  defer logger.Sync()

  logger.Info("hello world")
}

输出：

{"level":"info","ts":1587740198.9508286,"caller":"caller/main.go:9","msg":"hello world"}

不过如果想搞一个健壮的开发框架，不应该让自己跟某个日志库强绑定，更好的方法是开发一个日志的门面，程序里直接使用日志门面，再由门面调用日志库完成日志的记录。典型的 Java 的 slf4j 就是这个思路，程序里直接使用的是slf4j ，后面的 Logger 可以是 logback 也可以是 log4j 甚至是任何满足 slf4j 约定的日志库实现。

如果让我们用 Go 设计一个Log Facade，就需要我们自己在门面里获取调用者的函数名、文件位置了，那么在Go里面怎么实现这个功能呢?这就需要借助 runtime 标准库提供的 Caller 函数了。

本文主要介绍 runtime.Caller 的使用，上面说了那么多只是为了铺垫一下，学会它，在哪些地方可以应用上。

runtime.Caller

runtime.Caller 的函数签名如下：

func Caller(skip int) (pc uintptr, file string, line int, ok bool)

Caller 函数会报告当前 Go 程序调用栈所执行的函数的文件和行号信息。参数skip为要上溯的栈帧数，0 表示Caller的调用者(Caller所在的调用栈)，1 表示调用 Caller 调用者的调用者，以此类推。是不是有点晕，这

里举个例子：

func CallerA() {
  //获取的是 CallerA 这个函数的调用栈
  pc, file, lineNo, ok := runtime.Caller(0)
  //获取的是 CallerA函数的调用者的调用栈
  pc1, file1, lineNo1, ok1 := runtime.Caller(1)
}

函数的返回值为调用栈标识符、带路径的完整文件名、该调用在文件中的行号。如果无法获得信息，返回值 ok 会被设为 false。

获取调用者的函数名

runtime.Caller 返回值中第一个返回值是一个调用栈标识，通过它我们能拿到调用栈的函数信息 *runtime.Func，再进一步获取到调用者的函数名字，这里面会用到的函数和方法如下。

func FuncForPC(pc uintptr) *Func
func (*Func) Name

runtime.FuncForPC 函数返回一个表示调用栈标识符pc对应的调用栈的*Func;如果该调用栈标识符没有对应的调用栈，函数会返回nil。

Name 方法返回该调用栈所调用的函数的名字，上面说了runtime.FuncForPC 有可能会返回 nil，不过Name方法在实现的时候做了这种情况的判断，避免出现panic 的可能，所以我们可以放心大胆的使用。

func (f *Func) Name() string {
 if f == nil {
  return ""
 }
 fn := f.raw()
 if fn.isInlined() { // inlined version
  fi := (*funcinl)(unsafe.Pointer(fn))
  return fi.name
 }
 return funcname(f.funcInfo())
}

使用示例

下面看一个使用 runtime.Caller 和 runtime.FuncForPC 一起配合获取调用者信息的简单例子：

package main
import (
 "fmt"
 "path"
 "runtime"
)
func getCallerInfo(skip int) (info string) {
 pc, file, lineNo, ok := runtime.Caller(skip)
 if !ok {
  info = "runtime.Caller() failed"
  return
 }
 funcName := runtime.FuncForPC(pc).Name()
 fileName := path.Base(file) // Base函数返回路径的最后一个元素
 return fmt.Sprintf("FuncName:%s, file:%s, line:%d ", funcName, fileName, lineNo)
}
func main() {
 // 打印出getCallerInfo函数自身的信息
 fmt.Println(getCallerInfo(0))
 // 打印出getCallerInfo函数的调用者的信息
 fmt.Println(getCallerInfo(1))
}