Adding support for HPA

aerospike · Aug 6, 2024 · ea60954 · ea60954
1 parent 03662f3
commit ea60954
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 1 deletion.
diff --git a/api/v1/aerospikecluster_types.go b/api/v1/aerospikecluster_types.go
@@ -722,7 +722,8 @@ type AerospikeClusterStatus struct { //nolint:govet // for readability
 	Pods map[string]AerospikePodStatus `json:"pods" patchStrategy:"strategic"`
 
 	// Phase denotes the current phase of Aerospike cluster operation.
-	Phase AerospikeClusterPhase `json:"phase,omitempty"`
+	Phase    AerospikeClusterPhase `json:"phase,omitempty"`
+	Selector string                `json:"selector"`
 }
 
 // AerospikeNetworkType specifies the type of network address to use.
@@ -921,6 +922,7 @@ type AerospikePodStatus struct { //nolint:govet // for readability
 // +kubebuilder:printcolumn:name="HostNetwork",type=boolean,JSONPath=`.spec.podSpec.hostNetwork`
 // +kubebuilder:printcolumn:name="Age",type="date",JSONPath=".metadata.creationTimestamp"
 // +kubebuilder:printcolumn:name="Phase",type="string",JSONPath=".status.phase"
+//+kubebuilder:subresource:scale:specpath=.spec.size,statuspath=.status.size,selectorpath=.status.selector
 
 // AerospikeCluster is the schema for the AerospikeCluster API
 // +operator-sdk:csv:customresourcedefinitions:displayName="Aerospike Cluster",resources={{Service, v1},{Pod,v1},{StatefulSet,v1}}

diff --git a/config/crd/bases/asdb.aerospike.com_aerospikeclusters.yaml b/config/crd/bases/asdb.aerospike.com_aerospikeclusters.yaml
@@ -18282,6 +18282,8 @@ spec:
                         type: integer
                     type: object
                 type: object
+              selector:
+                type: string
               size:
                 description: Aerospike cluster size
                 format: int32
@@ -18893,11 +18895,17 @@ spec:
                 - skipWorkDirValidate
                 - skipXdrDlogFileValidate
                 type: object
+            required:
+            - selector
             type: object
         type: object
     served: true
     storage: true
     subresources:
+      scale:
+        labelSelectorPath: .status.selector
+        specReplicasPath: .spec.size
+        statusReplicasPath: .status.size
       status: {}
   - additionalPrinterColumns:
     - jsonPath: .spec.size

diff --git a/controllers/reconciler.go b/controllers/reconciler.go
@@ -10,6 +10,7 @@ import (
 	"github.com/go-logr/logr"
 	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/errors"
+	"k8s.io/apimachinery/pkg/labels"
 	k8sRuntime "k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/apimachinery/pkg/util/sets"
@@ -428,6 +429,9 @@ func (r *SingleClusterReconciler) updateStatus() error {
 		newAeroCluster.Status.IsReadinessProbeEnabled = clusterReadinessEnable
 	}
 
+	selector := labels.SelectorFromSet(utils.LabelsForAerospikeCluster(newAeroCluster.Name))
+	newAeroCluster.Status.Selector = selector.String()
+
 	err = r.patchStatus(newAeroCluster)
 	if err != nil {
 		return fmt.Errorf("error updating status: %w", err)

diff --git a/...bernetes-operator/crds/customresourcedefinition_aerospikeclusters.asdb.aerospike.com.yaml b/...bernetes-operator/crds/customresourcedefinition_aerospikeclusters.asdb.aerospike.com.yaml
@@ -18282,6 +18282,8 @@ spec:
                         type: integer
                     type: object
                 type: object
+              selector:
+                type: string
               size:
                 description: Aerospike cluster size
                 format: int32
@@ -18893,11 +18895,17 @@ spec:
                 - skipWorkDirValidate
                 - skipXdrDlogFileValidate
                 type: object
+            required:
+            - selector
             type: object
         type: object
     served: true
     storage: true
     subresources:
+      scale:
+        labelSelectorPath: .status.selector
+        specReplicasPath: .spec.size
+        statusReplicasPath: .status.size
       status: {}
   - additionalPrinterColumns:
     - jsonPath: .spec.size