feat(crd): introduce loadbalancer config

dragonflydb · Oct 27, 2023 · 0bb90ab · 0bb90ab
1 parent 7cf5b4b
commit 0bb90ab
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 0 deletions.
diff --git a/api/v1alpha1/dragonfly_types.go b/api/v1alpha1/dragonfly_types.go
@@ -88,6 +88,30 @@ type DragonflySpec struct {
 	// +optional
 	// +kubebuilder:validation:Optional
 	Snapshot *Snapshot `json:"snapshot,omitempty"`
+
+	// (Optional) Dragonfly LoadBalancer configuration
+	// +optional
+	// +kubebuilder:validation:Optional
+	LoadBalancerSpec *LoadBalancerService `json:"loadbalancerSpec,omitempty"`
+}
+
+type LoadBalancerService struct {
+	Enabled bool `json:"enabled,omitempty"`
+
+	// (Optional) Dragonfly LoadBalancerIP
+	// +optional
+	// +kubebuilder:validation:Optional
+	ServiceIp string `json:"serviceIp,omitempty"`
+
+	// (Optional) Dragonfly LoadBalancerSourceRanges
+	// +optional
+	// +kubebuilder:validation:Optional
+	SourceRanges []string `json:"sourceRanges,omitempty"`
+
+	// (Optional) Dragonfly LoadBalancerClass
+	// +optional
+	// +kubebuilder:validation:Optional
+	Class string `json:"class,omitempty"`
 }
 
 type Snapshot struct {

diff --git a/config/crd/bases/dragonflydb.io_dragonflies.yaml b/config/crd/bases/dragonflydb.io_dragonflies.yaml
@@ -1026,6 +1026,23 @@ spec:
               image:
                 description: Image is the Dragonfly image to use
                 type: string
+              loadbalancerSpec:
+                description: (Optional) Dragonfly LoadBalancer configuration
+                properties:
+                  class:
+                    description: (Optional) Dragonfly snapshot schedule
+                    type: string
+                  enabled:
+                    type: boolean
+                  serviceIp:
+                    description: (Optional) Dragonfly snapshot schedule
+                    type: string
+                  sourceRanges:
+                    description: (Optional) Dragonfly snapshot schedule
+                    items:
+                      type: string
+                    type: array
+                type: object
               replicas:
                 description: Replicas is the total number of Dragonfly instances including
                   the master

diff --git a/internal/resources/resources.go b/internal/resources/resources.go
@@ -313,6 +313,20 @@ func GetDragonflyResources(ctx context.Context, df *resourcesv1.Dragonfly) ([]cl
 		},
 	}
 
+	if df.Spec.LoadBalancerSpec != nil && df.Spec.LoadBalancerSpec.Enabled {
+		service.Spec.Type = corev1.ServiceTypeLoadBalancer
+
+		if df.Spec.LoadBalancerSpec.ServiceIp != "" {
+			service.Spec.LoadBalancerIP = df.Spec.LoadBalancerSpec.ServiceIp
+		}
+		if len(df.Spec.LoadBalancerSpec.SourceRanges) != 0 {
+			service.Spec.LoadBalancerSourceRanges = df.Spec.LoadBalancerSpec.SourceRanges
+		}
+		if df.Spec.LoadBalancerSpec.Class != "" {
+			service.Spec.LoadBalancerClass = &df.Spec.LoadBalancerSpec.Class
+		}
+	}
+
 	resources = append(resources, &service)
 
 	return resources, nil