Go语言Zap库Logger的定制化和封装使用详解

作者:kevinyan 时间:2024-04-25 15:18:52 

前言

日志无论对于程序还是程序员都非常重要,有多重要呢,想要长期在公司健健康康的干下去就得学会阶段性划水,阶段性划水的一大关键的就是干活快过预期但是装作。。。不对,这个开头不对劲,下面重来。

日志无论对于程序还是程序员都非常重要,程序员解决问题的快慢除了经验外,就是看日志能不能有效地记录问题发生的现场以及上下文等等。

那么让让程序记录有效的日志,除了程序内记日志的点位尽量精准外,还需要有一个称手的 Logger 。一个好的 Logger (日志记录器) 要能提供以下这些能力:

  • 支持把日志写入到多个输出流中,比如可以选择性的让测试、开发环境同时向控制台和日志文件输出日志,生产环境只输出到文件中。

  • 支持多级别的日志等级,比如常见的有:TRACEDEBUGINFOWARNERROR 等。

  • 支持结构化输出,结构化输出现在常用的就是JSON形式的,这样可以让统一日志平台,通过 logstash 之类的组件直接把日志聚合到日志平台上去。

  • 需要支持日志切割 -- log rotation, 按照日期、时间间隔或者文件大小对日志进行切割。

  • 在 Log Entry 中(就是每行记录)除了主动记录的信息外,还要包括如打印日志的函数、所在的文件、行号、记录时间等。

今天我带大家一起看看怎么在使用 Go 语言开发的项目里打造一个称手的 Logger,在这之前让我们先回到 2009 年,看看 Go 语言自诞生之初就提供给我们的内置 Logger。

Go 语言原生的Logger

Go 语言自带 log 内置包,为我们提供了一个默认的 Logger,可以直接使用。 这个库的详细用法可以在官方的文档里找到:pkg.go.dev/log

使用 log 记录日志,默认会输出到控制台中。比如下面这个例子:

package main
import (
"log"
"net/http"
)
func main() {
simpleHttpGet("www.google.com")
simpleHttpGet("https://www.baidu.com")
}
func simpleHttpGet(url string) {
resp, err := http.Get(url)
if err != nil {
log.Printf("Error fetching url %s : %s", url, err.Error())
} else {
log.Printf("Status Code for %s : %s", url, resp.Status)
resp.Body.Close()
}
return
}

这个例程中,分别向两个网址进行 GET 请求,然后记录了一下返回状态码 / 请求错误。 执行程序后会有类似输出:

2022/05/15 15:15:26 Error fetching url www.baidu.com : Get "www.baidu.com": unsupported protocol scheme "" 2022/05/15 15:15:26 Status Code for https://www.baidu.com : 200 OK

因为第一次请求的 URL 中协议头缺失, 所以不能成功发起请求,日志也很好的记录了错误信息。

Go 内置的 log 包当然也支持把日志输出到文件中,通过log.SetOutput 可以把任何 io.Writer 的实现设置成日志的输出。下面我们把上面那个例程修改成向文件输出日志。

大家可以自己试一下运行效果,这里不再做过多演示。

Go 语言原生Logger的缺点

原生 Logger 的优点,显而易见,简单、开箱即用,不用引用外部的三方库。我们可以按照开头处提出的对于一个 Logger 的五个标准再看一下默认Logger 是否能在项目里使用。

  • 仅限基本的日志级别

    • 只有一个Print选项。不支持INFO/DEBUG等多个级别。

  • 对于错误日志,它有FatalPanic

    • Fatal日志通过调用os.Exit(1)来结束程序

    • Panic日志在写入日志消息之后抛出一个panic

    • 但是它缺少一个ERROR日志级别,这个级别可以在不抛出panic或退出程序的情况下记录错误

  • 缺乏结构化日志格式的能力——只支持简单文本输出,不能把日志记录格式化成 JSON 格式。

  • 不提供日志切割的能力。

Zap 日志库

在 Go 的生态中,有不少可以选择的日志库,之前我们简单介绍过 logrus 这个库的使用:点我查看,它与Go的内置 log 库在 api 层面兼容,直接实现了log.Logger接口,支持把程序的系统级 Logger 切换成它。

不过 logrus 在性能敏感的场景下就显得不香了,用的更多的是 Uber 开源的 zap 日志库。由于 Uber 在当今 Go 生态中的贡献度很高,加之它本身业务—网约车的性能敏感场景,所以 Uber 开源的库很受欢迎。现在做项目,使用 Zap 做日志Logger 的非常多。程序员的内心OS应该是,不管我这并发高不高,上就完事了,万一哪天能从2个并发突然干成 2W 并发呢。

Zap 性能高的一大原因是:不用反射,日志里每个要写入的字段都得携带着类型

logger.Info(
 "Success..",
 zap.String("statusCode", resp.Status),
 zap.String("url", url))

