c6c2c525a6
Also fix some tests that were broken by bumping golang and dependencies in the previous commits. Note that in addition to changes made to satisfy the linter which do not impact the behavior of the code, this commit also adds ReadHeaderTimeout to all usages of http.Server to satisfy the linter (and because it seemed like a good suggestion).
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
|
|
}
|