mirror of
https://github.com/kemko/reproxy.git
synced 2026-01-04 01:05:50 +03:00
42 lines
988 B
Go
42 lines
988 B
Go
package lgr
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
)
|
|
|
|
// Writer holds lgr.L and wraps with io.Writer interface
|
|
type Writer struct {
|
|
L
|
|
level string // if defined added to each message
|
|
}
|
|
|
|
// Write to lgr.L
|
|
func (w *Writer) Write(p []byte) (n int, err error) {
|
|
w.Logf(w.level + string(p))
|
|
return len(p), nil
|
|
}
|
|
|
|
// ToWriter makes io.Writer for given lgr.L with optional level
|
|
func ToWriter(l L, level string) *Writer {
|
|
if level != "" && !strings.HasSuffix(level, " ") {
|
|
level += " "
|
|
}
|
|
return &Writer{l, level}
|
|
}
|
|
|
|
// ToStdLogger makes standard logger
|
|
func ToStdLogger(l L, level string) *log.Logger {
|
|
return log.New(ToWriter(l, level), "", 0)
|
|
}
|
|
|
|
// SetupStdLogger makes the default std logger with lgr.L
|
|
func SetupStdLogger(opts ...Option) {
|
|
logOpts := append([]Option{CallerDepth(3)}, opts...) // skip 3 more frames to compensate stdlog calls
|
|
l := New(logOpts...)
|
|
l.reTrace = reTraceStd // std logger split on log/ path
|
|
log.SetOutput(ToWriter(l, ""))
|
|
log.SetPrefix("")
|
|
log.SetFlags(0)
|
|
}
|