77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
package logger
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
logs "log"
|
|
"time"
|
|
|
|
"github.com/go-kit/log"
|
|
)
|
|
|
|
// 打印日志的API
|
|
type Logger interface {
|
|
Debug(string)
|
|
|
|
Info(string)
|
|
|
|
Warn(string)
|
|
|
|
Error(string)
|
|
}
|
|
|
|
var _ Logger = (*logger)(nil)
|
|
|
|
type logger struct {
|
|
kitLogger log.Logger
|
|
level Level
|
|
}
|
|
|
|
// 新建一个go-kit logger的接口封装
|
|
func New(out io.Writer, levelText string) (Logger, error) {
|
|
var level Level
|
|
err := level.UnmarshalText(levelText)
|
|
if err != nil {
|
|
return nil, fmt.Errorf(`{"level":"error","message":"%s: %s","ts":"%s"}`, err, levelText, time.RFC3339Nano)
|
|
}
|
|
l := log.NewJSONLogger(log.NewSyncWriter(out))
|
|
l = log.With(l, "ts", log.DefaultTimestampUTC)
|
|
return &logger{l, level}, err
|
|
}
|
|
|
|
func (l logger) Debug(msg string) {
|
|
if Debug.isAllowed(l.level) {
|
|
err := l.kitLogger.Log("level", Debug.String(), "message", msg)
|
|
if err != nil {
|
|
logs.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (l logger) Info(msg string) {
|
|
if Info.isAllowed(l.level) {
|
|
err := l.kitLogger.Log("level", Info.String(), "message", msg)
|
|
if err != nil {
|
|
logs.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (l logger) Warn(msg string) {
|
|
if Warn.isAllowed(l.level) {
|
|
err := l.kitLogger.Log("level", Warn.String(), "message", msg)
|
|
if err != nil {
|
|
logs.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (l logger) Error(msg string) {
|
|
if Error.isAllowed(l.level) {
|
|
err := l.kitLogger.Log("level", Error.String(), "message", msg)
|
|
if err != nil {
|
|
logs.Println(err)
|
|
}
|
|
}
|
|
}
|