Kube cluster upgrade

pkg/cluster/cluster.go

@@ -16,6 +16,7 @@ import (
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/client-go/pkg/api/v1"
 	"k8s.io/client-go/pkg/apis/apps/v1beta1"
+	policybeta1 "k8s.io/client-go/pkg/apis/policy/v1beta1"
 	"k8s.io/client-go/rest"
 	"k8s.io/client-go/tools/cache"
 
@@ -44,10 +45,11 @@ type Config struct {
 }
 
 type kubeResources struct {
-	Services    map[PostgresRole]*v1.Service
-	Endpoint    *v1.Endpoints
-	Secrets     map[types.UID]*v1.Secret
-	Statefulset *v1beta1.StatefulSet
+	Services            map[PostgresRole]*v1.Service
+	Endpoint            *v1.Endpoints
+	Secrets             map[types.UID]*v1.Secret
+	Statefulset         *v1beta1.StatefulSet
+	PodDisruptionBudget *policybeta1.PodDisruptionBudget
 	//Pods are treated separately
 	//PVCs are treated separately
 }
@@ -259,6 +261,12 @@ func (c *Cluster) Create() error {
 		}
 	}
 
+	pdb, err := c.createPodDisruptionBudget()
+	if err != nil {
+		return fmt.Errorf("could not create pod disruption budget: %v", err)
+	}
+	c.logger.Infof("pod disruption budget %q has been successfully created", util.NameFromMeta(pdb.ObjectMeta))
+
 	err = c.listResources()
 	if err != nil {
 		c.logger.Errorf("could not list resources: %v", err)
@@ -334,6 +342,12 @@ func (c *Cluster) compareStatefulSetWith(statefulSet *v1beta1.StatefulSet) *comp
 		needsRollUpdate = true
 		reasons = append(reasons, "new statefulset's terminationGracePeriodSeconds  doesn't match the current one")
 	}
+	if !reflect.DeepEqual(c.Statefulset.Spec.Template.Spec.Affinity, statefulSet.Spec.Template.Spec.Affinity) {
+		needsReplace = true
+		needsRollUpdate = true
+		reasons = append(reasons, "new statefulset's pod affinity doesn't match the current one")
+	}
+
 	// Some generated fields like creationTimestamp make it not possible to use DeepCompare on Spec.Template.ObjectMeta
 	if !reflect.DeepEqual(c.Statefulset.Spec.Template.Labels, statefulSet.Spec.Template.Labels) {
 		needsReplace = true
@@ -522,6 +536,15 @@ func (c *Cluster) Update(newSpec *spec.Postgresql) error {
 		c.logger.Infof("volumes have been updated successfully")
 	}
 
+	newPDB := c.generatePodDisruptionBudget()
+	if match, reason := c.samePDBWith(newPDB); !match {
+		c.logPDBChanges(c.PodDisruptionBudget, newPDB, true, reason)
+		if err := c.updatePodDisruptionBudget(newPDB); err != nil {
+			c.setStatus(spec.ClusterStatusUpdateFailed)
+			return fmt.Errorf("could not update pod disruption budget: %v", err)
+		}
+	}
+
 	c.setStatus(spec.ClusterStatusRunning)
 
 	return nil
@@ -555,6 +578,10 @@ func (c *Cluster) Delete() error {
 		}
 	}
 
+	if err := c.deletePodDisruptionBudget(); err != nil {
+		return fmt.Errorf("could not delete pod disruption budget: %v", err)
+	}
+
 	return nil
 }
 
@@ -690,11 +717,12 @@ func (c *Cluster) GetStatus() *spec.ClusterStatus {
 		Status:  c.Status,
 		Spec:    c.Spec,
 
-		MasterService:  c.GetServiceMaster(),
-		ReplicaService: c.GetServiceReplica(),
-		Endpoint:       c.GetEndpoint(),
-		StatefulSet:    c.GetStatefulSet(),
-		CurrentProcess: c.GetCurrentProcess(),
+		MasterService:       c.GetServiceMaster(),
+		ReplicaService:      c.GetServiceReplica(),
+		Endpoint:            c.GetEndpoint(),
+		StatefulSet:         c.GetStatefulSet(),
+		PodDisruptionBudget: c.GetPodDisruptionBudget(),
+		CurrentProcess:      c.GetCurrentProcess(),
 
 		Error: c.Error,
 	}
@@ -733,3 +761,13 @@ func (c *Cluster) ManualFailover(curMaster *v1.Pod, candidate spec.NamespacedNam
 
 	return nil
 }
+
+// Lock locks the cluster
+func (c *Cluster) Lock() {
+	c.mu.Lock()
+}
+
+// Unlock unlocks the cluster
+func (c *Cluster) Unlock() {
+	c.mu.Unlock()
+}