如何使用client-go动态切换Kubernetes上下文？

1. 切换指定上下文的核心实现 #

要切换到kubeconfig中指定的上下文，需先加载kubeconfig的原始结构，再通过配置覆盖指定目标上下文，最终生成客户端可用的配置。以下是完整实现代码，同时支持指定单个上下文和正则匹配集群关联的上下文两种场景：

package main

import (
	"context"
	"flag"
	"fmt"
	"path/filepath"
	"regexp"

	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/tools/clientcmd"
	"k8s.io/client-go/util/homedir"
)

func main() {
	var (
		kubeconfig    *string
		targetCtx     *string
		clusterRegexp *string
	)

	// 处理kubeconfig路径参数
	if home := homedir.HomeDir(); home != "" {
		kubeconfig = flag.String("kubeconfig", filepath.Join(home, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
	} else {
		kubeconfig = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
	}

	// 添加目标上下文参数
	targetCtx = flag.String("context", "", "target context name to switch to")
	// 添加集群名称正则匹配参数
	clusterRegexp = flag.String("cluster-regex", "", "regex pattern to match cluster names")

	flag.Parse()

	// 加载kubeconfig原始配置
	rawConfig, err := clientcmd.LoadFromFile(*kubeconfig)
	if err != nil {
		panic(err.Error())
	}

	var targetContexts []string

	// 处理正则匹配集群的场景
	if *clusterRegexp != "" {
		re, err := regexp.Compile(*clusterRegexp)
		if err != nil {
			panic(fmt.Sprintf("invalid regex pattern: %v", err))
		}

		// 遍历所有上下文，筛选出关联匹配集群的项
		for ctxName, ctx := range rawConfig.Contexts {
			if re.MatchString(ctx.Cluster) {
				targetContexts = append(targetContexts, ctxName)
			}
		}

		if len(targetContexts) == 0 {
			panic("no contexts found matching the cluster regex")
		}
	} else if *targetCtx != "" {
		// 处理指定单个上下文的场景
		targetContexts = append(targetContexts, *targetCtx)
	} else {
		// 默认使用当前上下文
		targetContexts = append(targetContexts, rawConfig.CurrentContext)
	}

	// 遍历所有目标上下文执行操作
	for _, ctxName := range targetContexts {
		fmt.Printf("=== Processing context: %s ===\n", ctxName)

		// 创建指定上下文的客户端配置
		clientConfig := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
			&clientcmd.ClientConfigLoadingRules{ExplicitPath: *kubeconfig},
			&clientcmd.ConfigOverrides{
				CurrentContext: ctxName,
			},
		)

		// 生成rest.Config用于创建K8s客户端
		restConfig, err := clientConfig.ClientConfig()
		if err != nil {
			fmt.Printf("Failed to get config for context %s: %v\n", ctxName, err)
			continue
		}

		// 创建K8s客户端集
		clientset, err := kubernetes.NewForConfig(restConfig)
		if err != nil {
			fmt.Printf("Failed to create clientset for context %s: %v\n", ctxName, err)
			continue
		}

		// 替换为你需要执行的任意操作（示例：获取全集群Pod数量）
		pods, err := clientset.CoreV1().Pods("").List(context.TODO(), metav1.ListOptions{})
		if err != nil {
			fmt.Printf("Failed to list pods in context %s: %v\n", ctxName, err)
			continue
		}
		fmt.Printf("There are %d pods in the cluster\n", len(pods.Items))
	}
}

2. 关键类型说明 #

• clientcmdapi.Config（RawConfig）：对应kubeconfig文件的完整结构，包含所有集群、上下文、用户的配置信息。
• ClientConfig：生成rest.Config的中间层，支持通过ConfigOverrides覆盖当前上下文等配置项。
• rest.Config：K8s客户端（clientset）使用的最终配置，包含集群地址、认证凭据、超时等核心参数。

3. 用法示例 #

# 切换到指定上下文执行操作
go run main.go -context my-gke-cluster

# 匹配所有名称包含gke的集群对应的上下文
go run main.go -cluster-regex "gke.*"

内容的提问来源于stack exchange，提问作者Matanya