Prevent scheduling on arm64 nodes

replicatedhq · Aug 10, 2020 · e8287f0 · e8287f0
1 parent 4fbb722
commit e8287f0
Show file tree

Hide file tree

Showing 6 changed files with 51 additions and 0 deletions.
diff --git a/pkg/kotsadm/affinity.go b/pkg/kotsadm/affinity.go
@@ -0,0 +1,36 @@
+package kotsadm
+
+import (
+	corev1 "k8s.io/api/core/v1"
+)
+
+func defaultKotsNodeAffinity() *corev1.NodeAffinity {
+	return &corev1.NodeAffinity{
+		RequiredDuringSchedulingIgnoredDuringExecution: &corev1.NodeSelector{
+			NodeSelectorTerms: []corev1.NodeSelectorTerm{
+				{
+					MatchExpressions: []corev1.NodeSelectorRequirement{
+						{
+							Key:      "kubernetes.io/os",
+							Operator: corev1.NodeSelectorOpIn,
+							Values: []string{
+								"linux",
+							},
+						},
+					},
+				},
+				{
+					MatchExpressions: []corev1.NodeSelectorRequirement{
+						{
+							Key:      "kubernetes.io/arch",
+							Operator: corev1.NodeSelectorOpNotIn,
+							Values: []string{
+								"arm64",
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+}
diff --git a/pkg/kotsadm/api_objects.go b/pkg/kotsadm/api_objects.go
@@ -345,6 +345,9 @@ func apiDeployment(deployOptions types.DeployOptions) *appsv1.Deployment {
 					}),
 				},
 				Spec: corev1.PodSpec{
+					Affinity: &corev1.Affinity{
+						NodeAffinity: defaultKotsNodeAffinity(),
+					},
 					SecurityContext:    &securityContext,
 					ServiceAccountName: "kotsadm-api",
 					RestartPolicy:      corev1.RestartPolicyAlways,

diff --git a/pkg/kotsadm/kotsadm_objects.go b/pkg/kotsadm/kotsadm_objects.go
@@ -337,6 +337,9 @@ func kotsadmDeployment(deployOptions types.DeployOptions) *appsv1.Deployment {
 					},
 				},
 				Spec: corev1.PodSpec{
+					Affinity: &corev1.Affinity{
+						NodeAffinity: defaultKotsNodeAffinity(),
+					},
 					SecurityContext: &securityContext,
 					Volumes: []corev1.Volume{
 						{

diff --git a/pkg/kotsadm/operator_objects.go b/pkg/kotsadm/operator_objects.go
@@ -230,6 +230,9 @@ func operatorDeployment(deployOptions types.DeployOptions) *appsv1.Deployment {
 					}),
 				},
 				Spec: corev1.PodSpec{
+					Affinity: &corev1.Affinity{
+						NodeAffinity: defaultKotsNodeAffinity(),
+					},
 					SecurityContext:    &securityContext,
 					ServiceAccountName: "kotsadm-operator",
 					RestartPolicy:      corev1.RestartPolicyAlways,

diff --git a/pkg/kotsadm/restore_job.go b/pkg/kotsadm/restore_job.go
@@ -46,6 +46,9 @@ func restoreJob(backupName string, namespace string, isOpenShift bool, kotsadmOp
 					}),
 				},
 				Spec: corev1.PodSpec{
+					Affinity: &corev1.Affinity{
+						NodeAffinity: defaultKotsNodeAffinity(),
+					},
 					SecurityContext:    &securityContext,
 					RestartPolicy:      corev1.RestartPolicyNever,
 					ServiceAccountName: "kotsadm",

diff --git a/pkg/kotsadm/schemahero_objects.go b/pkg/kotsadm/schemahero_objects.go
@@ -41,6 +41,9 @@ func migrationsPod(deployOptions types.DeployOptions) *corev1.Pod {
 			Labels:    types.GetKotsadmLabels(),
 		},
 		Spec: corev1.PodSpec{
+			Affinity: &corev1.Affinity{
+				NodeAffinity: defaultKotsNodeAffinity(),
+			},
 			SecurityContext:  &securityContext,
 			RestartPolicy:    corev1.RestartPolicyOnFailure,
 			ImagePullSecrets: pullSecrets,