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

package util

import "fmt"

type Progress struct {
	Key       string `json:"key,omitzero"`
	Total     int    `json:"total,omitzero"`
	Completed int    `json:"completed,omitzero"`
	fns       []func(p *Progress, delta int)
}

func NewProgress(key string, total int, fns ...func(p *Progress, delta int)) *Progress {
	if total == 0 {
		total = 100
	}
	return &Progress{Key: key, Total: total, fns: fns}
}

func (p *Progress) String() string {
	return fmt.Sprintf("%s (%d of %d)", p.Key, p.Completed, p.Total)
}

func (p *Progress) Increment(i int, logger Logger) {
	if p == nil {
		return
	}
	p.Completed += i
	p.call(i, logger)
}

func (p *Progress) call(delta int, logger Logger) {
	if logger != nil {
		logger.Debugf("%s [%d]", p.String(), delta)
	}
	for _, fn := range p.fns {
		fn(p, delta)
	}
}