mmelko commented on a change in pull request #1787:
URL: https://github.com/apache/camel-k/pull/1787#discussion_r512219029
##########
File path: pkg/trait/pdb.go
##########
@@ -0,0 +1,103 @@
+package trait
+
+import (
+ v1 "github.com/apache/camel-k/pkg/apis/camel/v1"
+ "github.com/google/go-cmp/cmp"
+ "k8s.io/api/policy/v1beta1"
+ metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+ "k8s.io/apimachinery/pkg/util/intstr"
+)
+
+type pdbTrait struct {
+ BaseTrait `property:",squash"`
+ MaxUnavailable string `property:"max-unavailable"
json:"maxUnavailable,omitempty"`
+ MinAvailable string `property:"min-available"
json:"minAvailable,omitempty"`
+}
+
+func newPdbTrait() Trait {
+ return &pdbTrait{
+ BaseTrait: NewBaseTrait("pdb", 900),
+ }
+}
+
+func (t *pdbTrait) Configure(e *Environment) (bool, error) {
+ return e.IntegrationInPhase(
+ v1.IntegrationPhaseRunning,
+ ), nil
+}
+
+func (t *pdbTrait) Apply(e *Environment) error {
+ pdbName := "pdb-" + e.Integration.Name + "-it"
+ var pdb *v1beta1.PodDisruptionBudget
+ var err error
+
+ if pdb, err =
t.Client.PolicyV1beta1().PodDisruptionBudgets(e.DetermineNamespace()).Get(pdbName,
metav1.GetOptions{}); err == nil {
+ if a := t.havePdbConfChanged(pdb); a {
+ err =
t.Client.PolicyV1beta1().PodDisruptionBudgets(e.DetermineNamespace()).Delete(pdbName,
&metav1.DeleteOptions{})
+ updated := t.generatePodDisruptionBudget(pdbName,
e.Integration)
+ if _, err =
t.Client.PolicyV1beta1().PodDisruptionBudgets(e.DetermineNamespace()).Create(updated);
err == nil {
+ t.L.Info("PodDisruptionBudget updated: ",
pdb.Spec)
+ }
+ }
+
+ } else {
+ t.L.Info("PodDisruptionBudget for the integration",
e.Integration.Name, "not found")
+ pdb = t.generatePodDisruptionBudget(pdbName, e.Integration)
+ if _, err =
t.Client.PolicyV1beta1().PodDisruptionBudgets(e.DetermineNamespace()).Create(pdb);
err == nil {
+ t.L.Error(err, "error during creating pdb", err)
+ }
+ }
+ return err
+}
+
+// check if configuration for pdb has changed, if yes update values in pdbtrait
+func (t *pdbTrait) havePdbConfChanged(pdb *v1beta1.PodDisruptionBudget) bool {
+ return !cmp.Equal(t.MaxUnavailable, pdb.Spec.MaxUnavailable) ||
!cmp.Equal(t.MinAvailable, pdb.Spec.MinAvailable)
+}
+func (t *pdbTrait) generatePodDisruptionBudget(pdbName string, integration
*v1.Integration) *v1beta1.PodDisruptionBudget {
+ controller := true
+ blockOwnerDeletion := true
+ spec := v1beta1.PodDisruptionBudgetSpec{
+ Selector: &metav1.LabelSelector{
+ MatchLabels: map[string]string{
+ "camel.apache.org/integration":
integration.Name,
+ },
+ },
+ }
+ var min, max intstr.IntOrString
+
+ if t.MinAvailable == "" && t.MaxUnavailable == "" {
+ //set the default
+ max = intstr.Parse("1")
+ spec.MaxUnavailable = &max
+ } else if t.MinAvailable != "" {
+ // sets to maxUnavailable to nil because only one parameter is
allowed to be set
+ min = intstr.Parse(t.MinAvailable)
+ spec.MinAvailable = &min
+ spec.MaxUnavailable = nil
+ t.MaxUnavailable = ""
+ } else {
+ //set maxAvailable
+ max = intstr.Parse(t.MaxUnavailable)
+ spec.MaxUnavailable = &max
+ }
+
+ return &v1beta1.PodDisruptionBudget{
+ ObjectMeta: metav1.ObjectMeta{
+ Name: pdbName,
+ Namespace: integration.Namespace,
+ Labels: integration.Labels,
+ OwnerReferences: []metav1.OwnerReference{
Review comment:
The problem was that I it was done according the labels .. but when I
looked there weren't any integration labels. I have to doublecheck
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
