summaryrefslogtreecommitdiff
path: root/internal/app/load_progress.go
blob: 38679fa3d0cf327f890ececaedff733fbb8a82b0 (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
package app

import "sync"

type loadProgress struct {
	mu        sync.Mutex
	total     int
	processed int
	done      bool
}

func (p *loadProgress) Reset() {
	if p == nil {
		return
	}
	p.mu.Lock()
	p.total = 0
	p.processed = 0
	p.done = false
	p.mu.Unlock()
}

func (p *loadProgress) SetTotal(total int) {
	if p == nil {
		return
	}
	p.mu.Lock()
	p.total = total
	p.mu.Unlock()
}

func (p *loadProgress) Increment() {
	if p == nil {
		return
	}
	p.mu.Lock()
	p.processed++
	p.mu.Unlock()
}

func (p *loadProgress) MarkDone() {
	if p == nil {
		return
	}
	p.mu.Lock()
	p.done = true
	p.mu.Unlock()
}

func (p *loadProgress) Snapshot() (processed, total int, done bool) {
	if p == nil {
		return 0, 0, true
	}
	p.mu.Lock()
	defer p.mu.Unlock()
	return p.processed, p.total, p.done
}