From fea217048591823280a888b6c26f68558e51dded Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 12 一月 2024 15:17:53 +0800 Subject: [PATCH] 增加admin_grpc环境变量 --- 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