feat: [CI-11274]: Added retries sending step status

harness · Feb 8, 2024 · d736476 · d736476
1 parent d7ab998
commit d736476
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/delegate/http.go b/delegate/http.go
@@ -31,8 +31,9 @@ const (
 )
 
 var (
-	registerTimeout   = 30 * time.Second
-	taskEventsTimeout = 60 * time.Second
+	registerTimeout      = 30 * time.Second
+	taskEventsTimeout    = 60 * time.Second
+	sendStatusRetryTimes = 5
 )
 
 // defaultClient is the default http.Client.
@@ -191,7 +192,15 @@ func (p *HTTPClient) Acquire(ctx context.Context, delegateID, taskID string) (*c
 func (p *HTTPClient) SendStatus(ctx context.Context, delegateID, taskID string, r *client.TaskResponse) error {
 	path := fmt.Sprintf(taskStatusEndpoint, taskID, delegateID, p.AccountID)
 	req := r
-	_, err := p.retry(ctx, path, "POST", req, nil, createBackoff(ctx, taskEventsTimeout), true) //nolint: bodyclose
+	retryNumber := 0
+	var err error
+	for retryNumber < sendStatusRetryTimes {
+		_, err = p.retry(ctx, path, "POST", req, nil, createBackoff(ctx, taskEventsTimeout), true) //nolint: bodyclose
+		if err == nil {
+			return nil
+		}
+		retryNumber++
+	}
 	return err
 }