2021-02-17 17:14:47 +08:00
|
|
|
package log
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"fmt"
|
2021-02-22 17:50:43 +08:00
|
|
|
"io"
|
2021-02-17 17:14:47 +08:00
|
|
|
"log"
|
|
|
|
"sync"
|
|
|
|
)
|
|
|
|
|
|
|
|
var _ Logger = (*stdLogger)(nil)
|
|
|
|
|
|
|
|
type stdLogger struct {
|
|
|
|
log *log.Logger
|
|
|
|
pool *sync.Pool
|
|
|
|
}
|
|
|
|
|
2021-05-17 16:34:28 +08:00
|
|
|
// NewStdLogger new a logger with writer.
|
2021-02-22 17:50:43 +08:00
|
|
|
func NewStdLogger(w io.Writer) Logger {
|
2021-02-17 17:14:47 +08:00
|
|
|
return &stdLogger{
|
2021-05-01 01:15:30 +08:00
|
|
|
log: log.New(w, "", 0),
|
2021-02-17 17:14:47 +08:00
|
|
|
pool: &sync.Pool{
|
|
|
|
New: func() interface{} {
|
|
|
|
return new(bytes.Buffer)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-05-01 01:15:30 +08:00
|
|
|
// Log print the kv pairs log.
|
2021-05-17 16:34:28 +08:00
|
|
|
func (l *stdLogger) Log(level Level, keyvals ...interface{}) error {
|
|
|
|
if len(keyvals) == 0 {
|
2021-05-01 01:15:30 +08:00
|
|
|
return nil
|
2021-02-17 17:14:47 +08:00
|
|
|
}
|
2021-07-18 15:01:08 +08:00
|
|
|
if (len(keyvals) & 1) == 1 {
|
2021-06-14 21:00:18 +08:00
|
|
|
keyvals = append(keyvals, "KEYVALS UNPAIRED")
|
2021-02-17 17:14:47 +08:00
|
|
|
}
|
2021-03-14 15:36:00 +08:00
|
|
|
buf := l.pool.Get().(*bytes.Buffer)
|
2021-05-17 16:34:28 +08:00
|
|
|
buf.WriteString(level.String())
|
|
|
|
for i := 0; i < len(keyvals); i += 2 {
|
2021-08-16 10:35:08 +08:00
|
|
|
_, _ = fmt.Fprintf(buf, " %s=%v", keyvals[i], keyvals[i+1])
|
2021-02-17 17:14:47 +08:00
|
|
|
}
|
2021-08-31 10:14:57 +08:00
|
|
|
_ = l.log.Output(4, buf.String()) //nolint:gomnd
|
2021-02-17 17:14:47 +08:00
|
|
|
buf.Reset()
|
2021-03-14 15:36:00 +08:00
|
|
|
l.pool.Put(buf)
|
2021-05-01 01:15:30 +08:00
|
|
|
return nil
|
2021-02-17 17:14:47 +08:00
|
|
|
}
|