milarin/statview
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed implementation of WaitForAllTasks

Browse Source
This commit is contained in:
milarin 2023-03-06 14:09:42 +01:00
parent 132a2a4200
commit 3cb1984024
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
view.go
View File

@ -19,6 +19,7 @@ type View[K comparable] struct {
wg *sync.WaitGroup
reportCh chan report[K]
doneCh chan struct{}
}
func New[K comparable]() *View[K] {
@ -32,6 +33,7 @@ func NewForWriter[K comparable](w io.Writer) *View[K] {
taskMap: map[K]struct{}{},
reportCh: make(chan report[K], 100),
doneCh: make(chan struct{}),
lastReports: map[K]report[K]{},
wg: &sync.WaitGroup{},
}
@ -48,6 +50,8 @@ func (v *View[K]) Show() {
v.Add(status.ID)
v.print()
}
v.doneCh <- struct{}{}
}
func (v *View[K]) SortFunc(f func(i, j K) bool) {
@ -79,5 +83,5 @@ func (v *View[K]) Done(task K) {
}
func (v *View[K]) WaitForAllTasks() {
v.wg.Wait()
<-v.doneCh
}