Golang 日志庫 Zap 如何自定義輸出目標(biāo)？
作者：路多辛
Zap是一個功能非常豐富、易于使用的 Golang日志庫。記錄日志時，一般會選擇輸出到控制臺（Stdout和Stderr）或者輸出到文件或者同時輸出到兩者。zap 庫也提供了對應(yīng)的選項。
Zap（https://github.com/uber-go/zap）是一個功能非常豐富、易于使用的 Golang日志庫。記錄日志時，一般會選擇輸出到控制臺（stdout和stderr）或者輸出到文件或者同時輸出到兩者。zap 庫也提供了對應(yīng)的選項。
指定日志輸出目標(biāo)
zap 配置項中（zap.Config）, 使用 OutputPaths 和 ErrorOutputPaths 可以指定日志輸出目標(biāo)，首先看下這個配置項的定義和說明：
// OutputPaths is a list of URLs or file paths to write logging output to.
// See Open for details.
OutputPaths []string `json:"outputPaths" yaml:"outputPaths"`
// ErrorOutputPaths is a list of URLs to write internal logger errors to.
// The default is standard error.
//
// Note that this setting only affects internal errors; for sample code that
// sends error-level logs to a different location from info- and debug-level
// logs, see the package-level AdvancedConfiguration example.
ErrorOutputPaths []string `json:"errorOutputPaths" yaml:"errorOutputPaths"`OutputPaths：是一個字符串類型的切片，可以指定URL、文件路徑或者 stdout，默認(rèn)是stdout。
ErrorOutputPaths：也是一個字符串類型的切片，和 OutputPaths 類似，默認(rèn)是stderr。
看個例子
package main
import (
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
func main() {
cfg := zap.NewProductionConfig()
cfg.OutputPaths = []string{"stdout"}
cfg.ErrorOutputPaths = []string{"stderr"}
cfg.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
logger, err := cfg.Build()
if err != nil {
panic("init logger error")
}
sugar := Logger.Sugar()
sugar.Info("這是一條測試日志")
}這個例子是將日志輸出到控制臺，如果想也把日志輸出到文件一份，修改下第 10 和 11 行，代碼如下：
cfg.OutputPaths = []string{"stdout", "./test.log"}
cfg.ErrorOutputPaths = []string{"stderr", "./test.log"}如果只想輸出到文件，代碼如下：
cfg.OutputPaths = []string{"./test.log"}
cfg.ErrorOutputPaths = []string{"./test.log"}自定義日志輸出方式
可以使用 zap.RegisterSink 函數(shù)來注冊自定義的的輸出方式，該函數(shù)第二個參數(shù)為一個自定義的工廠函數(shù), 接收一個 *url.URL 類型的指針, 返回一個 Sink 類型和一個 error，所以需要實現(xiàn) Sink 接口?？匆粋€具體的示例，自定義了一個通過 http 協(xié)議輸出到指定接口的輸出方式：
package main
import (
"bytes"
"errors"
"fmt"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"net/http"
"net/url"
)
type Http struct {
Remote *url.URL
}
func (h Http) Sync() error {
return nil
}
func (h Http) Close() error {
return nil
}
func (h Http) Write(p []byte) (n int, err error) {
req, err := http.NewRequest("POST", h.Remote.String(), bytes.NewBuffer(p))
if err != nil {
return 0, nil
}
req.Header.Set("Content-Type", "application/json")
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return 0, nil
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
return -1, errors.New("request failed")
}
return len(p), nil
}
// 實現(xiàn)工廠函數(shù)
func httpSink(url *url.URL) (sink zap.Sink, err error) {
h := Http{Remote: url}
return h, nil
}
func main() {
// 注冊自定義方法
if err := zap.RegisterSink("Http", httpSink); err != nil {
fmt.Println(err)
return
}
cfg := zap.NewProductionConfig()
cfg.OutputPaths = []string{"stdout", "http://127.0.0.1/log"}
cfg.ErrorOutputPaths = []string{"stderr", "http://127.0.0.1/log"}
cfg.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
logger, err := cfg.Build()
if err != nil {
panic(err)
}
logger.Sugar().Info("這是一條測試日志")
}然后啟動一個服務(wù)端，代碼如下：
package main
import (
"bytes"
"io/ioutil"
"log"
"net/http"
)
func main() {
http.HandleFunc("/log", func(w http.ResponseWriter, r *http.Request) {
body, _ := ioutil.ReadAll(r.Body)
log.Println("日志內(nèi)容內(nèi)容: ", bytes.NewBuffer(body).String())
})
log.Fatal(http.ListenAndServe(":80", nil))
}然后運行打印日志的代碼，可以看到服務(wù)端接收到了日志。
責(zé)任編輯：姜華
來源：
今日頭條
