github/Reloader
1
0
Fork 0
mirror of https://github.com/stakater/Reloader.git synced 2026-08-27 14:37:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Introduce a Generic ResourceReconciler and a generic BaseWorkload to de-duplicate a lot of code

Browse Source
This commit is contained in:
TheiLLeniumStudios
2026-01-05 00:59:57 +01:00
parent c785067a44
commit 5548ce559a
31 changed files with 1491 additions and 1356 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/pkg/workload/deployment.go
+21 -99
View File
@@ -1,124 +1,46 @@
package workload
import (
"context"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
)
// deploymentAccessor implements PodTemplateAccessor for Deployment.
type deploymentAccessor struct {
deployment *appsv1.Deployment
}
func (a *deploymentAccessor) GetPodTemplateSpec() *corev1.PodTemplateSpec {
return &a.deployment.Spec.Template
}
func (a *deploymentAccessor) GetObjectMeta() *metav1.ObjectMeta {
return &a.deployment.ObjectMeta
}
// DeploymentWorkload wraps a Kubernetes Deployment.
type DeploymentWorkload struct {
deployment *appsv1.Deployment
original *appsv1.Deployment
*BaseWorkload[*appsv1.Deployment]
}
// NewDeploymentWorkload creates a new DeploymentWorkload.
func NewDeploymentWorkload(d *appsv1.Deployment) *DeploymentWorkload {
original := d.DeepCopy()
accessor := &deploymentAccessor{deployment: d}
return &DeploymentWorkload{
deployment: d,
original: d.DeepCopy(),
BaseWorkload: NewBaseWorkload(d, original, accessor, KindDeployment),
}
}
// Ensure DeploymentWorkload implements WorkloadAccessor.
var _ WorkloadAccessor = (*DeploymentWorkload)(nil)
func (w *DeploymentWorkload) Kind() Kind {
return KindDeployment
}
func (w *DeploymentWorkload) GetObject() client.Object {
return w.deployment
}
func (w *DeploymentWorkload) GetName() string {
return w.deployment.Name
}
func (w *DeploymentWorkload) GetNamespace() string {
return w.deployment.Namespace
}
func (w *DeploymentWorkload) GetAnnotations() map[string]string {
return w.deployment.Annotations
}
func (w *DeploymentWorkload) GetPodTemplateAnnotations() map[string]string {
if w.deployment.Spec.Template.Annotations == nil {
w.deployment.Spec.Template.Annotations = make(map[string]string)
}
return w.deployment.Spec.Template.Annotations
}
func (w *DeploymentWorkload) SetPodTemplateAnnotation(key, value string) {
if w.deployment.Spec.Template.Annotations == nil {
w.deployment.Spec.Template.Annotations = make(map[string]string)
}
w.deployment.Spec.Template.Annotations[key] = value
}
func (w *DeploymentWorkload) GetContainers() []corev1.Container {
return w.deployment.Spec.Template.Spec.Containers
}
func (w *DeploymentWorkload) SetContainers(containers []corev1.Container) {
w.deployment.Spec.Template.Spec.Containers = containers
}
func (w *DeploymentWorkload) GetInitContainers() []corev1.Container {
return w.deployment.Spec.Template.Spec.InitContainers
}
func (w *DeploymentWorkload) SetInitContainers(containers []corev1.Container) {
w.deployment.Spec.Template.Spec.InitContainers = containers
}
func (w *DeploymentWorkload) GetVolumes() []corev1.Volume {
return w.deployment.Spec.Template.Spec.Volumes
}
func (w *DeploymentWorkload) Update(ctx context.Context, c client.Client) error {
return c.Patch(ctx, w.deployment, client.StrategicMergeFrom(w.original), client.FieldOwner(FieldManager))
}
// Ensure DeploymentWorkload implements Workload.
var _ Workload = (*DeploymentWorkload)(nil)
func (w *DeploymentWorkload) DeepCopy() Workload {
return &DeploymentWorkload{
deployment: w.deployment.DeepCopy(),
original: w.original.DeepCopy(),
}
}
func (w *DeploymentWorkload) ResetOriginal() {
w.original = w.deployment.DeepCopy()
}
func (w *DeploymentWorkload) GetEnvFromSources() []corev1.EnvFromSource {
var sources []corev1.EnvFromSource
for _, container := range w.deployment.Spec.Template.Spec.Containers {
sources = append(sources, container.EnvFrom...)
}
for _, container := range w.deployment.Spec.Template.Spec.InitContainers {
sources = append(sources, container.EnvFrom...)
}
return sources
}
func (w *DeploymentWorkload) UsesConfigMap(name string) bool {
return SpecUsesConfigMap(&w.deployment.Spec.Template.Spec, name)
}
func (w *DeploymentWorkload) UsesSecret(name string) bool {
return SpecUsesSecret(&w.deployment.Spec.Template.Spec, name)
}
func (w *DeploymentWorkload) GetOwnerReferences() []metav1.OwnerReference {
return w.deployment.OwnerReferences
return NewDeploymentWorkload(w.Object().DeepCopy())
}
// GetDeployment returns the underlying Deployment for special handling.
func (w *DeploymentWorkload) GetDeployment() *appsv1.Deployment {
return w.deployment
return w.Object()
}