Expose scale subresource to enable HPA's to modify the replicas of In…

…finispan CR (#2133)
infinispan · Sep 30, 2024 · 32d71f3 · 32d71f3
1 parent ce5f149
commit 32d71f3
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/api/v1/infinispan_types.go b/api/v1/infinispan_types.go
@@ -611,6 +611,9 @@ type InfinispanStatus struct {
 	// +optional
 	// +operator-sdk:csv:customresourcedefinitions:type=status,displayName="Operator Status"
 	Operator Operator `json:"operator,omitempty"`
+	// The Selector used to identify Infinispan cluster pods
+	// +optional
+	Selector string  `json:"selector"`
 }
 
 type OperandPhase string
@@ -666,6 +669,7 @@ type Operator struct {
 
 // +kubebuilder:object:root=true
 // +kubebuilder:subresource:status
+// +kubebuilder:subresource:scale:specpath=.spec.replicas,statuspath=.status.replicas,selectorpath=.status.selector
 // +operator-sdk:csv:customresourcedefinitions:displayName="Infinispan Cluster"
 
 // Infinispan is the Schema for the infinispans API

diff --git a/config/crd/bases/infinispan.org_infinispans.yaml b/config/crd/bases/infinispan.org_infinispans.yaml
@@ -2461,11 +2461,18 @@ spec:
                     description: The secret that contains user credentials.
                     type: string
                 type: object
+              selector:
+                description: The Operator Selector
+                type: string
               statefulSetName:
                 type: string
             type: object
         type: object
     served: true
     storage: true
     subresources:
+      scale:
+        labelSelectorPath: .status.selector
+        specReplicasPath: .spec.replicas
+        statusReplicasPath: .status.replicas
       status: {}
diff --git a/pkg/reconcile/pipeline/infinispan/handler/provision/statefulsets.go b/pkg/reconcile/pipeline/infinispan/handler/provision/statefulsets.go
@@ -60,9 +60,16 @@ func ClusterStatefulSet(i *ispnv1.Infinispan, ctx pipeline.Context) {
 		return
 	}
 
+	selector, err := metav1.LabelSelectorAsSelector(statefulSet.Spec.Selector)
+
+	if err != nil {
+		ctx.Requeue(fmt.Errorf("error retrieving StatefulSet labels: %w", err))
+		return
+	}
 	_ = ctx.UpdateInfinispan(func() {
 		i.Status.Replicas = &i.Spec.Replicas
 		i.Status.StatefulSetName = statefulSet.Name
+		i.Status.Selector = selector.String()
 	})
 }