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
|
package logger
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
|
"context"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/rs/zerolog"
|
"github.com/rs/zerolog"
|
||||||
)
|
)
|
||||||
@ -38,9 +42,24 @@ func New(opts NewLoggerOpts) (Logger, error) {
|
|||||||
level = zerolog.DebugLevel
|
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{
|
return &logger{
|
||||||
zeroLogger: &l,
|
zeroLogger: &l,
|
||||||
}, nil
|
}, nil
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user