84 lines
2.5 KiB
Go
84 lines
2.5 KiB
Go
// Copyright 2021-2022 the Pinniped contributors. All Rights Reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package deploymentref
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
appsv1 "k8s.io/api/apps/v1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
|
|
"go.pinniped.dev/internal/downward"
|
|
"go.pinniped.dev/internal/kubeclient"
|
|
"go.pinniped.dev/internal/ownerref"
|
|
)
|
|
|
|
// getTempClient is stubbed out for testing.
|
|
//
|
|
// We would normally pass a kubernetes.Interface into New(), but the client we want to create in
|
|
// the calling code depends on the return value of New() (i.e., on the kubeclient.Option for the
|
|
// OwnerReference).
|
|
var getTempClient = func() (kubernetes.Interface, error) { //nolint:gochecknoglobals
|
|
client, err := kubeclient.New()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return client.Kubernetes, nil
|
|
}
|
|
|
|
func New(podInfo *downward.PodInfo) (kubeclient.Option, *appsv1.Deployment, *corev1.Pod, error) {
|
|
tempClient, err := getTempClient()
|
|
if err != nil {
|
|
return nil, nil, nil, fmt.Errorf("cannot create temp client: %w", err)
|
|
}
|
|
|
|
deployment, pod, err := getDeploymentAndPod(tempClient, podInfo)
|
|
if err != nil {
|
|
return nil, nil, nil, fmt.Errorf("cannot get deployment: %w", err)
|
|
}
|
|
|
|
// work around stupid behavior of WithoutVersionDecoder.Decode
|
|
deployment.APIVersion, deployment.Kind = appsv1.SchemeGroupVersion.WithKind("Deployment").ToAPIVersionAndKind()
|
|
|
|
return kubeclient.WithMiddleware(ownerref.New(deployment)), deployment, pod, nil
|
|
}
|
|
|
|
func getDeploymentAndPod(kubeClient kubernetes.Interface, podInfo *downward.PodInfo) (*appsv1.Deployment, *corev1.Pod, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
|
|
ns := podInfo.Namespace
|
|
|
|
pod, err := kubeClient.CoreV1().Pods(ns).Get(ctx, podInfo.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("could not get pod: %w", err)
|
|
}
|
|
|
|
podOwner := metav1.GetControllerOf(pod)
|
|
if podOwner == nil {
|
|
return nil, nil, fmt.Errorf("pod %s/%s is missing owner", ns, podInfo.Name)
|
|
}
|
|
|
|
rs, err := kubeClient.AppsV1().ReplicaSets(ns).Get(ctx, podOwner.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("could not get replicaset: %w", err)
|
|
}
|
|
|
|
rsOwner := metav1.GetControllerOf(rs)
|
|
if rsOwner == nil {
|
|
return nil, nil, fmt.Errorf("replicaset %s/%s is missing owner", ns, podInfo.Name)
|
|
}
|
|
|
|
d, err := kubeClient.AppsV1().Deployments(ns).Get(ctx, rsOwner.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("could not get deployment: %w", err)
|
|
}
|
|
|
|
return d, pod, nil
|
|
}
|