如何用Golang实现多集群部署_Golang 多集群管理与实践

package main

import (
    "k8s.io/client-go/kubernetes"
    "k8s.io/client-go/tools/clientcmd"
    "k8s.io/client-go/rest"
    "fmt"
)

func getClientset(kubeconfigPath string) (*kubernetes.Clientset, error) {
    config, err := clientcmd.BuildConfigFromFlags("", kubeconfigPath)
    if err != nil {
        return nil, err
    }
    return kubernetes.NewForConfig(config)
}

func main() {
    clusters := map[string]string{
        "cluster-east": "/path/to/east.kubeconfig",
        "cluster-west": "/path/to/west.kubeconfig",
    }

    for name, kubeconfig := range clusters {
        clientset, err := getClientset(kubeconfig)
        if err != nil {
            fmt.Printf("无法连接集群 %s: %v\n", name, err)
            continue
        }

        // 获取节点列表作为测试
        nodes, err := clientset.CoreV1().Nodes().List(context.TODO(), metav1.ListOptions{})
        if err != nil {
            fmt.Printf("查询集群 %s 节点失败: %v\n", name, err)
            continue
        }
        fmt.Printf("集群 %s 有 %d 个节点\n", name, len(nodes.Items))
    }
}

type DeploymentTask struct {
    ClusterName   string
    Clientset     *kubernetes.Clientset
    Namespace     string
    ImageVersion  string
}

func deployToCluster(task DeploymentTask) error {
    // 更新 Deployment 的 Image 字段
    deploymentClient := task.Clientset.AppsV1().Deployments(task.Namespace)
    dep, err := deploymentClient.Get(context.TODO(), "my-app", metav1.GetOptions{})
    if err != nil {
        return err
    }

    dep.Spec.Template.Spec.Containers[0].Image = task.ImageVersion
    _, err = deploymentClient.Update(context.TODO(), dep, metav1.UpdateOptions{})
    return err
}

var wg sync.WaitGroup
for _, task := range tasks {
    wg.Add(1)
    go func(t DeploymentTask) {
        defer wg.Done()
        if err := deployToCluster(t); err != nil {
            fmt.Printf("部署到 %s 失败: %v\n", t.ClusterName, err)
        } else {
            fmt.Printf("成功部署到 %s\n", t.ClusterName)
        }
    }(task)
}
wg.Wait()

func isClusterHealthy(clientset *kubernetes.Clientset) bool {
    _, err := clientset.Discovery().ServerVersion()
    return err == nil
}