重构Golang获取K8s资源函数 应使用何种参数与返回值类型？

你可以通过Go 1.18及以上版本支持的泛型特性实现类型安全的通用函数，无需额外定义枚举类型的resourceType参数，编译期即可完成类型校验，不需要运行时类型断言。

完整实现代码 #

import (
    "context"
    corev1 "k8s.io/api/core/v1"
    metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
    "k8s.io/client-go/kubernetes"
)

// 以下为你的原有全局变量，可根据实际业务调整为函数参数传入
var (
    namespace string
    clientset *kubernetes.Clientset
)

// 原有ListOptions生成逻辑保留
func getListOption(name string) metav1.ListOptions {
    return metav1.ListOptions{
        // 你的原有筛选逻辑，示例为按名称匹配
        FieldSelector: "metadata.name=" + name,
    }
}

// 泛型约束：适配所有client-go返回的资源列表结构
type K8sResourceList[T any] interface {
    GetItems() []T
}

// 通用资源获取函数
// 泛型参数说明：
// T: 目标资源类型，例如 corev1.Pod、corev1.Service
// L: 对应资源的列表返回类型，例如 corev1.PodList、corev1.ServiceList
func getResource[T any, L K8sResourceList[T]](name string, listExecutor func(ctx context.Context, opts metav1.ListOptions) (L, error)) []T {
    listResult, err := listExecutor(context.TODO(), getListOption(name))
    if err != nil {
        panic(err.Error())
    }
    return listResult.GetItems()
}

调用示例 #

// 获取指定名称的Pod列表
pods := getResource("test-pod", func(ctx context.Context, opts metav1.ListOptions) (*corev1.PodList, error) {
    return clientset.CoreV1().Pods(namespace).List(ctx, opts)
})

// 获取指定名称的Service列表
services := getResource("test-svc", func(ctx context.Context, opts metav1.ListOptions) (*corev1.ServiceList, error) {
    return clientset.CoreV1().Services(namespace).List(ctx, opts)
})

低版本Go兼容方案（无泛型） #

如果使用的Go版本低于1.18不支持泛型，可以返回空接口，调用方自行做类型断言：

func getResource(name string, listExecutor func(ctx context.Context, opts metav1.ListOptions) (interface{}, error)) interface{} {
    listResult, err := listExecutor(context.TODO(), getListOption(name))
    if err != nil {
        panic(err.Error())
    }
    return listResult
}

// 调用示例
pods := getResource("test-pod", func(ctx context.Context, opts metav1.ListOptions) (interface{}, error) {
    return clientset.CoreV1().Pods(namespace).List(ctx, opts)
}).(*corev1.PodList).Items

优化建议 #

生产环境建议将函数内的panic改为返回error，符合Go的错误处理规范，避免程序直接崩溃。

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