add buffered io to logs writer

This commit is contained in:
Sergey Chubaryan 2024-08-18 14:55:16 +03:00
parent 66250f7122
commit 278ef8c991
2 changed files with 46 additions and 2 deletions

View 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()
}

View File

@ -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