operator: add node selector

Signed-off-by: Prashanth Dintyala <saiprashanth173@gmail.com>
NVIDIA · Apr 12, 2021 · fef8bc0 · fef8bc0
1 parent 31a3111
commit fef8bc0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/operator/pkg/resources/proxy/statefulset.go b/operator/pkg/resources/proxy/statefulset.go
@@ -132,6 +132,7 @@ func proxyPodSpec(ais *aisv1.AIStore) corev1.PodSpec {
 			},
 		},
 		Affinity:           cmn.NewAISPodAffinity(ais, ais.Spec.ProxySpec.Affinity, PodLabels(ais)),
+		NodeSelector:       ais.Spec.ProxySpec.NodeSelector,
 		ServiceAccountName: cmn.ServiceAccountName(ais),
 		SecurityContext:    ais.Spec.ProxySpec.SecurityContext,
 		Volumes:            cmn.NewAISVolumes(ais, aiscmn.Proxy),

diff --git a/operator/pkg/resources/target/statefulset.go b/operator/pkg/resources/target/statefulset.go
@@ -128,6 +128,7 @@ func NewTargetSS(ais *aisv1.AIStore) *apiv1.StatefulSet {
 					ServiceAccountName: cmn.ServiceAccountName(ais),
 					SecurityContext:    ais.Spec.TargetSpec.SecurityContext,
 					Affinity:           cmn.NewAISPodAffinity(ais, ais.Spec.TargetSpec.Affinity, ls),
+					NodeSelector:       ais.Spec.TargetSpec.NodeSelector,
 					Volumes:            cmn.NewAISVolumes(ais, aiscmn.Target),
 					Tolerations:        ais.Spec.TargetSpec.Tolerations,
 				},