mirror of
https://github.com/gogrlx/nats-server.git
synced 2026-04-02 03:38:42 -07:00
Changes for max log files option (active plus backups); remove redundant lexical sort of backups; adjust test
This commit is contained in:
@@ -19,7 +19,6 @@ import (
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
@@ -131,14 +130,14 @@ type fileLogger struct {
|
||||
out int64
|
||||
canRotate int32
|
||||
sync.Mutex
|
||||
l *Logger
|
||||
f writerAndCloser
|
||||
limit int64
|
||||
olimit int64
|
||||
pid string
|
||||
time bool
|
||||
closed bool
|
||||
archiveLimit int
|
||||
l *Logger
|
||||
f writerAndCloser
|
||||
limit int64
|
||||
olimit int64
|
||||
pid string
|
||||
time bool
|
||||
closed bool
|
||||
maxNumFiles int
|
||||
}
|
||||
|
||||
func newFileLogger(filename, pidPrefix string, time bool) (*fileLogger, error) {
|
||||
@@ -173,9 +172,9 @@ func (l *fileLogger) setLimit(limit int64) {
|
||||
}
|
||||
}
|
||||
|
||||
func (l *fileLogger) setArchiveLimit(limit int) {
|
||||
func (l *fileLogger) setMaxNumFiles(max int) {
|
||||
l.Lock()
|
||||
l.archiveLimit = limit
|
||||
l.maxNumFiles = max
|
||||
l.Unlock()
|
||||
}
|
||||
|
||||
@@ -200,9 +199,7 @@ func (l *fileLogger) logDirect(label, format string, v ...interface{}) int {
|
||||
return len(entry)
|
||||
}
|
||||
|
||||
func (l *fileLogger) archivePurge(fname string) {
|
||||
// Evaluate number of saved backups for purge
|
||||
// l readlock held
|
||||
func (l *fileLogger) logPurge(fname string) {
|
||||
var backups []string
|
||||
lDir := filepath.Dir(fname)
|
||||
lBase := filepath.Base(fname)
|
||||
@@ -215,27 +212,24 @@ func (l *fileLogger) archivePurge(fname string) {
|
||||
if entry.IsDir() || entry.Name() == lBase || !strings.HasPrefix(entry.Name(), lBase) {
|
||||
continue
|
||||
}
|
||||
stamp, found := strings.CutPrefix(entry.Name(), fmt.Sprintf("%s%s", lBase, "."))
|
||||
if found {
|
||||
if stamp, found := strings.CutPrefix(entry.Name(), fmt.Sprintf("%s%s", lBase, ".")); found {
|
||||
_, err := time.Parse("2006:01:02:15:04:05.999999999", strings.Replace(stamp, ".", ":", 5))
|
||||
if err == nil {
|
||||
backups = append(backups, entry.Name())
|
||||
}
|
||||
}
|
||||
}
|
||||
archives := len(backups)
|
||||
if archives > l.archiveLimit {
|
||||
// Oldest to latest
|
||||
sort.Slice(backups, func(i, j int) bool {
|
||||
return backups[i] < backups[j]
|
||||
})
|
||||
for i := 0; i < archives-l.archiveLimit; i++ {
|
||||
if err := os.Remove(fmt.Sprintf("%s%s%s", lDir, string(os.PathSeparator), backups[i])); err != nil {
|
||||
currBackups := len(backups)
|
||||
maxBackups := l.maxNumFiles - 1
|
||||
if currBackups > maxBackups {
|
||||
// backups sorted oldest to latest based on timestamped lexical filename (ReadDir)
|
||||
for i := 0; i < currBackups-maxBackups; i++ {
|
||||
if err := os.Remove(filepath.Join(lDir, string(os.PathSeparator), backups[i])); err != nil {
|
||||
l.logDirect(l.l.errorLabel, "Unable to remove backup log file %q (%v), will attempt next rotation", backups[i], err)
|
||||
// Bail fast, we'll try again next rotation
|
||||
return
|
||||
}
|
||||
l.logDirect(l.l.infoLabel, "Removed archived log file %q", backups[i])
|
||||
l.logDirect(l.l.infoLabel, "Purged log file %q", backups[i])
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -275,8 +269,8 @@ func (l *fileLogger) Write(b []byte) (int, error) {
|
||||
n := l.logDirect(l.l.infoLabel, "Rotated log, backup saved as %q", bak)
|
||||
l.out = int64(n)
|
||||
l.limit = l.olimit
|
||||
if l.archiveLimit > 0 {
|
||||
l.archivePurge(fname)
|
||||
if l.maxNumFiles > 0 {
|
||||
l.logPurge(fname)
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -310,16 +304,16 @@ func (l *Logger) SetSizeLimit(limit int64) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// SetArchiveLimit sets the number of archived log files that will be retained
|
||||
func (l *Logger) SetArchiveLimit(limit int) error {
|
||||
// SetMaxNumFiles sets the number of archived log files that will be retained
|
||||
func (l *Logger) SetMaxNumFiles(max int) error {
|
||||
l.Lock()
|
||||
if l.fl == nil {
|
||||
l.Unlock()
|
||||
return fmt.Errorf("can set log archive limit only for file logger")
|
||||
return fmt.Errorf("can set log max number of files only for file logger")
|
||||
}
|
||||
fl := l.fl
|
||||
l.Unlock()
|
||||
fl.setArchiveLimit(limit)
|
||||
fl.setMaxNumFiles(max)
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
@@ -72,14 +72,14 @@ func (s *Server) ConfigureLogger() {
|
||||
l.SetSizeLimit(opts.LogSizeLimit)
|
||||
}
|
||||
}
|
||||
if opts.LogMaxArchives > 0 {
|
||||
if opts.LogMaxFiles > 0 {
|
||||
if l, ok := log.(*srvlog.Logger); ok {
|
||||
al := int(opts.LogMaxArchives)
|
||||
if int64(al) != opts.LogMaxArchives {
|
||||
al := int(opts.LogMaxFiles)
|
||||
if int64(al) != opts.LogMaxFiles {
|
||||
// set to default (no max) on overflow
|
||||
al = 0
|
||||
}
|
||||
l.SetArchiveLimit(al)
|
||||
l.SetMaxNumFiles(al)
|
||||
}
|
||||
}
|
||||
} else if opts.RemoteSyslog != "" {
|
||||
|
||||
@@ -311,7 +311,7 @@ type Options struct {
|
||||
PortsFileDir string `json:"-"`
|
||||
LogFile string `json:"-"`
|
||||
LogSizeLimit int64 `json:"-"`
|
||||
LogMaxArchives int64 `json:"-"`
|
||||
LogMaxFiles int64 `json:"-"`
|
||||
Syslog bool `json:"-"`
|
||||
RemoteSyslog string `json:"-"`
|
||||
Routes []*url.URL `json:"-"`
|
||||
@@ -1000,8 +1000,8 @@ func (o *Options) processConfigFileLine(k string, v interface{}, errors *[]error
|
||||
o.LogFile = v.(string)
|
||||
case "logfile_size_limit", "log_size_limit":
|
||||
o.LogSizeLimit = v.(int64)
|
||||
case "logfile_max_archives", "log_max_archives":
|
||||
o.LogMaxArchives = v.(int64)
|
||||
case "logfile_max_num", "log_max_num":
|
||||
o.LogMaxFiles = v.(int64)
|
||||
case "syslog":
|
||||
o.Syslog = v.(bool)
|
||||
trackExplicitVal(o, &o.inConfig, "Syslog", o.Syslog)
|
||||
|
||||
@@ -39,7 +39,7 @@ func TestLogMaxArchives(t *testing.T) {
|
||||
totEntriesExpected int
|
||||
}{
|
||||
{
|
||||
"Default implicit, no max archives, expect 0 purged archives",
|
||||
"Default implicit, no max logs, expect 0 purged logs",
|
||||
`
|
||||
port: -1
|
||||
log_file: %s
|
||||
@@ -48,52 +48,52 @@ func TestLogMaxArchives(t *testing.T) {
|
||||
9,
|
||||
},
|
||||
{
|
||||
"Default explicit, no max archives, expect 0 purged archives",
|
||||
"Default explicit, no max logs, expect 0 purged logs",
|
||||
`
|
||||
port: -1
|
||||
log_file: %s
|
||||
logfile_size_limit: 100
|
||||
logfile_max_archives: 0
|
||||
logfile_max_num: 0
|
||||
`,
|
||||
9,
|
||||
},
|
||||
{
|
||||
"Default explicit - negative val, no max archives, expect 0 purged archives",
|
||||
"Default explicit - negative val, no max logs, expect 0 purged logs",
|
||||
`
|
||||
port: -1
|
||||
log_file: %s
|
||||
logfile_size_limit: 100
|
||||
logfile_max_archives: -42
|
||||
logfile_max_num: -42
|
||||
`,
|
||||
9,
|
||||
},
|
||||
{
|
||||
"1-archive limit, expect 7 purged archives",
|
||||
"1-max num, expect 8 purged logs",
|
||||
`
|
||||
port: -1
|
||||
log_file: %s
|
||||
logfile_size_limit: 100
|
||||
logfile_max_archives: 1
|
||||
logfile_max_num: 1
|
||||
`,
|
||||
2,
|
||||
1,
|
||||
},
|
||||
{
|
||||
"5-archive limit, expect 4 purged archives",
|
||||
"5-max num, expect 4 purged logs; use opt alias",
|
||||
`
|
||||
port: -1
|
||||
log_file: %s
|
||||
logfile_size_limit: 100
|
||||
logfile_max_archives: 5
|
||||
log_size_limit: 100
|
||||
log_max_num: 5
|
||||
`,
|
||||
6,
|
||||
5,
|
||||
},
|
||||
{
|
||||
"100-archive limit, expect 0 purged archives",
|
||||
"100-max num, expect 0 purged logs",
|
||||
`
|
||||
port: -1
|
||||
log_file: %s
|
||||
logfile_size_limit: 100
|
||||
logfile_max_archives: 100
|
||||
logfile_max_num: 100
|
||||
`,
|
||||
9,
|
||||
},
|
||||
@@ -117,6 +117,8 @@ func TestLogMaxArchives(t *testing.T) {
|
||||
t.Fatalf("No NATS Server object returned")
|
||||
}
|
||||
s.Shutdown()
|
||||
// Windows filesystem can be a little pokey on the flush, so wait a bit after shutdown...
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
entries, err := os.ReadDir(d)
|
||||
if err != nil {
|
||||
t.Fatalf("Error reading dir: %v", err)
|
||||
|
||||
Reference in New Issue
Block a user