上面向日志里写入了一条记录,Message 是 "Success.." 另外写入了两个字符串键值对。 Zap 针对日志里要写入的字段,每个类型都有一个对应的方法把字段转成 zap.Field 类型 。比如:

zap.Int('key', 123)
zap.Bool('key', true)
zap.Error('err', err)
zap.Any('arbitraryType', &User{})

还有很多中这种类型方法,就不一一列举啦。这种记录日志的方式造成在使用体验上稍稍有点差,不过考虑到性能上收益这点使用体验上的损失也能接受。

下面我们先来学习一下 Zap 的使用方法,再对项目中使用 Zap 时做些自定义的配置和封装,让它变得更好用,最重要的是匹配上我们开头提出的关于好的 Logger 的五条标准。

Zap 的使用方法

安装zap

首先说一下,zap 的安装方式,直接运行以下命令下载 zap 到本地的依赖库中。

go get -u go.uber.org/zap

设置 Logger

我们先说 zap 提供的配置好的 Logger ,稍后会对它进行自定义。

  • 通过调用zap.NewProduction()zap.NewDevelopment()zap.Example() 这三个方法,都可以创建 Logger。

  • 上面三个方法都可以创建 Logger,他们都对 Logger 进行了不同的配置,比如zap.NewProduction()创建的 Logger 在记录日志时会自动记录调用函数的信息、打日志的时间等,这三个不用纠结,直接都用zap.NewProduction(),且在项目中使用的时候,我们不会直接用 zap 配置好的 Logger ,需要再做更细致的定制。

zap 的 Logger 提供了记录不同等级的日志的方法,像从低到高的日志等级一般有:Debug、Info、Warn、Error 这些级别都有对应的方法。他们的使用方式都一样,下面是 Info 方法的方法签名。

func (log *Logger) Info(msg string, fields ...Field) {
if ce := log.check(InfoLevel, msg); ce != nil {
ce.Write(fields...)
}
}

方法的第一个参数是日志里 msg 字段要记录的信息,msg是日志行记录里一个固定的字段,要再添加其他字段到日志,直接传递 zap.Field 类型的参数即可,上面我们已经说过zap.Field类型的字段,就是由 zap.String("key", "value") 这类方法创建出来的。由于 Info 方法签名里 fileds参数声明是可变参数,所以支持添加任意多个字段到日志行记录里, 比如例程里的:

logger.Info("Success..", zap.String("statusCode", resp.Status), zap.String("url", url))

即日志行记录里,除了 msg 字段,还添加了statusCodeurl两个自定义字段。 上面例程里使用的zap.NewProduction()创建的 Logger 会向控制台输出JSON格式的日志行,比如上面使用Info方法后,控制台会有类似下面的输出。

{"level":"info","ts":1558882294.665447,"caller":"basiclogger/UberGoLogger.go:31","msg":"Success..","statusCode":"200 OK","url":"https://www.baidu.com"}

定制 Zap 的 Logger

下面我们把 zap 做进一步的自定义配置,让日志不光能输出到控制台,也能输出到文件,再把日志时间由时间戳格式,换成更容易被人类看懂的DateTime时间格式。

下面少说话,直接上代码,必要的解释放在了注释里。

var logger *zap.Logger
func init() {
encoderConfig := zap.NewProductionEncoderConfig()
 // 设置日志记录中时间的格式
encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
 // 日志Encoder 还是JSONEncoder,把日志行格式化成JSON格式的
encoder := zapcore.NewJSONEncoder(encoderConfig)
file, _ := os.OpenFile("/tmp/test.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 644)
fileWriteSyncer = zapcore.AddSync(file)
core := zapcore.NewTee(
// 同时向控制台和文件写日志, 生产环境记得把控制台写入去掉,日志记录的基本是Debug 及以上,生产环境记得改成Info
zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), zapcore.DebugLevel),
zapcore.NewCore(encoder, fileWriteSyncer, zapcore.DebugLevel),
)
logger = zap.New(core)
}

日志切割

Zap 本身不支持日志切割,可以借助另外一个库 lumberjack 协助完成切割。

func getFileLogWriter() (writeSyncer zapcore.WriteSyncer) {
// 使用 lumberjack 实现 log rotate
lumberJackLogger := &lumberjack.Logger{
Filename:   "/tmp/test.log",
MaxSize:    100, // 单个文件最大100M
MaxBackups: 60, // 多于 60 个日志文件后,清理较旧的日志
MaxAge:     1, // 一天一切割
Compress:   false,
}
return zapcore.AddSync(lumberJackLogger)
}

封装 Logger

我们不能每次使用日志,都这么设置一番,所以最好的还是把这些配置初始化放在一个单独的包里,这样在项目中初始化一次即可。

