wait_group.go

type waitGroup struct {
	sync.WaitGroup
}

func (wg *waitGroup) DoneChan() chan struct{} {
	ch := make(chan struct{})
	go func() {
		defer close(ch)
		wg.Wait()
	}()
	return ch
}