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

日志中输出文件名,行号及函数名是个比较有用的功能,那么在logrus中如何作到呢?

1. 在自带Formatter中输出

logrus有两个片自的Formatter,分别是:TextFormatter和JSONFormatter。(如果不了解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)

返回值中的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

你可能感兴趣的:(go)