mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 04:50:52 +08:00
3e4858a624
This commit introduces a new configuration option `logs-max-total-size-mb` that allows users to set a maximum total size (in MB) for log files in the logs directory. When this limit is exceeded, the oldest log files will be automatically deleted to stay within the specified size. Setting this value to 0 (the default) disables this feature. This change enhances log management by preventing excessive disk space usage.
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package logging
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestEnforceLogDirSizeLimitDeletesOldest(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
writeLogFile(t, filepath.Join(dir, "old.log"), 60, time.Unix(1, 0))
|
|
writeLogFile(t, filepath.Join(dir, "mid.log"), 60, time.Unix(2, 0))
|
|
protected := filepath.Join(dir, "main.log")
|
|
writeLogFile(t, protected, 60, time.Unix(3, 0))
|
|
|
|
deleted, err := enforceLogDirSizeLimit(dir, 120, protected)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if deleted != 1 {
|
|
t.Fatalf("expected 1 deleted file, got %d", deleted)
|
|
}
|
|
|
|
if _, err := os.Stat(filepath.Join(dir, "old.log")); !os.IsNotExist(err) {
|
|
t.Fatalf("expected old.log to be removed, stat error: %v", err)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(dir, "mid.log")); err != nil {
|
|
t.Fatalf("expected mid.log to remain, stat error: %v", err)
|
|
}
|
|
if _, err := os.Stat(protected); err != nil {
|
|
t.Fatalf("expected protected main.log to remain, stat error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestEnforceLogDirSizeLimitSkipsProtected(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
protected := filepath.Join(dir, "main.log")
|
|
writeLogFile(t, protected, 200, time.Unix(1, 0))
|
|
writeLogFile(t, filepath.Join(dir, "other.log"), 50, time.Unix(2, 0))
|
|
|
|
deleted, err := enforceLogDirSizeLimit(dir, 100, protected)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if deleted != 1 {
|
|
t.Fatalf("expected 1 deleted file, got %d", deleted)
|
|
}
|
|
|
|
if _, err := os.Stat(protected); err != nil {
|
|
t.Fatalf("expected protected main.log to remain, stat error: %v", err)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(dir, "other.log")); !os.IsNotExist(err) {
|
|
t.Fatalf("expected other.log to be removed, stat error: %v", err)
|
|
}
|
|
}
|
|
|
|
func writeLogFile(t *testing.T, path string, size int, modTime time.Time) {
|
|
t.Helper()
|
|
|
|
data := make([]byte, size)
|
|
if err := os.WriteFile(path, data, 0o644); err != nil {
|
|
t.Fatalf("write file: %v", err)
|
|
}
|
|
if err := os.Chtimes(path, modTime, modTime); err != nil {
|
|
t.Fatalf("set times: %v", err)
|
|
}
|
|
}
|