forked from argoproj/argo-workflows
/
retry.go
80 lines (68 loc) · 2.03 KB
/
retry.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package commands
import (
"os"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"github.com/argoproj/pkg/errors"
"github.com/argoproj/argo/cmd/argo/commands/client"
workflowpkg "github.com/argoproj/argo/pkg/apiclient/workflow"
"github.com/argoproj/argo/workflow/packer"
"github.com/argoproj/argo/workflow/util"
)
func NewRetryCommand() *cobra.Command {
var (
cliSubmitOpts cliSubmitOpts
)
var command = &cobra.Command{
Use: "retry WORKFLOW",
Short: "retry a workflow",
Run: func(cmd *cobra.Command, args []string) {
if len(args) == 0 {
cmd.HelpFunc()(cmd, args)
os.Exit(1)
}
if client.ArgoServer != "" {
apiServerWFRetry(args[0], cliSubmitOpts)
} else {
kubeClient := InitKubeClient()
wfClient := InitWorkflowClient()
wf, err := wfClient.Get(args[0], metav1.GetOptions{})
if err != nil {
log.Fatal(err)
}
err = packer.DecompressWorkflow(wf)
if err != nil {
log.Fatal(err)
}
wf, err = util.RetryWorkflow(kubeClient, wfClient, wf)
if err != nil {
log.Fatal(err)
}
printWorkflow(wf, cliSubmitOpts.output, DefaultStatus)
waitOrWatch([]string{wf.Name}, cliSubmitOpts)
}
},
}
command.Flags().StringVarP(&cliSubmitOpts.output, "output", "o", "", "Output format. One of: name|json|yaml|wide")
command.Flags().BoolVarP(&cliSubmitOpts.wait, "wait", "w", false, "wait for the workflow to complete")
command.Flags().BoolVar(&cliSubmitOpts.watch, "watch", false, "watch the workflow until it completes")
return command
}
func apiServerWFRetry(wfName string, opts cliSubmitOpts) {
conn := client.GetClientConn()
defer conn.Close()
ns, _, _ := client.Config.Namespace()
wfApiClient, ctx := GetWFApiServerGRPCClient(conn)
wfReq := workflowpkg.WorkflowRetryRequest{
Name: wfName,
Namespace: ns,
}
wf, err := wfApiClient.RetryWorkflow(ctx, &wfReq)
if err != nil {
errors.CheckError(err)
return
}
printWorkflow(wf, opts.output, DefaultStatus)
waitOrWatch([]string{wf.Name}, opts)
}