25 lines
325 B
Go
25 lines
325 B
Go
|
package poller
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
type routineGroup struct {
|
||
|
|
waitGroup sync.WaitGroup
|
||
|
|
}
|
||
|
|
|
||
|
|
func newRoutineGroup() *routineGroup {
|
||
|
|
return new(routineGroup)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *routineGroup) Run(fn func()) {
|
||
|
|
g.waitGroup.Add(1)
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
defer g.waitGroup.Done()
|
||
|
|
fn()
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *routineGroup) Wait() {
|
||
|
|
g.waitGroup.Wait()
|
||
|
|
}
|