Add NodeSelector

olliewalsh · olliewalsh · commit dcf0a1d32ce6 · 2024-11-20T23:39:16.000Z
diff --git a/api/bases/telemetry.openstack.org_autoscalings.yaml b/api/bases/telemetry.openstack.org_autoscalings.yaml
@@ -103,6 +103,12 @@ spec:
                     items:
                       type: string
                     type: array
+                  nodeSelector:
+                    additionalProperties:
+                      type: string
+                    description: NodeSelector to target subset of worker nodes running
+                      this service
+                    type: object
                   notifierImage:
                     type: string
                   override:
diff --git a/api/bases/telemetry.openstack.org_ceilometers.yaml b/api/bases/telemetry.openstack.org_ceilometers.yaml
@@ -149,6 +149,12 @@ spec:
                 items:
                   type: string
                 type: array
+              nodeSelector:
+                additionalProperties:
+                  type: string
+                description: NodeSelector to target subset of worker nodes running
+                  this service
+                type: object
               notificationImage:
                 type: string
               passwordSelector:
diff --git a/api/bases/telemetry.openstack.org_telemetries.yaml b/api/bases/telemetry.openstack.org_telemetries.yaml
@@ -106,6 +106,12 @@ spec:
                         items:
                           type: string
                         type: array
+                      nodeSelector:
+                        additionalProperties:
+                          type: string
+                        description: NodeSelector to target subset of worker nodes
+                          running this service
+                        type: object
                       notifierImage:
                         type: string
                       override:
@@ -467,6 +473,12 @@ spec:
                     items:
                       type: string
                     type: array
+                  nodeSelector:
+                    additionalProperties:
+                      type: string
+                    description: NodeSelector to target subset of worker nodes running
+                      this service
+                    type: object
                   notificationImage:
                     type: string
                   passwordSelector:
@@ -1823,6 +1835,12 @@ spec:
                         type: string
                     type: object
                 type: object
+              nodeSelector:
+                additionalProperties:
+                  type: string
+                description: NodeSelector to target subset of worker nodes running
+                  this service
+                type: object
             type: object
           status:
             description: TelemetryStatus defines the observed state of Telemetry
diff --git a/api/v1beta1/autoscaling_types.go b/api/v1beta1/autoscaling_types.go
@@ -122,6 +122,10 @@ type AodhCore struct {
 	// +operator-sdk:csv:customresourcedefinitions:type=spec
 	// TLS - Parameters related to the TLS
 	TLS tls.API `json:"tls,omitempty"`
+
+	// +kubebuilder:validation:Optional
+	// NodeSelector to target subset of worker nodes running this service
+	NodeSelector *map[string]string `json:"nodeSelector,omitempty"`
 }
 
 // APIOverrideSpec to override the generated manifest of several child resources.
diff --git a/api/v1beta1/ceilometer_types.go b/api/v1beta1/ceilometer_types.go
@@ -112,6 +112,10 @@ type CeilometerSpecCore struct {
 	// +operator-sdk:csv:customresourcedefinitions:type=spec
 	// KSMTLS - Parameters related to the TLS for kube-state-metrics
 	KSMTLS tls.SimpleService `json:"ksmTls,omitempty"`
+
+	// +kubebuilder:validation:Optional
+	// NodeSelector to target subset of worker nodes running this service
+	NodeSelector *map[string]string `json:"nodeSelector,omitempty"`
 }
 
 // CeilometerStatus defines the observed state of Ceilometer
diff --git a/api/v1beta1/telemetry_types.go b/api/v1beta1/telemetry_types.go
@@ -71,6 +71,10 @@ type TelemetrySpecBase struct {
 	// +kubebuilder:validation:Optional
 	// Logging - Parameters related to the logging
 	Logging LoggingSection `json:"logging,omitempty"`
+
+	// +kubebuilder:validation:Optional
+	// NodeSelector to target subset of worker nodes running this service
+	NodeSelector *map[string]string `json:"nodeSelector,omitempty"`
 }
 
 // CeilometerSection defines the desired state of the ceilometer service
diff --git a/api/v1beta1/zz_generated.deepcopy.go b/api/v1beta1/zz_generated.deepcopy.go
diff --git a/config/crd/bases/telemetry.openstack.org_autoscalings.yaml b/config/crd/bases/telemetry.openstack.org_autoscalings.yaml
@@ -103,6 +103,12 @@ spec:
                     items:
                       type: string
                     type: array
+                  nodeSelector:
+                    additionalProperties:
+                      type: string
+                    description: NodeSelector to target subset of worker nodes running
+                      this service
+                    type: object
                   notifierImage:
                     type: string
                   override:
diff --git a/config/crd/bases/telemetry.openstack.org_ceilometers.yaml b/config/crd/bases/telemetry.openstack.org_ceilometers.yaml
@@ -149,6 +149,12 @@ spec:
                 items:
                   type: string
                 type: array
