fix(cli): Add retry to retry ;). Fixes argoproj#4101 (argoproj#4105)

simster7 · Sep 23, 2020 · c4d9102 · c4d9102
1 parent ff195f2
commit c4d9102
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/workflow/util/util.go b/workflow/util/util.go
@@ -665,7 +665,19 @@ func RetryWorkflow(kubeClient kubernetes.Interface, hydrator hydrator.Interface,
  default:
  return nil, errors.Errorf(errors.CodeBadRequest, "workflow must be Failed/Error to retry")
  }
+ var updated *wfv1.Workflow
+ err := wait.ExponentialBackoff(retry.DefaultRetry, func() (bool, error) {
+ var err error
+ updated, err = retryWorkflow(kubeClient, hydrator, wfClient, wf, restartSuccessful, nodeFieldSelector)
+ return err == nil, err
+ })
+ if err != nil {
+ return nil, err
+ }
+ return updated, err
+}
 
+func retryWorkflow(kubeClient kubernetes.Interface, hydrator hydrator.Interface, wfClient v1alpha1.WorkflowInterface, wf *wfv1.Workflow, restartSuccessful bool, nodeFieldSelector string) (*wfv1.Workflow, error) {
  err := hydrator.Hydrate(wf)
  if err != nil {
  return nil, err