Michael McGuinness
92334ad1dd
Set Up S3 integration * cfginterfaceandfirsts3funcs * addlocalstack * generallypassesfullsuite * addedmultipleattemptedpings * cleanup * stabiliseplusskip
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package wait
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
var (
|
|
// VisitStop is used as a return value from [VisitFunc] to stop the walk.
|
|
// It is not returned as an error by any function.
|
|
VisitStop = errors.New("stop the walk")
|
|
|
|
// VisitRemove is used as a return value from [VisitFunc] to have the current node removed.
|
|
// It is not returned as an error by any function.
|
|
VisitRemove = errors.New("remove this strategy")
|
|
)
|
|
|
|
// VisitFunc is a function that visits a strategy node.
|
|
// If it returns [VisitStop], the walk stops.
|
|
// If it returns [VisitRemove], the current node is removed.
|
|
type VisitFunc func(root Strategy) error
|
|
|
|
// Walk walks the strategies tree and calls the visit function for each node.
|
|
func Walk(root *Strategy, visit VisitFunc) error {
|
|
if root == nil {
|
|
return errors.New("root strategy is nil")
|
|
}
|
|
|
|
if err := walk(root, visit); err != nil {
|
|
if errors.Is(err, VisitRemove) || errors.Is(err, VisitStop) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// walk walks the strategies tree and calls the visit function for each node.
|
|
// It returns an error if the visit function returns an error.
|
|
func walk(root *Strategy, visit VisitFunc) error {
|
|
if *root == nil {
|
|
// No strategy.
|
|
return nil
|
|
}
|
|
|
|
// Allow the visit function to customize the behaviour of the walk before visiting the children.
|
|
if err := visit(*root); err != nil {
|
|
if errors.Is(err, VisitRemove) {
|
|
*root = nil
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
if s, ok := (*root).(*MultiStrategy); ok {
|
|
var i int
|
|
for range s.Strategies {
|
|
if err := walk(&s.Strategies[i], visit); err != nil {
|
|
if errors.Is(err, VisitRemove) {
|
|
s.Strategies = append(s.Strategies[:i], s.Strategies[i+1:]...)
|
|
if errors.Is(err, VisitStop) {
|
|
return VisitStop
|
|
}
|
|
continue
|
|
}
|
|
|
|
return err
|
|
}
|
|
i++
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|