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

需求

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

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

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

不同的输出类型，根据目前市面上大部分的日志库，我们把日志分为以下六种类型：INFO、WARN、DEBUG、TRACE、ERROR 和 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。

原文地址：https://caixw.io/posts/2017/how-to-design-log-package-with-go.html