add buffered io to logs writer
This commit is contained in:
parent
66250f7122
commit
278ef8c991
25
src/logger/bufio_wrapper.go
Normal file
25
src/logger/bufio_wrapper.go
Normal file
@ -0,0 +1,25 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type BufioWrapper struct {
|
||||
*bufio.Writer
|
||||
m *sync.RWMutex
|
||||
}
|
||||
|
||||
func (b *BufioWrapper) Write(p []byte) (nn int, err error) {
|
||||
b.m.RLock()
|
||||
defer b.m.RUnlock()
|
||||
|
||||
return b.Writer.Write(p)
|
||||
}
|
||||
|
||||
func (b *BufioWrapper) Flush() error {
|
||||
b.m.Lock()
|
||||
defer b.m.Unlock()
|
||||
|
||||
return b.Writer.Flush()
|
||||
}
|
||||
@ -1,8 +1,12 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"context"
|
||||
"io"
|
||||
"os"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
@ -38,9 +42,24 @@ func New(opts NewLoggerOpts) (Logger, error) {
|
||||
level = zerolog.DebugLevel
|
||||
}
|
||||
|
||||
writer := io.MultiWriter(writers...)
|
||||
writer := bufio.NewWriterSize(io.MultiWriter(writers...), 32*1024)
|
||||
wrapper := &BufioWrapper{writer, &sync.RWMutex{}}
|
||||
go func() {
|
||||
tmr := time.NewTicker(500 * time.Millisecond)
|
||||
defer tmr.Stop()
|
||||
|
||||
l := zerolog.New(writer).Level(level).With().Timestamp().Logger()
|
||||
for {
|
||||
wrapper.Flush()
|
||||
|
||||
select {
|
||||
case <-context.Background().Done():
|
||||
return
|
||||
case <-tmr.C:
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
l := zerolog.New(wrapper).Level(level).With().Timestamp().Logger()
|
||||
return &logger{
|
||||
zeroLogger: &l,
|
||||
}, nil
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user