eosswedenorg/thalos
1
0
Fork 0
mirror of https://github.com/eosswedenorg/thalos synced 2026-06-18 04:40:03 +02:00
Code Releases Activity

internal/log/RotatingFile.go: use option functions.

Browse source
This commit is contained in:
Henrik Hautakoski 2024-02-21 17:22:06 +01:00
parent bcdaa9cea4
commit 90b10f39ae
1 changed files with 19 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

29
internal/log/RotatingFile.go
View file

@ -24,7 +24,7 @@ func open(filename string) (*os.File, error) {
} }
// Open a new rotating file. // Open a new rotating file.
func NewRotatingFile(filename string, maxSize int64, maxAge time.Duration) (*RotatingFile, error) { func NewRotatingFile(filename string, opts ...RotatingFileOption) (*RotatingFile, error) {
if err := os.MkdirAll(path.Dir(filename), 0o766); err != nil && !os.IsExist(err) { if err := os.MkdirAll(path.Dir(filename), 0o766); err != nil && !os.IsExist(err) {
return nil, err return nil, err
} }
@ -39,14 +39,18 @@ func NewRotatingFile(filename string, maxSize int64, maxAge time.Duration) (*Rot
return nil, err return nil, err
} }
return &RotatingFile{ file := &RotatingFile{
fd: fd, fd: fd,
size: stat.Size(), size: stat.Size(),
maxSize: maxSize, ts: time.Now(),
ts: time.Now(), format: "2006-01-02_150405",
maxAge: maxAge, }
format: "2006-01-02_150405",
}, nil for _, opt := range opts {
opt(file)
}
return file, nil
} }
// Open a new rotating file using a config struct. // Open a new rotating file using a config struct.
@ -55,7 +59,12 @@ func NewRotatingFileFromConfig(config Config, suffix string) (*RotatingFile, err
suffix = "_" + suffix suffix = "_" + suffix
} }
return NewRotatingFile(config.GetFilePath()+suffix+".log", int64(config.MaxFileSize), config.MaxTime) filename := config.GetFilePath() + suffix + ".log"
return NewRotatingFile(filename, func(f *RotatingFile) {
f.maxAge = config.MaxTime
f.maxSize = int64(config.MaxFileSize)
})
} }
func (w *RotatingFile) newFilename(name string) string { func (w *RotatingFile) newFilename(name string) string {