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

WaitForAllTasks reimplemented

Browse Source
This commit is contained in:
milarin 2023-03-10 23:15:09 +01:00
parent cb3703217d
commit 18f58d0bf1
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
view.go
View File

@ -20,6 +20,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] {
@ -33,6 +34,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{}, 1),
lastReports: map[K]report[K]{}, lastReports: map[K]report[K]{},
wg: &sync.WaitGroup{}, wg: &sync.WaitGroup{},
} }
@ -63,6 +65,8 @@ func (v *View[K]) consumeReports(printReports bool) {
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) {
@ -96,3 +100,7 @@ func (v *View[K]) Done(task K) {
v.wg.Done() v.wg.Done()
} }
func (v *View[K]) WaitForAllTasks() {
<-v.doneCh
}