initial import
This commit is contained in:
37
core/threading/routinegroup.go
Normal file
37
core/threading/routinegroup.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package threading
|
||||
|
||||
import "sync"
|
||||
|
||||
type RoutineGroup struct {
|
||||
waitGroup sync.WaitGroup
|
||||
}
|
||||
|
||||
func NewRoutineGroup() *RoutineGroup {
|
||||
return new(RoutineGroup)
|
||||
}
|
||||
|
||||
// Don't reference the variables from outside,
|
||||
// because outside variables can be changed by other goroutines
|
||||
func (g *RoutineGroup) Run(fn func()) {
|
||||
g.waitGroup.Add(1)
|
||||
|
||||
go func() {
|
||||
defer g.waitGroup.Done()
|
||||
fn()
|
||||
}()
|
||||
}
|
||||
|
||||
// Don't reference the variables from outside,
|
||||
// because outside variables can be changed by other goroutines
|
||||
func (g *RoutineGroup) RunSafe(fn func()) {
|
||||
g.waitGroup.Add(1)
|
||||
|
||||
GoSafe(func() {
|
||||
defer g.waitGroup.Done()
|
||||
fn()
|
||||
})
|
||||
}
|
||||
|
||||
func (g *RoutineGroup) Wait() {
|
||||
g.waitGroup.Wait()
|
||||
}
|
||||
Reference in New Issue
Block a user