+              nodeSelector:
+                additionalProperties:
+                  type: string
+                description: NodeSelector to target subset of worker nodes running
+                  this service
+                type: object
               notificationImage:
                 type: string
               passwordSelector:
diff --git a/config/crd/bases/telemetry.openstack.org_telemetries.yaml b/config/crd/bases/telemetry.openstack.org_telemetries.yaml
@@ -106,6 +106,12 @@ spec:
                         items:
                           type: string
                         type: array
+                      nodeSelector:
+                        additionalProperties:
+                          type: string
+                        description: NodeSelector to target subset of worker nodes
+                          running this service
+                        type: object
                       notifierImage:
                         type: string
                       override:
@@ -467,6 +473,12 @@ spec:
                     items:
                       type: string
                     type: array
+                  nodeSelector:
+                    additionalProperties:
+                      type: string
+                    description: NodeSelector to target subset of worker nodes running
+                      this service
+                    type: object
                   notificationImage:
                     type: string
                   passwordSelector:
@@ -1823,6 +1835,12 @@ spec:
                         type: string
                     type: object
                 type: object
+              nodeSelector:
+                additionalProperties:
+                  type: string
+                description: NodeSelector to target subset of worker nodes running
+                  this service
+                type: object
             type: object
           status:
             description: TelemetryStatus defines the observed state of Telemetry
diff --git a/controllers/telemetry_controller.go b/controllers/telemetry_controller.go
@@ -251,6 +251,10 @@ func (r TelemetryReconciler) reconcileCeilometer(ctx context.Context, instance *
 		return ctrl.Result{}, nil
 	}
 
+	if instance.Spec.Ceilometer.NodeSelector == nil {
+		instance.Spec.Ceilometer.NodeSelector = instance.Spec.NodeSelector
+	}
+
 	helper.GetLogger().Info("Reconciling Ceilometer", ceilometerNamespaceLabel, instance.Namespace, ceilometerNameLabel, ceilometer.ServiceName)
 	op, err := controllerutil.CreateOrPatch(ctx, helper.GetClient(), ceilometerInstance, func() error {
 		instance.Spec.Ceilometer.CeilometerSpec.DeepCopyInto(&ceilometerInstance.Spec)
@@ -328,6 +332,10 @@ func (r TelemetryReconciler) reconcileAutoscaling(ctx context.Context, instance
 		return ctrl.Result{}, nil
 	}
 
+	if instance.Spec.Autoscaling.Aodh.NodeSelector == nil {
+		instance.Spec.Autoscaling.Aodh.NodeSelector = instance.Spec.NodeSelector
+	}
+
 	helper.GetLogger().Info("Reconciling Autoscaling", autoscalingNamespaceLabel, instance.Namespace, autoscalingNameLabel, autoscalingName)
 	op, err := controllerutil.CreateOrPatch(ctx, helper.GetClient(), autoscalingInstance, func() error {
 		instance.Spec.Autoscaling.AutoscalingSpec.DeepCopyInto(&autoscalingInstance.Spec)
diff --git a/pkg/autoscaling/aodh_statefulset.go b/pkg/autoscaling/aodh_statefulset.go
@@ -200,6 +200,10 @@ func AodhStatefulSet(
 		},
 	}
 
+	if instance.Spec.Aodh.NodeSelector != nil {
+		pod.Spec.NodeSelector = *instance.Spec.Aodh.NodeSelector
+	}
+
 	statefulset := &appsv1.StatefulSet{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      ServiceName,
diff --git a/pkg/autoscaling/dbsync.go b/pkg/autoscaling/dbsync.go
@@ -92,5 +92,9 @@ func DbSyncJob(instance *autoscalingv1beta1.Autoscaling, labels map[string]strin
 		},
 	}
 
+	if instance.Spec.Aodh.NodeSelector != nil {
+		job.Spec.Template.Spec.NodeSelector = *instance.Spec.Aodh.NodeSelector
+	}
+
 	return job
 }
diff --git a/pkg/availability/statefulset.go b/pkg/availability/statefulset.go
@@ -170,5 +170,9 @@ func KSMStatefulSet(
 		},
 	}
 
+	if instance.Spec.NodeSelector != nil {
+		ss.Spec.Template.Spec.NodeSelector = *instance.Spec.NodeSelector
+	}
+
 	return ss, nil
 }
diff --git a/pkg/ceilometer/statefulset.go b/pkg/ceilometer/statefulset.go
@@ -183,6 +183,10 @@ func StatefulSet(
 		},
 	}
 
+	if instance.Spec.NodeSelector != nil {
+		pod.Spec.NodeSelector = *instance.Spec.NodeSelector
+	}
+
 	statefulset := &appsv1.StatefulSet{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      ServiceName,