除了上面的那些配置外,我们的配置里还少了些日志调用方的信息,比如函数名、文件位置、行号等,这样在排查问题看日志的时候,定位问题的时效会提高不少。

我们对 Logger 再做一下封装。

// 发送私信 go-logger 给公众号「网管叨bi叨」
// 可获得完整代码和使用Demo
package zlog
// 简单封装一下对 zap 日志库的使用
// 使用方式:
// zlog.Debug("hello", zap.String("name", "Kevin"), zap.Any("arbitraryObj", dummyObject))
// zlog.Info("hello", zap.String("name", "Kevin"), zap.Any("arbitraryObj", dummyObject))
// zlog.Warn("hello", zap.String("name", "Kevin"), zap.Any("arbitraryObj", dummyObject))
var logger *zap.Logger
func init() {
......
}
func getFileLogWriter() (writeSyncer zapcore.WriteSyncer) {
......
}
func Info(message string, fields ...zap.Field) {
callerFields := getCallerInfoForLog()
fields = append(fields, callerFields...)
logger.Info(message, fields...)
}
func Debug(message string, fields ...zap.Field) {
callerFields := getCallerInfoForLog()
fields = append(fields, callerFields...)
logger.Debug(message, fields...)
}
func Error(message string, fields ...zap.Field) {
callerFields := getCallerInfoForLog()
fields = append(fields, callerFields...)
logger.Error(message, fields...)
}
func Warn(message string, fields ...zap.Field) {
callerFields := getCallerInfoForLog()
fields = append(fields, callerFields...)
logger.Warn(message, fields...)
}
func getCallerInfoForLog() (callerFields []zap.Field) {
pc, file, line, ok := runtime.Caller(2) // 回溯两层,拿到写日志的调用方的函数信息
if !ok {
return
}
funcName := runtime.FuncForPC(pc).Name()
funcName = path.Base(funcName) //Base函数返回路径的最后一个元素,只保留函数名
callerFields = append(callerFields, zap.String("func", funcName), zap.String("file", file), zap.Int("line", line))
return
}

为啥不用 zap.New(core, zap.AddCaller())这种方式,在日志行里添加调用方的信息呢?主要还是想更灵活点,能自己制定对应的日志字段,所以把 Caller的几个信息放到单独的字段里,等把日志收集到日志平台上去后,查询日志的时候也更利于检索。

在下面的例程中尝试使用我们封装好的日志 Logger 做个简单的测试。

package main
import (
"example.com/utils/zlog"
)
type User strunct {
 Name  stirng
}
func main() {
 user := &User{
   "Name": "Kevin"
 }
 zlog.Info("test log", zap.Any("user", user))
}

输出类似下面的输出。

{"level":"info","ts":"2022-05-15T21:22:22.687+0800","msg":"test log","res":{"Name":"Kevin"},"func":"main.Main","file":"/Users/Kevin/go/src/example.com/demo/zap.go","line":84}

来源:https://juejin.cn/post/7110892495827009572

标签:Go语言,Zap,Logger,定制化,封装
0
投稿

猜你喜欢

  • Python实现文件按照日期命名的方法

    2022-10-25 19:40:09
  • Vue+Java+Base64实现条码解析的示例

    2024-05-02 17:08:27
  • WEB设计经验-来自Microsoft

    2008-05-15 07:30:00
  • 在matlab中创建类似字典的数据结构方式

    2021-07-08 01:22:53
  • 简单的Python2.7编程初学经验总结

    2021-03-18 01:27:11
  • python批量下载壁纸的实现代码

    2023-11-09 18:13:05
  • Python使用scipy.fft进行大学经典的傅立叶变换

    2022-09-10 20:26:13
  • Numpy np.array()函数使用方法指南

    2023-05-23 09:33:37
  • JavaScript中Infinity(无穷数)的使用和注意事项

    2023-08-21 00:59:13
  • 详解vue-cli 脚手架 安装

    2024-05-09 10:40:06
  • Python 网络编程之TCP客户端/服务端功能示例【基于socket套接字】

    2023-12-12 07:25:21
  • Python3基础教程之递归函数简单示例

    2022-07-19 17:19:16
  • python中numpy数组与list相互转换实例方法

    2023-11-29 01:48:50
  • Mysql性能优化案例 - 覆盖索引分享

    2024-01-27 15:16:59
  • 基于Python编写一个宝石消消乐小游戏

    2021-10-25 05:46:06
  • Python实现为PDF去除水印的示例代码

    2023-03-05 04:27:56
  • getAllResponseHeaders获取网页的http头信息代码

    2010-03-31 14:31:00
  • vue计时器的实现方法

    2024-04-30 10:24:27
  • 使用python判断你是青少年还是老年人

    2021-03-18 14:42:23
  • Python 代码性能优化技巧分享

    2023-10-15 05:00:56
  • asp之家 网络编程 m.aspxhome.com