linka-cloud/grpc
2
0
Fork 0
mirror of https://github.com/linka-cloud/grpc.git synced 2024-11-24 03:46:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

logger: add Logger method to access *logrus.Logger

Browse Source 
Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
This commit is contained in:
Adphi 2022-10-18 18:26:39 +02:00
parent c7096975b1
commit 23f1b78389
Signed by: adphi
GPG Key ID: 46BE4062DB2397FF
1 changed files with 12 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
logger/logger.go
View File

@ -64,6 +64,7 @@ type Logger interface {
Logr() logr.Logger Logr() logr.Logger
FieldLogger() logrus.FieldLogger FieldLogger() logrus.FieldLogger
Logger() *logrus.Logger
} }
type logger struct { type logger struct {
@ -167,25 +168,12 @@ func (l *logger) Panicln(args ...interface{}) {
} }
func (l *logger) WriterLevel(level logrus.Level) *io.PipeWriter { func (l *logger) WriterLevel(level logrus.Level) *io.PipeWriter {
switch t := l.fl.(type) { return l.Logger().WriterLevel(level)
case *logrus.Logger:
return t.WriterLevel(level)
case *logrus.Entry:
return t.WriterLevel(level)
}
panic(fmt.Sprintf("unexpected logger type %T", l.fl))
} }
func (l *logger) SetLevel(level logrus.Level) Logger { func (l *logger) SetLevel(level logrus.Level) Logger {
switch t := l.fl.(type) { l.Logger().SetLevel(level)
case *logrus.Logger:
t.SetLevel(level)
return l return l
case *logrus.Entry:
t.Logger.SetLevel(level)
return l
}
panic(fmt.Sprintf("unexpected logger type %T", l.fl))
} }
func (l *logger) WithField(key string, value interface{}) Logger { func (l *logger) WithField(key string, value interface{}) Logger {
@ -212,14 +200,17 @@ func (l *logger) FieldLogger() logrus.FieldLogger {
return l.fl return l.fl
} }
func (l *logger) SetOutput(w io.Writer) Logger { func (l *logger) Logger() *logrus.Logger {
switch t := l.fl.(type) { switch t := l.fl.(type) {
case *logrus.Logger: case *logrus.Logger:
t.SetOutput(w) return t
return l
case *logrus.Entry: case *logrus.Entry:
t.Logger.SetOutput(w) return t.Logger
return l
} }
panic(fmt.Sprintf("unexpected logger type %T", l.fl)) panic(fmt.Sprintf("unexpected logger type %T", l.fl))
} }
func (l *logger) SetOutput(w io.Writer) Logger {
l.Logger().SetOutput(w)
return l
}