netmaker/netclient/ncutils/util.go
2022-03-21 11:27:25 -04:00

26 lines
633 B
Go

package ncutils
import (
"fmt"
"time"
"github.com/gravitl/netmaker/logger"
)
// BackOff - back off any function while there is an error
func BackOff(isExponential bool, maxTime int, f interface{}) (interface{}, error) {
// maxTime seconds
startTime := time.Now()
sleepTime := time.Second
for time.Now().Before(startTime.Add(time.Second * time.Duration(maxTime))) {
if result, err := f.(func() (interface{}, error))(); err == nil {
return result, nil
}
time.Sleep(sleepTime)
if isExponential {
sleepTime = sleepTime << 1
}
logger.Log(1, "retrying...")
}
return nil, fmt.Errorf("could not find result")
}