如何为 Go 设计一个通用的日志包

时间:2023-03-09 20:32:16
如何为 Go 设计一个通用的日志包

需求

一个通用的日志包,应该满足以下几个需求:

兼容 log.Logger,标准库大量使用了 log.Logger 作为其错误内容的输出通道,比如 net/http.Server.ErrorLog,所以兼容 log.Logger 是首要的需求;

自定义配置,不同的运行环境,往往需要不同的日志输出配置,在不重新编译源码的情况下,就能改变配置也应该成为一项标配;

不同的输出类型,根据目前市面上大部分的日志库,我们把日志分为以下六种类型:INFOWARNDEBUGTRACEERROR 和 CRITICAL

对日志内容的自定义处理,比如将日志按时间段进行切割,或是重要内容发邮箱进行提醒等。

配置文件

根据以上的需求,我们可以拟定一个以下格式的 XML 配置文件:

<?xml version="1.0" encoding="utf-8"?>
<logs>
<info prefix="INFO" flag="log.lstdflag" >
<!-- 向控制台输出信息 -->
<console output="stderr" foreground="red" background="blue" />
</info> <error prefix="ERROR" flag="log.lstdflag" />
<!-- 向控制台输出信息 -->
<console output="stderr" foreground="red" background="blue" /> <!-- 输出到 /var/logs/debug 目录下,如果大于 5M,则新建文件 -->
<rotate dir="/var/logs/debug/" size="5M" />
</error> <critical prefix="CRITICAL" flag="log.lstdflag" >
<!-- 向指定的邮箱发送错误信息 -->
<smtp username="u"
password="p"
host="smtp.example.com"
subject="CRITICAL"
sendTo="u1@example.com;u2@example.com" />
</critical>
</logs>

*元素固定为 logs

二级元素为 INFO 和 WARN 等六种固定的类型,在加载时,将其初始为一个 *log.Logger 实例,初始化参数可以直接从 XML 的属性和子元素中获取;

二级以下的元素,统一初始化为 io.Writer 实例,当作 log.New() 的参数,用于初始化其父元素。用户可通过接口自定义这些元素,若有初始化参数,则从其 XML 属性值中获取。

这样就达到了一个*度相对比较高的配置文件。

接口设计

根据以上要求,我们大致上可以确定下来接口的内容。

一个从配置文件进行初始化的接口:

func InitFromConfig(path string) error

以及各个输出类型的操作函数,最主要的就是获取 *log.Logger 的接口,以及一些常用的简化调用。以 INFO 类型为例,我们可以定义以下几个函数,其它类型也都相同,只是函数名不同。

func INFO() *log.Logger  // 返回与 INFO 类型对应的 *log.Logger 实例

// 针对 INFO 类型日志的一些常用操作进行封装
func Info(v ...interface{}) // 相当于 INFO().Println()
func Infof(format string, v...interface{}) // 相当于 INFO().Printf()

我们还需要公开一个注册接口,用于注册用户自定义的元素:

type Init func(args map[string]string)(io.Writer, error)

func Register(name string, fn Init)

使用

只要在 main() 中正确定加载了配置文件,之后随处都可以调用:

// main.go
func main() {
err := logs.InitFromConfig("./config/logs.xml")
if err !=nil {
panic(err)
} // do something
} // file1.go
func h1(w http.ResponseWriter, r *http.Request) {
if !auth() {
logs.Infof("鉴权失败")
w.WriteHeader(http.StatusUnauthorized)
return
}
} // file2.go
func getServer(port string, h http.Handler) *http.Server {
return &http.Server {
Addr: port,
Handler: h,
ErrorLog: logs.INFO(),
}
}

实现

完整的实现代码在:https://github.com/issue9/logs