82 lines
1.4 KiB
Go
82 lines
1.4 KiB
Go
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
|
||
|
|
kitlog "github.com/go-kit/kit/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type GoKitIOLogger struct {
|
||
|
|
fields Fields
|
||
|
|
level LogLevel
|
||
|
|
logger kitlog.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewGoKitIOLogger(out io.Writer) Logger {
|
||
|
|
kl := kitlog.NewJSONLogger(out)
|
||
|
|
|
||
|
|
return &GoKitIOLogger{
|
||
|
|
fields: make(Fields),
|
||
|
|
level: LevelInfo,
|
||
|
|
logger: kl,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) SetLogLevel(loglevel LogLevel) {
|
||
|
|
kl.level = loglevel
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) WithField(key string, value interface{}) Logger {
|
||
|
|
return kl.With(Fields{
|
||
|
|
key: value,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) WithErr(err error) Logger {
|
||
|
|
return kl.With(Fields{
|
||
|
|
"error": err,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) With(fields Fields) Logger {
|
||
|
|
newFields := make(Fields)
|
||
|
|
for k, v := range kl.fields {
|
||
|
|
newFields[k] = v
|
||
|
|
}
|
||
|
|
for k, v := range fields {
|
||
|
|
newFields[k] = v
|
||
|
|
}
|
||
|
|
|
||
|
|
return &GoKitIOLogger{
|
||
|
|
fields: newFields,
|
||
|
|
level: kl.level,
|
||
|
|
logger: kl.logger,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) Debug(message string) {
|
||
|
|
if kl.level == LevelDebug {
|
||
|
|
kl.log(message)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) Info(message string) {
|
||
|
|
if kl.level != LevelError {
|
||
|
|
kl.log(message)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) Error(message string) {
|
||
|
|
kl.log(message)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (kl *GoKitIOLogger) log(message string) {
|
||
|
|
kv := make([]interface{}, len(kl.fields)*2)
|
||
|
|
for k, v := range kl.fields {
|
||
|
|
kv = append(kv, k, v)
|
||
|
|
}
|
||
|
|
kv = append(kv, "level", kl.level, "message", message)
|
||
|
|
|
||
|
|
kl.logger.Log(kv...)
|
||
|
|
}
|