github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-01-23 09:00:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: add test for logging rotate size (#3587)

Browse Source
This commit is contained in:
Kevin Wan 2023-09-24 22:28:03 +08:00 committed by GitHub
parent 814852f0b8
commit 4222fd97bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
core/logx/rotatelogger.go
View File

@ -298,6 +298,7 @@ func (l *RotateLogger) initialize() error {
if l.fp, err = os.OpenFile(l.filename, os.O_APPEND|os.O_WRONLY, defaultFileMode); err != nil { if l.fp, err = os.OpenFile(l.filename, os.O_APPEND|os.O_WRONLY, defaultFileMode); err != nil {
return err return err
} }
l.currentSize = fileInfo.Size() l.currentSize = fileInfo.Size()
} }

15
core/logx/rotatelogger_test.go
View File

@ -517,6 +517,21 @@ func TestGzipFile(t *testing.T) {
}) })
} }
func TestRotateLogger_WithExistingFile(t *testing.T) {
const body = "foo"
filename, err := fs.TempFilenameWithText(body)
assert.Nil(t, err)
if len(filename) > 0 {
defer os.Remove(filename)
}
rule := NewSizeLimitRotateRule(filename, "-", 1, 100, 3, false)
logger, err := NewLogger(filename, rule, false)
assert.Nil(t, err)
assert.Equal(t, int64(len(body)), logger.currentSize)
assert.Nil(t, logger.Close())
}
func BenchmarkRotateLogger(b *testing.B) { func BenchmarkRotateLogger(b *testing.B) {
filename := "./test.log" filename := "./test.log"
filename2 := "./test2.log" filename2 := "./test2.log"