From 794b44835f98df719f2b1186b2ad90a11f9a35b0 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期四, 08 六月 2023 20:28:11 +0800 Subject: [PATCH] fix --- src/util/util.go | 16 ++++++++++------ 1 files changed, 10 insertions(+), 6 deletions(-) diff --git a/src/util/util.go b/src/util/util.go index 1831dae..9e0170e 100644 --- a/src/util/util.go +++ b/src/util/util.go @@ -3,13 +3,11 @@ 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" @@ -150,13 +148,19 @@ return nil } -func GetClient() (*kubernetes.Clientset, error) { +var kubeConfig string + +func GetClient(cluster string) (*kubernetes.Clientset, error) { // 閰嶇疆 Kubernetes 闆嗙兢鐨� kubeconfig 璺緞 - kubeconfig := flag.String("kubeconfig", filepath.Join(HomeDir(), ".kube", "config"), "kubeconfig file") - flag.Parse() + //if kubeConfig == nil { + // kubeConfig = flag.String("kubeconfig", filepath.Join(HomeDir(), ".kube", cluster, "config"), "kubeconfig file") + // flag.Parse() + //} + + kubeConfig = fmt.Sprintf("%s/.kube/%s/config", HomeDir(), cluster) // 鍒涘缓 Kubernetes 瀹㈡埛绔� - config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig) + config, err := clientcmd.BuildConfigFromFlags("", kubeConfig) if err != nil { panic(err.Error()) } -- Gitblit v1.8.0