add customizable retry

Signed-off-by: Markus Blaschke <mblaschke82@gmail.com>
webdevops · Mar 3, 2024 · 3824d71 · 3824d71
1 parent 164825d
commit 3824d71
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/config/opts.go b/config/opts.go
@@ -15,8 +15,11 @@ type (
 
 		Fenecon struct {
 			Request struct {
-				Timeout  time.Duration `long:"fenecon.request.timeout"  env:"FENECON_REQUEST_TIMEOUT"  description:"Request timeout" default:"10s"`
-				Parallel int           `long:"fenecon.request.parallel"  env:"FENECON_REQUEST_PARALLEL"  description:"Number of parallel requests" default:"10"`
+				Timeout          time.Duration `long:"fenecon.request.timeout"       env:"FENECON_REQUEST_TIMEOUT"       description:"Request timeout"              default:"10s"`
+				Parallel         int           `long:"fenecon.request.parallel"      env:"FENECON_REQUEST_PARALLEL"      description:"Number of parallel requests"  default:"3"`
+				RetryCount       int           `long:"fenecon.request.retries"       env:"FENECON_REQUEST_RETRIES"       description:"Request retries"              default:"1"`
+				RetryWaitTime    time.Duration `long:"fenecon.request.waittime"      env:"FENECON_REQUEST_WAITTIME"      description:"Request retries"              default:"2s"`
+				RetryMaxWaitTime time.Duration `long:"fenecon.request.maxwaittime"   env:"FENECON_REQUEST_MAXWAITTIME"   description:"Request retries"              default:"5s"`
 			}
 
 			Auth struct {

diff --git a/fenecon/prober.go b/fenecon/prober.go
@@ -79,6 +79,12 @@ func (fp *FeneconProber) SetParallelRequests(val int) {
 	fp.parallelRequests = val
 }
 
+func (fp *FeneconProber) SetRetry(retry int, waitTime, maxWaitTime time.Duration) {
+	fp.client.SetRetryCount(retry)
+	fp.client.SetRetryWaitTime(waitTime)
+	fp.client.SetRetryMaxWaitTime(maxWaitTime)
+}
+
 func (fp *FeneconProber) SetTimeout(timeout time.Duration) {
 	fp.client.SetTimeout(timeout)
 }

diff --git a/probe.go b/probe.go
@@ -22,6 +22,11 @@ func newFeneconProber(ctx context.Context, registry *prometheus.Registry, logger
 	sp.SetUserAgent(UserAgent + gitTag)
 	sp.SetTimeout(opts.Fenecon.Request.Timeout)
 	sp.SetParallelRequests(opts.Fenecon.Request.Parallel)
+	sp.SetRetry(
+		opts.Fenecon.Request.RetryCount,
+		opts.Fenecon.Request.RetryWaitTime,
+		opts.Fenecon.Request.RetryMaxWaitTime,
+	)
 	if len(opts.Fenecon.Auth.Password) >= 1 {
 		sp.SetHttpAuth(opts.Fenecon.Auth.Username, opts.Fenecon.Auth.Password)
 	}