49 lines
983 B
Go
49 lines
983 B
Go
package rotater
|
|
|
|
import (
|
|
"github.com/lestrrat-go/file-rotatelogs"
|
|
"sync"
|
|
)
|
|
|
|
type rotateSyncer struct {
|
|
setOnceOnclose sync.Once
|
|
onceOnclose sync.Once
|
|
onClose func()
|
|
*rotatelogs.RotateLogs
|
|
}
|
|
|
|
// NewRotater create a RotateSyncer writer.
|
|
func NewRotater(filename string, options ...Option) (RotateSyncer, error) {
|
|
if rotateLogger, err := rotatelogs.New(filename, options...); err != nil {
|
|
return nil, err
|
|
} else {
|
|
return &rotateSyncer{RotateLogs: rotateLogger}, nil
|
|
}
|
|
}
|
|
func (r *rotateSyncer) SetOnClose(closeFunc func()) {
|
|
r.setOnceOnclose.Do(func() {
|
|
r.onClose = closeFunc
|
|
})
|
|
}
|
|
|
|
func (r *rotateSyncer) Rotate() error {
|
|
return r.RotateLogs.Rotate()
|
|
}
|
|
func (r *rotateSyncer) Close() error {
|
|
r.onceOnclose.Do(func() {
|
|
if r.onClose != nil {
|
|
r.onClose()
|
|
r.onClose = nil
|
|
}
|
|
})
|
|
return r.RotateLogs.Close()
|
|
}
|
|
|
|
func (r *rotateSyncer) Sync() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *rotateSyncer) Write(bs []byte) (int, error) {
|
|
return s.RotateLogs.Write(bs)
|
|
}
|