feat(controller): Support nodeSelector

Add nodeSelector key to API spec to control NodeSelector set on statefulset.
hyperspike · Feb 13, 2025 · 5f01063 · 5f01063
1 parent 5d5e4ca
commit 5f01063
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/api/v1/valkey_types.go b/api/v1/valkey_types.go
@@ -82,6 +82,9 @@ type ValkeySpec struct {
 
 	// Service Password
 	ServicePassword *corev1.SecretKeySelector `json:"servicePassword,omitempty"`
+
+	// Node Selector
+	NodeSelector map[string]string `json:"nodeSelector,omitempty"`
 }
 
 // ExternalAccess defines the external access configuration

diff --git a/api/v1/zz_generated.deepcopy.go b/api/v1/zz_generated.deepcopy.go
diff --git a/config/crd/bases/hyperspike.io_valkeys.yaml b/config/crd/bases/hyperspike.io_valkeys.yaml
@@ -211,6 +211,11 @@ spec:
               image:
                 description: Image to use
                 type: string
+              nodeSelector:
+                additionalProperties:
+                  type: string
+                description: Node Selector
+                type: object
               nodes:
                 default: 3
                 description: Number of shards

diff --git a/internal/controller/valkey_controller.go b/internal/controller/valkey_controller.go
@@ -2102,6 +2102,7 @@ func (r *ValkeyReconciler) upsertStatefulSet(ctx context.Context, valkey *hyperv
 							},
 						},
 					},
+					NodeSelector: valkey.Spec.NodeSelector,
 					Containers: []corev1.Container{
 						{
 							Image: image,
-Original file line number
+Diff line change
@@ Expand Up @@
     							},
     						},
     					},
+    					NodeSelector: valkey.Spec.NodeSelector,
     					Containers: []corev1.Container{
     						{
     							Image: image,
@@ Expand Down @@