Skip to content

Commit

Permalink
feat: Add logger.NewWriter (#42)
Browse files Browse the repository at this point in the history
  • Loading branch information
jiping-s authored Jan 25, 2024
1 parent 8ee7136 commit 4b74920
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 0 deletions.
5 changes: 5 additions & 0 deletions logger/logger.go
Original file line number Diff line number Diff line change
Expand Up @@ -492,6 +492,11 @@ func (logger Logger) WithFields(fields map[string]interface{}) Logger {
return wrapLogger(entry, logger)
}

// NewWriter creates an io.Writer on this logger. Each .Write() call would log a message.
func (logger Logger) NewWriter(level LogLevel) io.Writer {
return newLogWriter(logger, level)
}

func buildSprintPrefixes(fields map[string]interface{}) []string {
prefixList := make([]string, 0, 3)

Expand Down
31 changes: 31 additions & 0 deletions logger/loggerwriter.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
package logger

type logWriter struct {
lg Logger
level LogLevel
}

func newLogWriter(lg Logger, level LogLevel) *logWriter {
return &logWriter{lg, level}
}

func (lw *logWriter) Write(p []byte) (n int, err error) {
s := string(p)
switch lw.level {
case PanicLevel:
lw.lg.Panic(s)
case FatalLevel:
lw.lg.Fatal(s)
case ErrorLevel:
lw.lg.Error(s)
case WarnLevel:
lw.lg.Warn(s)
case InfoLevel:
lw.lg.Info(s)
case DebugLevel:
lw.lg.Debug(s)
case TraceLevel:
lw.lg.Trace(s)
}
return len(s), nil
}

0 comments on commit 4b74920

Please sign in to comment.