如何通过编程方式获取StatefulSet Pod的DNS名称

嘿，这个场景我经常碰到！既然你已经用client-go创建了StatefulSet，那获取Pod的DNS名称其实很直接，有两种可靠的方式，我给你详细说下：

方法一：查询实际运行的Pod并构造DNS名称（推荐） #

这种方法会直接从K8s API拉取当前属于该StatefulSet的所有运行中Pod，然后根据Headless Service的名称拼接出标准的DNS地址，准确性最高，毕竟能过滤掉未启动或异常的Pod。

步骤&代码示例 #

1. 先获取目标StatefulSet的详情，拿到它关联的Headless Service名称（StatefulSet的spec.serviceName字段是必填项，用来绑定Headless Service）
2. 用StatefulSet的标签选择器过滤出所有属于它的Pod
3. 逐个拼接Pod的DNS名称

import (
    "context"
    "fmt"
    "log"

    "k8s.io/client-go/kubernetes"
    appsv1 "k8s.io/api/apps/v1"
    corev1 "k8s.io/api/core/v1"
    metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
    "k8s.io/apimachinery/pkg/labels"
)

func getStatefulSetPodDNSNames(clientset *kubernetes.Clientset, stsName, namespace string) ([]string, error) {
    // 1. 获取StatefulSet对象
    sts, err := clientset.AppsV1().StatefulSets(namespace).Get(context.TODO(), stsName, metav1.GetOptions{})
    if err != nil {
        return nil, fmt.Errorf("failed to get statefulset: %w", err)
    }
    serviceName := sts.Spec.ServiceName
    if serviceName == "" {
        return nil, fmt.Errorf("statefulset %s has no serviceName configured", stsName)
    }

    // 2. 筛选属于该StatefulSet的Pod
    labelSelector := labels.SelectorFromSet(sts.Spec.Selector.MatchLabels).String()
    podList, err := clientset.CoreV1().Pods(namespace).List(context.TODO(), metav1.ListOptions{
        LabelSelector: labelSelector,
    })
    if err != nil {
        return nil, fmt.Errorf("failed to list pods: %w", err)
    }

    // 3. 生成每个Pod的DNS名称
    var dnsNames []string
    for _, pod := range podList.Items {
        // 标准格式：{podName}.{serviceName}.{namespace}.svc.cluster.local
        dnsName := fmt.Sprintf("%s.%s.%s.svc.cluster.local", pod.Name, serviceName, namespace)
        dnsNames = append(dnsNames, dnsName)
    }

    return dnsNames, nil
}

方法二：根据StatefulSet规则生成DNS名称（适合确定副本都正常运行的场景） #

StatefulSet的Pod名称是严格按照{statefulset-name}-{序号}的规则生成的（从0开始），如果你确定所有副本都已正常启动，也可以直接根据副本数生成DNS名称，省去查询Pod的步骤：

func generateStatefulSetPodDNSNames(clientset *kubernetes.Clientset, stsName, namespace string) ([]string, error) {
    sts, err := clientset.AppsV1().StatefulSets(namespace).Get(context.TODO(), stsName, metav1.GetOptions{})
    if err != nil {
        return nil, fmt.Errorf("failed to get statefulset: %w", err)
    }
    serviceName := sts.Spec.ServiceName
    if serviceName == "" {
        return nil, fmt.Errorf("statefulset %s has no serviceName configured", stsName)
    }

    replicas := sts.Spec.Replicas
    if replicas == nil {
        replicas = new(int32)
        *replicas = 1
    }

    var dnsNames []string
    for i := 0; i < int(*replicas); i++ {
        podName := fmt.Sprintf("%s-%d", stsName, i)
        dnsName := fmt.Sprintf("%s.%s.%s.svc.cluster.local", podName, serviceName, namespace)
        dnsNames = append(dnsNames, dnsName)
    }

    return dnsNames, nil
}

注意事项 #

• 一定要用StatefulSet的spec.serviceName来拼接，而不是直接用StatefulSet名称，因为Headless Service的名称可以自定义，不一定和StatefulSet同名
• 方法一更可靠，因为它只返回实际存在的Pod的DNS，方法二可能会生成对应未启动Pod的无效地址

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