如何扩展Kubernetes client-go以整合TrafficSplit自定义资源？

我正在编写一个监控Kubernetes Service对象的控制器：当Service包含特定标签时，会创建TrafficSplit资源。由于原生Kubernetes Go客户端不支持TrafficSplit对象，我目前通过引入SMI SDK的客户端来实现，但现在想把TrafficSplit的操作能力整合到原生kubeClient对象中，而非维护独立的客户端实例。

当前实现代码如下：

import (
    splitClientV1alpha1 "github.com/servicemeshinterface/smi-sdk-go/pkg/gen/client/split/clientset/versioned/typed/split/v1alpha1"
    "k8s.io/client-go/kubernetes"
    "path/filepath"
    "github.com/mitchellh/go-homedir"
    "k8s.io/client-go/tools/clientcmd"
    "fmt"
)

// 获取本地kube配置文件
kubehome := filepath.Join(homedir.HomeDir(), ".kube", "config")
// 从文件构建配置
kubeConfig, err = clientcmd.BuildConfigFromFlags("", kubehome)
if err != nil {
    return fmt.Errorf("error loading kubernetes configuration: %w", err)
}
// 创建原生Kubernetes客户端对象
kubeClient, err := kubernetes.NewForConfig(kubeConfig)
if err != nil {
    return fmt.Errorf("error creating kubernetes client: %w", err)
}
// 创建专用于自定义资源的客户端集
splitClient, err := splitClientV1alpha1.NewForConfig(kubeConfig)
if err != nil {
    return fmt.Errorf("error creating split client: %s", err)
}

请问是否有方法可以将TrafficSplit的schema扩展到kubeClient对象中，实现统一的客户端调用？

当然可以实现！原生kubernetes.Clientset是固定的结构体，无法直接修改扩展，但我们可以通过两种优雅的方式将TrafficSplit的操作能力整合到统一的客户端实例中：

方法一：组合客户端结构体（推荐，快速实现） #

最简便的方式是封装一个自定义结构体，将原生K8s客户端和TrafficSplit客户端组合在一起，这样在代码中只需要传递一个对象即可操作所有资源：

import (
    splitClientV1alpha1 "github.com/servicemeshinterface/smi-sdk-go/pkg/gen/client/split/clientset/versioned/typed/split/v1alpha1"
    "k8s.io/client-go/kubernetes"
    "k8s.io/client-go/rest"
    "fmt"
)

// CustomKubeClient 整合原生K8s客户端与TrafficSplit客户端
type CustomKubeClient struct {
    // 嵌入原生Clientset，继承所有原生资源操作方法
    *kubernetes.Clientset
    // 添加TrafficSplit的操作接口
    TrafficSplit splitClientV1alpha1.SplitV1alpha1Interface
}

// NewCustomKubeClient 初始化整合后的客户端
func NewCustomKubeClient(config *rest.Config) (*CustomKubeClient, error) {
    // 创建原生客户端
    kubeClient, err := kubernetes.NewForConfig(config)
    if err != nil {
        return nil, fmt.Errorf("error creating kubernetes client: %w", err)
    }
    // 创建TrafficSplit客户端
    splitClient, err := splitClientV1alpha1.NewForConfig(config)
    if err != nil {
        return nil, fmt.Errorf("error creating split client: %w", err)
    }
    // 返回整合后的客户端
    return &CustomKubeClient{
        Clientset:    kubeClient,
        TrafficSplit: splitClient,
    }, nil
}

使用方式 #

初始化后，你可以像操作原生资源一样调用TrafficSplit的方法，完全不需要维护两个独立客户端：

// 初始化整合客户端
customClient, err := NewCustomKubeClient(kubeConfig)
if err != nil {
    // 处理错误
}

// 操作原生Service资源
service, err := customClient.CoreV1().Services("default").Get(context.TODO(), "my-service", metav1.GetOptions{})
// 操作TrafficSplit资源
trafficSplit, err := customClient.TrafficSplit.TrafficSplits("default").Create(context.TODO(), &splitv1alpha1.TrafficSplit{}, metav1.CreateOptions{})

方法二：使用client-gen生成整合的Clientset（适合长期维护场景） #

如果你的项目需要更贴近原生Clientset的调用体验，或者需要整合更多自定义资源，可以使用Kubernetes的client-gen工具生成一个包含原生资源和TrafficSplit的自定义Clientset：

1. 编写client-gen配置文件：在项目中创建client-gen的配置（比如client-gen.yaml），指定要包含的原生API组（如core, apps）和TrafficSplit的API组（split.smi-spec.io/v1alpha1）。
2. 运行client-gen生成代码：通过client-gen工具根据配置生成整合后的Clientset代码，生成的客户端会同时支持原生K8s资源和TrafficSplit的操作。

这种方式需要熟悉client-gen的使用规则，适合复杂的长期项目，但能提供最原生的客户端调用体验。

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