From daf4e3acc0f2b85b02e3900b01bbbf61f432e85a Mon Sep 17 00:00:00 2001
From: Sergey Dudoladov <sergey.dudoladov@zalando.de>
Date: Fri, 5 Apr 2019 12:08:52 +0200
Subject: [PATCH] add basic affinity setup

---
 pkg/cluster/k8sres.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/pkg/cluster/k8sres.go b/pkg/cluster/k8sres.go
index 0a47d84a2..c9b7e1897 100644
--- a/pkg/cluster/k8sres.go
+++ b/pkg/cluster/k8sres.go
@@ -1279,6 +1279,25 @@ func (c *Cluster) generateLogicalBackupJob() (*batchv1beta1.CronJob, error) {
 		[]v1.VolumeMount{},
 	)
 
+	labels := map[string]string{
+		"version":    c.Name,
+		"spilo-role": "replica",
+	}
+	podAffinityTerm := v1.PodAffinityTerm{
+		LabelSelector: &metav1.LabelSelector{
+			MatchLabels: labels,
+		},
+		TopologyKey: "kubernetes.io/hostname",
+	}
+	podAffinity := v1.Affinity{
+		PodAffinity: &v1.PodAffinity{
+			PreferredDuringSchedulingIgnoredDuringExecution: []v1.WeightedPodAffinityTerm{{
+				Weight:          1,
+				PodAffinityTerm: podAffinityTerm,
+			},
+			},
+		}}
+
 	if podTemplate, err = generatePodTemplate(
 		c.Namespace,
 		c.labelsSet(true),
@@ -1300,6 +1319,7 @@ func (c *Cluster) generateLogicalBackupJob() (*batchv1beta1.CronJob, error) {
 	// pods of k8s jobs support only "OnFailure" or "Never"
 	// but the default is "Always"
 	podTemplate.Spec.RestartPolicy = "OnFailure"
+	podTemplate.Spec.Affinity = &podAffinity
 
 	jobSpec := batchv1.JobSpec{Template: *podTemplate}