mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 13:56:27 +01:00
30 lines
449 B
Go
30 lines
449 B
Go
package genny
|
|
|
|
import "sync"
|
|
|
|
type Group struct {
|
|
Generators []*Generator
|
|
moot sync.RWMutex
|
|
}
|
|
|
|
func (gg *Group) Add(g *Generator) {
|
|
m := &gg.moot
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
gg.Generators = append(gg.Generators, g)
|
|
}
|
|
|
|
func (gg *Group) Merge(g2 *Group) {
|
|
for _, g := range g2.Generators {
|
|
gg.Add(g)
|
|
}
|
|
}
|
|
|
|
func (gg *Group) With(r *Runner) {
|
|
m := &gg.moot
|
|
m.RLock()
|
|
defer m.RUnlock()
|
|
for _, g := range gg.Generators {
|
|
r.With(g)
|
|
}
|
|
}
|