fix 修改 util Cluster 根据ClusterId 保存并读取kube config

wangpengfei

2023-06-08 2ad213b923c4475efe7d76dfdcbb8323616a6a3a

 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,17 +148,19 @@
   return nil
}

var kubeConfig *string
var kubeConfig string

func GetClient() (*kubernetes.Clientset, error) {
func GetClient(cluster string) (*kubernetes.Clientset, error) {
   // 配置 Kubernetes 集群的 kubeconfig 路径
   if kubeConfig == nil {
      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())
   }