forked from goharbor/harbor
-
Notifications
You must be signed in to change notification settings - Fork 0
/
file_sweeper.go
70 lines (58 loc) · 1.36 KB
/
file_sweeper.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package sweeper
import (
"fmt"
"io/ioutil"
"os"
"path"
"strings"
"time"
)
const (
oneDay = 24 * time.Hour
)
// FileSweeper is used to sweep the file logs
type FileSweeper struct {
duration int
workDir string
}
// NewFileSweeper is constructor of FileSweeper
func NewFileSweeper(workDir string, duration int) *FileSweeper {
return &FileSweeper{
workDir: workDir,
duration: duration,
}
}
// Sweep logs
func (fs *FileSweeper) Sweep() (int, error) {
cleared := 0
logFiles, err := ioutil.ReadDir(fs.workDir)
if err != nil {
return 0, fmt.Errorf("getting outdated log files under '%s' failed with error: %s", fs.workDir, err)
}
// Nothing to sweep
if len(logFiles) == 0 {
return 0, nil
}
// Start to sweep log files
// Record all errors
errs := make([]string, 0)
for _, logFile := range logFiles {
if logFile.ModTime().Add(time.Duration(fs.duration) * oneDay).Before(time.Now()) {
logFilePath := path.Join(fs.workDir, logFile.Name())
if err := os.Remove(logFilePath); err != nil {
errs = append(errs, fmt.Sprintf("remove log file '%s' error: %s", logFilePath, err))
continue // go on for next one
}
cleared++
}
}
if len(errs) > 0 {
err = fmt.Errorf("%s", strings.Join(errs, "\n"))
}
// Return error with high priority
return cleared, err
}
// Duration for sweeping
func (fs *FileSweeper) Duration() int {
return fs.duration
}