Add rebalanceStrategy in QdrantCluster spec (#109)

* Add rebalanceStrategy in QdrantCluster spec

* Update docs

Author: bashofmann

api/v1/qdrantcluster_types.go

@@ -82,6 +82,15 @@ const (
 	GPUTypeAmd    GPUType = "amd"
 )
 
+type RebalanceStrategy string
+
+//goland:noinspection GoUnusedConst
+const (
+	ByCount        RebalanceStrategy = "by_count"
+	BySize         RebalanceStrategy = "by_size"
+	ByCountAndSize RebalanceStrategy = "by_count_and_size"
+)
+
 // QdrantClusterSpec defines the desired state of QdrantCluster
 // +kubebuilder:pruning:PreserveUnknownFields
 type QdrantClusterSpec struct {
@@ -162,6 +171,10 @@ type QdrantClusterSpec struct {
 	// to inspect the attached storage.
 	// +optional
 	StartupDelaySeconds *int `json:"startupDelaySeconds,omitempty"`
+	// RebalanceStrategy specifies the strategy to use for automaticially rebalancing shards the cluster.
+	// Cluster-manager needs to be enabled for this feature to work.
+	// +optional
+	RebalanceStrategy *RebalanceStrategy `json:"rebalanceStrategy,omitempty"`
 }
 
 // Validate if there are incorrect settings in the CRD

api/v1/zz_generated.deepcopy.go

@@ -565,6 +565,11 @@ spec:
                       the feature gate PDBUnhealthyPodEvictionPolicy is enabled (enabled by default).
                     type: string
                 type: object
+              rebalanceStrategy:
+                description: |-
+                  RebalanceStrategy specifies the strategy to use for automaticially rebalancing shards the cluster.
+                  Cluster-manager needs to be enabled for this feature to work.
+                type: string
               resources:
                 description: Resources specifies the resources to allocate for each
                   Qdrant node.

charts/qdrant-kubernetes-api/templates/region-crds/qdrant.io_qdrantclusters.yaml

@@ -564,6 +564,11 @@ spec:
                       the feature gate PDBUnhealthyPodEvictionPolicy is enabled (enabled by default).
                     type: string
                 type: object
+              rebalanceStrategy:
+                description: |-
+                  RebalanceStrategy specifies the strategy to use for automaticially rebalancing shards the cluster.
+                  Cluster-manager needs to be enabled for this feature to work.
+                type: string
               resources:
                 description: Resources specifies the resources to allocate for each
                   Qdrant node.

crds/qdrant.io_qdrantclusters.yaml

@@ -733,6 +733,7 @@ _Appears in:_
 | `podDisruptionBudget` _[PodDisruptionBudgetSpec](https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.28/#poddisruptionbudgetspec-v1-policy)_ | PodDisruptionBudget specifies the pod disruption budget for the cluster. |  |  |
 | `restartAllPodsConcurrently` _boolean_ | RestartAllPodsConcurrently specifies whether to restart all pods concurrently (also called one-shot-restart).<br />If enabled, all the pods in the cluster will be restarted concurrently in situations where multiple pods<br />need to be restarted like when RestartedAtAnnotationKey is added/updated or the Qdrant version need to be upgraded.<br />This helps sharded but not replicated clusters to reduce downtime to possible minimum during restart. |  |  |
 | `startupDelaySeconds` _integer_ | If StartupDelaySeconds is set (> 0), an additional 'sleep <value>' will be emitted to the pod startup.<br />The sleep will be added when a pod is restarted, it will not force any pod to restart.<br />This feature can be used for debugging the core, e.g. if a pod is in crash loop, it provided a way<br />to inspect the attached storage. |  |  |
+| `rebalanceStrategy` _[RebalanceStrategy](#rebalancestrategy)_ | RebalanceStrategy specifies the strategy to use for automaticially rebalancing shards the cluster.<br />Cluster-manager needs to be enabled for this feature to work. |  |  |
 
 
 
@@ -1023,6 +1024,19 @@ _Appears in:_
 | `fsGroup` _integer_ | FsGroup specifies file system group to run the Qdrant process as. |  |  |
 
 
+#### RebalanceStrategy
+
+_Underlying type:_ _string_
+
+
+
+
+
+_Appears in:_
+- [QdrantClusterSpec](#qdrantclusterspec)
+
+
+
 #### RegionCapabilities