argo workflow官方提供的k8s客户端代码示例

柔情只为你懂 2023-10-02 20:15 125阅读 0赞

package main
    
    import (
    	"flag"
    	"fmt"
    	"os/user"
    	"path/filepath"
    
    	wfv1 "github.com/argoproj/argo/pkg/apis/workflow/v1alpha1"
    	wfclientset "github.com/argoproj/argo/pkg/client/clientset/versioned"
    	"github.com/argoproj/pkg/errors"
    	corev1 "k8s.io/api/core/v1"
    	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
    	"k8s.io/apimachinery/pkg/fields"
    	"k8s.io/client-go/tools/clientcmd"
    )
    
    var (
    	helloWorldWorkflow = wfv1.Workflow{
        
    		ObjectMeta: metav1.ObjectMeta{
        
    			GenerateName: "hello-world-",
    		},
    		Spec: wfv1.WorkflowSpec{
        
    			Entrypoint: "whalesay",
    			Templates: []wfv1.Template{
        
    				{
        
    					Name: "whalesay",
    					Container: &corev1.Container{
        
    						Image:   "docker/whalesay:latest",
    						Command: []string{
        "cowsay", "hello world"},
    					},
    				},
    			},
    		},
    	}
    )
    
    func main() {
        
    	// get current user to determine home directory
    	usr, err := user.Current()
    	checkErr(err)
    
    	// get kubeconfig file location
    	kubeconfig := flag.String("kubeconfig", filepath.Join(usr.HomeDir, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
    	flag.Parse()
    
    	// use the current context in kubeconfig
    	config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig)
    	checkErr(err)
    	namespace := "default"
    
    	// create the workflow client
    	wfClient := wfclientset.NewForConfigOrDie(config).ArgoprojV1alpha1().Workflows(namespace)
    
    	// submit the hello world workflow
    	createdWf, err := wfClient.Create(&helloWorldWorkflow)
    	checkErr(err)
    	fmt.Printf("Workflow %s submitted\n", createdWf.Name)
    
    	// wait for the workflow to complete
    	fieldSelector := fields.ParseSelectorOrDie(fmt.Sprintf("metadata.name=%s", createdWf.Name))
    	watchIf, err := wfClient.Watch(metav1.ListOptions{
        FieldSelector: fieldSelector.String()})
    	errors.CheckError(err)
    	defer watchIf.Stop()
    	for next := range watchIf.ResultChan() {
        
    		wf, ok := next.Object.(*wfv1.Workflow)
    		if !ok {
        
    			continue
    		}
    		if !wf.Status.FinishedAt.IsZero() {
        
    			fmt.Printf("Workflow %s %s at %v\n", wf.Name, wf.Status.Phase, wf.Status.FinishedAt)
    			break
    		}
    	}
    }
    
    func checkErr(err error) {
        
    	if err != nil {
        
    		panic(err.Error())
    	}
    }

参考链接：  
[https://github.com/quantopian/argo/blob/master/docs/rest-api.md][https_github.com_quantopian_argo_blob_master_docs_rest-api.md]

[https_github.com_quantopian_argo_blob_master_docs_rest-api.md]: https://github.com/quantopian/argo/blob/master/docs/rest-api.md