logrus中输出文件名、行号及函数名

日志中输出文件名,行号及函数名是个比较有用的功能,那么在logrus中如何作到呢?
1. 在自带Formatter中输出 logrus有两个自带的Formatter,分别是:TextFormatter和JSONFormatter。要在这两个Formatter中输出文件名,行号和函数名,只需要设置

logrus.SetReportCaller(true)

1.1 在TextFormatter中输出
func Demo(){ logrus.Info("i'm demo")}func main(){ logrus.SetReportCaller(true) logrus.SetFormatter(&logrus.TextFormatter{ //以下设置只是为了使输出更美观 DisableColors:true, TimestampFormat:"2006-01-02 15:03:04", })Demo() }//输出 time="2021-05-11 14:02:45" level=info msg="i'm demo" func=main.Demo file="/home/ballqiu/go/log/demo.go:30"

说明:
  • func为函数名
  • file格式为文件名:行号
  • JsonFormatter中的使用类似
1.2 定制文件名和函数名的输出 上例中的文件名因为包含了路径信息,因此特别长。如果我们只要想文件名,不想输出路径,以便使得日志更简短,怎么做呢?可以设置Formatter中的CallerPrettyfier,它的函数原型是:
func(*runtime.Frame) (function string, file string)

【logrus中输出文件名、行号及函数名】返回值中的function是函数名, file是文件名。
例:
func Demo(){ logrus.Info("i'm demo")}func main(){ logrus.SetReportCaller(true)logrus.SetFormatter(&logrus.JSONFormatter{ TimestampFormat:"2006-01-02 15:03:04",CallerPrettyfier: func(frame *runtime.Frame) (function string, file string) { //处理文件名 fileName := path.Base(frame.File) return frame.Function, fileName }, })Demo() }//输出 {"file":"demo.go","func":"main.Demo","level":"info","msg":"i'm demo","time":"2021-05-11 14:02:57"}

说明:
  • 参数frame中包含了文件名,函数名,行号,等一系列信息。
  • path.Base(frame.File)去掉了文件名中的路径部分。
  • TextFormatter中的使用与JSONFormatter完全一致
2.在自定义Formatter中输出 如果你还不了解如何自定义Formatter,可以参看logrus自定义日志输出格式。
方法:
  1. 设置logrus.SetReportCaller(true)
  2. 通过entry.Caller拿到相应的信息进行输出
例:
type MyFormatter struct {} func (m *MyFormatter) Format(entry *logrus.Entry) ([]byte, error){ var b *bytes.Buffer if entry.Buffer != nil { b = entry.Buffer } else { b = &bytes.Buffer{} }timestamp := entry.Time.Format("2006-01-02 15:04:05") var newLog string//HasCaller()为true才会有调用信息 if entry.HasCaller() { fName := filepath.Base(entry.Caller.File) newLog = fmt.Sprintf("[%s] [%s] [%s:%d %s] %s\n", timestamp, entry.Level, fName, entry.Caller.Line, entry.Caller.Function, entry.Message) } else{ newLog = fmt.Sprintf("[%s] [%s] %s\n", timestamp, entry.Level, entry.Message) }b.WriteString(newLog) return b.Bytes(), nil }func Demo(){ logrus.Info("i'm demo") }func main(){ logrus.SetReportCaller(true)logrus.SetFormatter(&MyFormatter{})Demo() }//输出 [2021-05-11 15:08:46] [info] [demo.go:38 main.Demo] i'm demo

说明:
  • entry.Caller.File:文件名
  • entry.Caller.Line: 行号
  • entry.Caller.Function:函数名
  • entry.Caller中还有调用栈相关信息,有需要可以在日志中加入
  • entry.HasCaller() 的判断是必须的,否则如果外部没有设置logrus.SetReportCaller(true),entry.Caller.*的调用会引发Panic

    推荐阅读