blob: aeec15e63933970ebcca83af8d7023a25109c563 (
plain)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
package model
import "sync"
// ScanProgress tracks the state of an in-progress or recently completed scan.
type ScanProgress struct {
mu sync.RWMutex
Running bool `json:"running"`
CurrentSet string `json:"current_set,omitempty"`
SetsTotal int `json:"sets_total"`
SetsDone int `json:"sets_done"`
FilesTotal int `json:"files_total"`
FilesDone int `json:"files_done"`
LastError string `json:"last_error,omitempty"`
}
// Start marks a scan running and resets counters for the given set count.
func (p *ScanProgress) Start(setsTotal int) {
p.mu.Lock()
defer p.mu.Unlock()
p.Running = true
p.SetsTotal = setsTotal
p.SetsDone = 0
p.FilesTotal = 0
p.FilesDone = 0
p.LastError = ""
}
// SetCurrentSet records the set currently being scanned.
func (p *ScanProgress) SetCurrentSet(name string) {
p.mu.Lock()
defer p.mu.Unlock()
p.CurrentSet = name
}
// IncrementFile increments the completed file count.
func (p *ScanProgress) IncrementFile() {
p.mu.Lock()
defer p.mu.Unlock()
p.FilesDone++
}
// SetFilesTotal records the number of files to scan.
func (p *ScanProgress) SetFilesTotal(total int) {
p.mu.Lock()
defer p.mu.Unlock()
p.FilesTotal = total
}
// AddFilesTotal adds to the total number of files discovered during the scan.
func (p *ScanProgress) AddFilesTotal(total int) {
p.mu.Lock()
defer p.mu.Unlock()
p.FilesTotal += total
}
// IncrementSet increments the completed set count.
func (p *ScanProgress) IncrementSet() {
p.mu.Lock()
defer p.mu.Unlock()
p.SetsDone++
}
// Done marks the scan complete and records an error message when provided.
func (p *ScanProgress) Done(err error) {
p.mu.Lock()
defer p.mu.Unlock()
p.Running = false
p.CurrentSet = ""
if err != nil {
p.LastError = err.Error()
}
}
// Copy returns a race-safe snapshot of the scan progress.
func (p *ScanProgress) Copy() ScanProgress {
p.mu.RLock()
defer p.mu.RUnlock()
return ScanProgress{
Running: p.Running,
CurrentSet: p.CurrentSet,
SetsTotal: p.SetsTotal,
SetsDone: p.SetsDone,
FilesTotal: p.FilesTotal,
FilesDone: p.FilesDone,
LastError: p.LastError,
}
}
|
