mirror of
https://github.com/ScuroNeko/mtg.git
synced 2026-09-01 08:24:03 +03:00
Add zerolog implementation
This commit is contained in:
@@ -0,0 +1,115 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"github.com/9seconds/mtg/v2/mtglib"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
const loggerFieldName = "logger"
|
||||
|
||||
type zeroLogContextVarType uint8
|
||||
|
||||
const (
|
||||
zeroLogContextVarTypeUnknown zeroLogContextVarType = iota
|
||||
zeroLogContextVarTypeStr
|
||||
zeroLogContextVarTypeInt
|
||||
)
|
||||
|
||||
type zeroLogContext struct {
|
||||
name string
|
||||
log *zerolog.Logger
|
||||
|
||||
ctxVarType zeroLogContextVarType
|
||||
ctxVarName string
|
||||
ctxVarStr string
|
||||
ctxVarInt int
|
||||
|
||||
parent *zeroLogContext
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) Named(name string) mtglib.Logger {
|
||||
loggerName := z.name
|
||||
if loggerName == "" {
|
||||
loggerName = name
|
||||
} else {
|
||||
loggerName += "." + name
|
||||
}
|
||||
|
||||
return &zeroLogContext{
|
||||
name: loggerName,
|
||||
log: z.log,
|
||||
parent: z,
|
||||
}
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) BindInt(name string, value int) mtglib.Logger {
|
||||
return &zeroLogContext{
|
||||
name: z.name,
|
||||
log: z.log,
|
||||
ctxVarType: zeroLogContextVarTypeInt,
|
||||
ctxVarInt: value,
|
||||
ctxVarName: name,
|
||||
parent: z,
|
||||
}
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) BindStr(name, value string) mtglib.Logger {
|
||||
return &zeroLogContext{
|
||||
name: z.name,
|
||||
log: z.log,
|
||||
ctxVarType: zeroLogContextVarTypeStr,
|
||||
ctxVarStr: value,
|
||||
ctxVarName: name,
|
||||
parent: z,
|
||||
}
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) Info(msg string) {
|
||||
z.InfoError(msg, nil)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) Warning(msg string) {
|
||||
z.WarningError(msg, nil)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) Debug(msg string) {
|
||||
z.DebugError(msg, nil)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) InfoError(msg string, err error) {
|
||||
z.emitLog(z.log.Info(), msg, err)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) WarningError(msg string, err error) {
|
||||
z.emitLog(z.log.Warn(), msg, err)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) DebugError(msg string, err error) {
|
||||
z.emitLog(z.log.Debug(), msg, err)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) emitLog(evt *zerolog.Event, msg string, err error) {
|
||||
z.attachCtx(evt)
|
||||
|
||||
for current := z.parent; current != nil; current = current.parent {
|
||||
current.attachCtx(evt)
|
||||
}
|
||||
|
||||
evt.Str(loggerFieldName, z.name).Err(err).Msg(msg)
|
||||
}
|
||||
|
||||
func (z *zeroLogContext) attachCtx(evt *zerolog.Event) {
|
||||
switch z.ctxVarType {
|
||||
case zeroLogContextVarTypeStr:
|
||||
evt.Str(z.ctxVarName, z.ctxVarStr)
|
||||
case zeroLogContextVarTypeInt:
|
||||
evt.Int(z.ctxVarName, z.ctxVarInt)
|
||||
case zeroLogContextVarTypeUnknown:
|
||||
}
|
||||
}
|
||||
|
||||
func NewZeroLogger(log zerolog.Logger) mtglib.Logger {
|
||||
return &zeroLogContext{
|
||||
log: &log,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user