/
pod_disruption_budget.go
51 lines (41 loc) · 1.51 KB
/
pod_disruption_budget.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package resource
import (
"fmt"
"github.com/itayankri/valhalla-operator/internal/status"
policyv1 "k8s.io/api/policy/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
)
type PodDisruptionBudgetBuilder struct {
*ValhallaResourceBuilder
}
func (builder *ValhallaResourceBuilder) PodDisruptionBudget() *PodDisruptionBudgetBuilder {
return &PodDisruptionBudgetBuilder{builder}
}
func (builder *PodDisruptionBudgetBuilder) Build() (client.Object, error) {
return &policyv1.PodDisruptionBudget{
ObjectMeta: metav1.ObjectMeta{
Name: builder.Instance.ChildResourceName(HorizontalPodAutoscalerSuffix),
Namespace: builder.Instance.Namespace,
},
}, nil
}
func (builder *PodDisruptionBudgetBuilder) Update(object client.Object) error {
name := builder.Instance.ChildResourceName(PodDisruptionBudgetSuffix)
pdb := object.(*policyv1.PodDisruptionBudget)
pdb.Spec.MinAvailable = builder.Instance.Spec.GetMinAvailable()
pdb.Spec.Selector = &metav1.LabelSelector{
MatchLabels: map[string]string{
"app": name,
},
}
if err := controllerutil.SetControllerReference(builder.Instance, pdb, builder.Scheme); err != nil {
return fmt.Errorf("failed setting controller reference: %v", err)
}
return nil
}
func (*PodDisruptionBudgetBuilder) ShouldDeploy(resources []runtime.Object) bool {
return status.IsPersistentVolumeClaimBound(resources) && status.IsJobCompleted(resources)
}