add tolerations

Signed-off-by: Oleg Vasilev <[email protected]>
neondatabase · Aug 29, 2024 · 837f32e · 837f32e
1 parent 577759d
commit 837f32e
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/neonvm/controllers/vm_controller.go b/neonvm/controllers/vm_controller.go
@@ -1373,6 +1373,20 @@ func podSpec(
 		return nil, fmt.Errorf("marshal VM Status: %w", err)
 	}
 
+	tolerations := append([]corev1.Toleration{}, vm.Spec.Tolerations...)
+	tolerations = append(tolerations,
+		corev1.Toleration{
+			Key:               "node.kubernetes.io/not-ready",
+			TolerationSeconds: lo.ToPtr(int64(30)),
+			Effect:            "NoExecute",
+		},
+		corev1.Toleration{
+			Key:               "node.kubernetes.io/unreachable",
+			TolerationSeconds: lo.ToPtr(int64(30)),
+			Effect:            "NoExecute",
+		},
+	)
+
 	pod := &corev1.Pod{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:        vm.Status.PodName,
@@ -1387,7 +1401,7 @@ func podSpec(
 			TerminationGracePeriodSeconds: vm.Spec.TerminationGracePeriodSeconds,
 			NodeSelector:                  vm.Spec.NodeSelector,
 			ImagePullSecrets:              vm.Spec.ImagePullSecrets,
-			Tolerations:                   vm.Spec.Tolerations,
+			Tolerations:                   tolerations,
 			ServiceAccountName:            vm.Spec.ServiceAccountName,
 			SchedulerName:                 vm.Spec.SchedulerName,
 			Affinity:                      affinity,