github.com/nsqio/nsq@v1.3.0/internal/util/wait_group_wrapper.go (about)

     1  package util
     2  
     3  import (
     4  	"sync"
     5  )
     6  
     7  type WaitGroupWrapper struct {
     8  	sync.WaitGroup
     9  }
    10  
    11  func (w *WaitGroupWrapper) Wrap(cb func()) {
    12  	w.Add(1)
    13  	go func() {
    14  		cb()
    15  		w.Done()
    16  	}()
    17  }