wangpengfei
2023-06-02 064c0874e5fd041c4641ef873d1bf72ac98a184d
src/util/util.go
@@ -3,9 +3,13 @@
import (
   "context"
   "crypto/tls"
   "flag"
   "fmt"
   "k8s.io/client-go/kubernetes"
   "k8s.io/client-go/tools/clientcmd"
   "net/http"
   "os"
   "path/filepath"
   "time"
   "golang.org/x/crypto/ssh"
@@ -145,3 +149,22 @@
   fmt.Println("kubectl has been installed on the remote server.")
   return nil
}
func GetClient() (*kubernetes.Clientset, error) {
   // 配置 Kubernetes 集群的 kubeconfig 路径
   kubeconfig := flag.String("kubeconfig", filepath.Join(HomeDir(), ".kube", "config"), "kubeconfig file")
   flag.Parse()
   // 创建 Kubernetes 客户端
   config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig)
   if err != nil {
      panic(err.Error())
   }
   clientset, err := kubernetes.NewForConfig(config)
   if err != nil {
      panic(err.Error())
   }
   return clientset, nil
}