Add gpu type to cluster spec (#84)

* Add gpu type to cluster spec

* Fix typos

* Do not mark constants as unused in Goland. They are used in other projects.

Author: bashofmann

api/v1/qdrantcluster_types.go

@@ -9,11 +9,13 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
+//goland:noinspection GoUnusedConst
 const (
 	KindQdrantCluster     = "QdrantCluster"
 	ResourceQdrantCluster = "qdrantclusters"
 )
 
+//goland:noinspection GoUnusedConst
 const (
 	// RestartedAtAnnotationKey is the annotation key to trigger a restart.
 	// The annotation should be placed on the QdrantCluster instance.
@@ -72,6 +74,14 @@ func GetQdrantClusterCrdForHash(qc QdrantCluster) QdrantCluster {
 	return result
 }
 
+type GPUType string
+
+//goland:noinspection GoUnusedConst
+const (
+	GPUTypeNvidia GPUType = "nvidia"
+	GPUTypeAmd    GPUType = "amd"
+)
+
 // QdrantClusterSpec defines the desired state of QdrantCluster
 // +kubebuilder:pruning:PreserveUnknownFields
 type QdrantClusterSpec struct {
@@ -125,6 +135,9 @@ type QdrantClusterSpec struct {
 	// Service specifies the configuration of the Qdrant Kubernetes Service.
 	// +optional
 	Service *KubernetesService `json:"service,omitempty"`
+	// GPU specifies GPU configuration for the cluster. If this field is not set, no GPU will be used.
+	// +optional
+	GPU *GPU `json:"gpu,omitempty"`
 	// StatefulSet specifies the configuration of the Qdrant Kubernetes StatefulSet.
 	// +optional
 	StatefulSet *KubernetesStatefulSet `json:"statefulSet,omitempty"`
@@ -151,7 +164,7 @@ type QdrantClusterSpec struct {
 	StartupDelaySeconds *int `json:"startupDelaySeconds,omitempty"`
 }
 
-// Validates if there are incorrect settings in the CRD
+// Validate if there are incorrect settings in the CRD
 func (s QdrantClusterSpec) Validate() error {
 	if err := s.Resources.Validate("Spec.Resources"); err != nil {
 		return err
@@ -167,6 +180,19 @@ func (s QdrantClusterSpec) GetServicePerNode() bool {
 	return *s.ServicePerNode
 }
 
+type GPU struct {
+	// GPUType specifies the type of the GPU to use.
+	// +kubebuilder:validation:Enum=nvidia;amd
+	GPUType GPUType `json:"gpuType"`
+}
+
+func (g *GPU) GetGPUType() GPUType {
+	if g == nil {
+		return ""
+	}
+	return g.GPUType
+}
+
 type KubernetesService struct {
 	// Type specifies the type of the Service: "ClusterIP", "NodePort", "LoadBalancer".
 	// +kubebuilder:default="ClusterIP"
@@ -296,7 +322,7 @@ type Resources struct {
 	Requests ResourceRequests `json:"requests,omitempty"`
 }
 
-// Validates if there are incorrect settings in the CRD
+// Validate if there are incorrect settings in the CRD
 func (s Resources) Validate(base string) error {
 	if _, err := resource.ParseQuantity(s.CPU); err != nil {
 		return fmt.Errorf("%s.CPU error: %w", base, err)
@@ -336,7 +362,7 @@ type ResourceRequests struct {
 	Memory string `json:"memory,omitempty"`
 }
 
-// Validates if there are incorrect settings in the CRD
+// Validate if there are incorrect settings in the CRD
 func (s ResourceRequests) Validate(base string) error {
 	if s.CPU != "" {
 		if _, err := resource.ParseQuantity(s.CPU); err != nil {
@@ -688,6 +714,7 @@ func (n *StorageClassNames) GetSnapshots() *string {
 
 type ClusterPhase string
 
+//goland:noinspection GoUnusedConst
 const (
 	ClusterActiveStateSuffix = "ing"
 	ClusterFailedStatePrefix = "FailedTo"
@@ -716,6 +743,7 @@ const (
 
 type ClusterCondition string
 
+//goland:noinspection GoUnusedConst
 const (
 	ClusterConditionAcceptingConnection ClusterCondition = "AcceptingConnection"
 	ClusterConditionRecoveryMode        ClusterCondition = "RecoveryMode"

api/v1/qdrantclusterrestore_types.go

@@ -2,6 +2,7 @@ package v1
 
 import metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 
+//goland:noinspection GoUnusedConst
 const (
 	KindQdrantClusterRestore     = "QdrantClusterRestore"
 	ResourceQdrantClusterRestore = "qdrantclusterrestores"
@@ -31,6 +32,7 @@ type RestoreDestination struct {
 
 type RestorePhase string
 
+//goland:noinspection GoUnusedConst
 const (
 	RestoreRunning   RestorePhase = "Running"
 	RestoreSkipped   RestorePhase = "Skipped"

api/v1/qdrantclusterscheduledsnapshot_types.go

@@ -2,6 +2,7 @@ package v1
 
 import metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 
+//goland:noinspection GoUnusedConst
 const (
 	KindQdrantClusterScheduledSnapshot     = "QdrantClusterScheduledSnapshot"
 	ResourceQdrantClusterScheduledSnapshot = "qdrantclusterscheduledsnapshots"
@@ -25,6 +26,7 @@ type QdrantClusterScheduledSnapshotSpec struct {
 
 type ScheduledSnapshotPhase string
 
+//goland:noinspection GoUnusedConst
 const (
 	ScheduleActive   ScheduledSnapshotPhase = "Active"
 	ScheduleDisabled ScheduledSnapshotPhase = "Disabled"

api/v1/qdrantclustersnapshot_types.go

@@ -2,6 +2,7 @@ package v1
 
 import metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 
+//goland:noinspection GoUnusedConst
 const (
 	KindQdrantClusterSnapshot     = "QdrantClusterSnapshot"
 	ResourceQdrantClusterSnapshot = "qdrantclustersnapshots"
@@ -28,6 +29,7 @@ type QdrantClusterSnapshotSpec struct {
 
 type QdrantClusterSnapshotPhase string
 
+//goland:noinspection GoUnusedConst
 const (
 	SnapshotRunning   QdrantClusterSnapshotPhase = "Running"
 	SnapshotSkipped   QdrantClusterSnapshotPhase = "Skipped"

api/v1/region_types.go

@@ -6,6 +6,7 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
+//goland:noinspection GoUnusedConst
 const (
 	KindQdrantCloudRegion     = "QdrantCloudRegion"
 	ResourceQdrantCloudRegion = "qdrantcloudregions"
@@ -62,6 +63,7 @@ type HelmRelease struct {
 
 type RegionPhase string
 
+//goland:noinspection GoUnusedConst
 const (
 	RegionPhaseReady    RegionPhase = "Ready"
 	RegionPhaseNotReady RegionPhase = "NotReady"
@@ -70,6 +72,7 @@ const (
 
 type MetricSource string
 
+//goland:noinspection GoUnusedConst
 const (
 	KubeletMetricSource MetricSource = "kubelet"
 	ApiMetricSource     MetricSource = "api"
@@ -189,6 +192,7 @@ type RegionCapabilities struct {
 
 type KubernetesDistribution string
 
+//goland:noinspection GoUnusedConst
 const (
 	K8sDistributionUnknown   KubernetesDistribution = "unknown"
 	K8sDistributionAWS       KubernetesDistribution = "aws"
@@ -208,6 +212,7 @@ const (
 
 type ComponentPhase string
 
+//goland:noinspection GoUnusedConst
 const (
 	ComponentPhaseReady    ComponentPhase = "Ready"
 	ComponentPhaseNotReady ComponentPhase = "NotReady"

api/v1/zz_generated.deepcopy.go

@@ -296,6 +296,19 @@ spec:
                         type: object
                     type: object
                 type: object
+              gpu:
+                description: GPU specifies GPU configuration for the cluster. If this
+                  field is not set, no GPU will be used.
+                properties:
+                  gpuType:
+                    description: GPUType specifies the type of the GPU to use.
+                    enum:
+                    - nvidia
+                    - amd
+                    type: string
+                required:
+                - gpuType
+                type: object
               id:
                 description: Id specifies the unique identifier of the cluster
                 type: string

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

@@ -295,6 +295,19 @@ spec:
                         type: object
                     type: object
                 type: object
+              gpu:
+                description: GPU specifies GPU configuration for the cluster. If this
+                  field is not set, no GPU will be used.
+                properties:
+                  gpuType:
+                    description: GPUType specifies the type of the GPU to use.
+                    enum:
+                    - nvidia
+                    - amd
+                    type: string
+                required:
+                - gpuType
+                type: object
               id:
                 description: Id specifies the unique identifier of the cluster
                 type: string

crds/qdrant.io_qdrantclusters.yaml

@@ -92,6 +92,35 @@ _Appears in:_
 | `message` _string_ | Message specifies the info explaining the current phase of the component |  |  |
 
 
+#### GPU
+
+
+
+
+
+
+
+_Appears in:_
+- [QdrantClusterSpec](#qdrantclusterspec)
+
+| Field | Description | Default | Validation |
+| --- | --- | --- | --- |
+| `gpuType` _[GPUType](#gputype)_ | GPUType specifies the type of the GPU to use. |  | Enum: [nvidia amd] <br /> |
+
+
+#### GPUType
+
+_Underlying type:_ _string_
+
+
+
+
+
+_Appears in:_
+- [GPU](#gpu)
+
+
+
 #### HelmRelease
 
 
@@ -663,6 +692,7 @@ _Appears in:_
 | `config` _[QdrantConfiguration](#qdrantconfiguration)_ | Config specifies the Qdrant configuration setttings for the clusters. |  |  |
 | `ingress` _[Ingress](#ingress)_ | Ingress specifies the ingress for the cluster. |  |  |
 | `service` _[KubernetesService](#kubernetesservice)_ | Service specifies the configuration of the Qdrant Kubernetes Service. |  |  |
+| `gpu` _[GPU](#gpu)_ | GPU specifies GPU configuration for the cluster. If this field is not set, no GPU will be used. |  |  |
 | `statefulSet` _[KubernetesStatefulSet](#kubernetesstatefulset)_ | StatefulSet specifies the configuration of the Qdrant Kubernetes StatefulSet. |  |  |
 | `storageClassNames` _[StorageClassNames](#storageclassnames)_ | StorageClassNames specifies the storage class names for db and snapshots. |  |  |
 | `topologySpreadConstraints` _[TopologySpreadConstraint](https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.28/#topologyspreadconstraint-v1-core)_ | TopologySpreadConstraints specifies the topology spread constraints for the cluster. |  |  |