golang中log包自定义输出日志格式与写入到文件

2024-02-07 08:30:41

自定义 Golang 中的 Log 日志格式及输出到文件

在 Golang 开发中，日志记录是监控和调试应用程序的关键。log 包 提供了丰富的功能，允许您轻松地记录和管理日志消息。除了预定义的格式化程序，您还可以自定义输出格式和将日志写入文件，以满足特定需求。

自定义输出日志格式

默认情况下，log 包使用 TextFormatter 将日志记录为纯文本，包括时间戳、日志级别和消息。然而，您可以通过实现 io.Writer 接口来自定义自己的格式化程序：

import (
	"fmt"
	"io"
	"log"
	"os"
	"strings"
)

type MyLogFormatter struct {
	level  string
	prefix string
}

func (f *MyLogFormatter) Write(p []byte) (n int, err error) {
	level, prefix := parseLogRecord(p)
	f.level = level
	f.prefix = prefix
	return fmt.Fprintf(os.Stdout, "[%s] [%s] %s", f.level, f.prefix, p)
}

func parseLogRecord(p []byte) (level string, prefix string) {
	level = strings.TrimSpace(string(p[:strings.Index(string(p), "]")]))
	prefix = strings.TrimSpace(string(p[strings.Index(string(p), "]")+1:strings.Index(string(p), " ")])
	return level, prefix
}

func main() {
	formatter := &MyLogFormatter{}
	logger := log.New(formatter, "", log.Ldate|log.Ltime|log.Lshortfile)
	logger.Println("自定义日志消息")
}

在这个例子中，MyLogFormatter 结构体实现了 Write() 方法，它解析日志记录并将其格式化为自定义的格式，包括日志级别和前缀。

将日志输出到文件

除了标准输出之外，您还可以在 Golang 中将日志写入文件：

import (
	"io"
	"log"
	"os"
)

func main() {
	file, err := os.OpenFile("my_log.txt", os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
	if err != nil {
		panic(err)
	}
	log.SetOutput(file)
	log.Println("日志写入文件")
}