/
reporter.go
70 lines (59 loc) · 1.56 KB
/
reporter.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
// Copyright 2018 GRAIL, Inc. All rights reserved.
// Use of this source code is governed by the Apache 2.0
// license that can be found in the LICENSE file.
package traverse
import (
"fmt"
"os"
"sync"
)
// A Reporter receives events from an ongoing traversal. Reporters
// can be passed as options into Traverse, and are used to monitor
// progress of long-running traversals.
type Reporter interface {
// Init is called when processing is about to begin. Parameter
// n indicates the number of tasks to be executed by the traversal.
Init(n int)
// Complete is called after the traversal has completed.
Complete()
// Begin is called when task i is begun.
Begin(i int)
// End is called when task i has completed.
End(i int)
}
// NewSimpleReporter returns a new reporter that prints the number
// of queued, running, and completed tasks to stderr.
func NewSimpleReporter(name string) Reporter {
return &simpleReporter{name: name}
}
type simpleReporter struct {
name string
mu sync.Mutex
queued, running, done int
}
func (r *simpleReporter) Init(n int) {
r.mu.Lock()
r.queued = n
r.update()
r.mu.Unlock()
}
func (r *simpleReporter) Complete() {
fmt.Fprintf(os.Stderr, "\n")
}
func (r *simpleReporter) Begin(i int) {
r.mu.Lock()
r.queued--
r.running++
r.update()
r.mu.Unlock()
}
func (r *simpleReporter) End(i int) {
r.mu.Lock()
r.running--
r.done++
r.update()
r.mu.Unlock()
}
func (r *simpleReporter) update() {
fmt.Fprintf(os.Stderr, "%s: (queued: %d -> running: %d -> done: %d) \r", r.name, r.queued, r.running, r.done)
}