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