[Feature] [Scheduler] Deployment Scale (#1764)

ajanikow · web-flow · commit f89e64d707cb · 2024-11-12T17:55:27.000+01:00
diff --git a/.golangci.yaml b/.golangci.yaml
@@ -131,6 +131,8 @@ linters-settings:
         pkg: k8s.io/api/storage/v1
       - alias: meta
         pkg: k8s.io/apimachinery/pkg/apis/meta/v1
+      - alias: autoscaling
+        pkg: k8s.io/api/autoscaling/v1
       - alias: typedCore
         pkg: k8s.io/client-go/kubernetes/typed/core/v1
       - alias: ugrpc
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -21,6 +21,7 @@
 - (Feature) (Platform) Storage
 - (Maintenance) Extract GRPC Client Package
 - (Feature) (Platform) Chart
+- (Feature) (Scheduler) Deployment Scale Functionality
 
 ## [1.2.43](https://github.com/arangodb/kube-arangodb/tree/1.2.43) (2024-10-14)
 - (Feature) ArangoRoute CRD
diff --git a/chart/kube-arangodb-arm64/crds/scheduler-deployment.yaml b/chart/kube-arangodb-arm64/crds/scheduler-deployment.yaml
@@ -20,3 +20,6 @@ spec:
       storage: true
       subresources:
         status: {}
+        scale:
+          specReplicasPath: .spec.replicas
+          statusReplicasPath: .status.replicas
diff --git a/chart/kube-arangodb-enterprise-arm64/crds/scheduler-deployment.yaml b/chart/kube-arangodb-enterprise-arm64/crds/scheduler-deployment.yaml
@@ -20,3 +20,6 @@ spec:
       storage: true
       subresources:
         status: {}
+        scale:
+          specReplicasPath: .spec.replicas
+          statusReplicasPath: .status.replicas
diff --git a/chart/kube-arangodb-enterprise/crds/scheduler-deployment.yaml b/chart/kube-arangodb-enterprise/crds/scheduler-deployment.yaml
@@ -20,3 +20,6 @@ spec:
       storage: true
       subresources:
         status: {}
+        scale:
+          specReplicasPath: .spec.replicas
+          statusReplicasPath: .status.replicas
diff --git a/chart/kube-arangodb/crds/scheduler-deployment.yaml b/chart/kube-arangodb/crds/scheduler-deployment.yaml
@@ -20,3 +20,6 @@ spec:
       storage: true
       subresources:
         status: {}
+        scale:
+          specReplicasPath: .spec.replicas
+          statusReplicasPath: .status.replicas
diff --git a/pkg/apis/scheduler/v1beta1/types_deployment.go b/pkg/apis/scheduler/v1beta1/types_deployment.go
@@ -38,6 +38,8 @@ type ArangoSchedulerDeploymentList struct {
 }
 
 // +genclient
+// +genclient:method=GetScale,verb=get,subresource=scale,result=k8s.io/api/autoscaling/v1.Scale
+// +genclient:method=UpdateScale,verb=update,subresource=scale,input=k8s.io/api/autoscaling/v1.Scale,result=k8s.io/api/autoscaling/v1.Scale
 // +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
 
 // ArangoSchedulerDeployment wraps apps. ArangoSchedulerDeployment with profile details
diff --git a/pkg/crd/crds/scheduler-deployment.yaml b/pkg/crd/crds/scheduler-deployment.yaml
@@ -20,3 +20,6 @@ spec:
       storage: true
       subresources:
         status: {}
+        scale:
+          specReplicasPath: .spec.replicas
+          statusReplicasPath: .status.replicas
diff --git a/pkg/generated/clientset/versioned/typed/scheduler/v1beta1/arangoschedulerdeployment.go b/pkg/generated/clientset/versioned/typed/scheduler/v1beta1/arangoschedulerdeployment.go
diff --git a/pkg/generated/clientset/versioned/typed/scheduler/v1beta1/fake/fake_arangoschedulerdeployment.go b/pkg/generated/clientset/versioned/typed/scheduler/v1beta1/fake/fake_arangoschedulerdeployment.go
diff --git a/pkg/util/k8sutil/inspector/generic/mod.go b/pkg/util/k8sutil/inspector/generic/mod.go
@@ -23,6 +23,7 @@ package generic
 import (
 	"context"
 
+	autoscaling "k8s.io/api/autoscaling/v1"
 	meta "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
 )
@@ -61,6 +62,19 @@ type DeleteInterface[S meta.Object] interface {
 	Delete(ctx context.Context, name string, opts meta.DeleteOptions) error
 }
 
+type ScaleGet interface {
+	GetScale(ctx context.Context, name string, options meta.GetOptions) (*autoscaling.Scale, error)
+}
+
+type ScaleUpdate interface {
+	UpdateScale(ctx context.Context, name string, scale *autoscaling.Scale, opts meta.UpdateOptions) (*autoscaling.Scale, error)
+}
+
+type Scale interface {
+	ScaleGet
+	ScaleUpdate
+}
+
 type ReadClient[S meta.Object] interface {
 	GetInterface[S]
 }

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,8 @@ type ArangoSchedulerDeploymentList struct {`
`38`	`38`	`}`
`39`	`39`
`40`	`40`	`// +genclient`
	`41`	`+// +genclient:method=GetScale,verb=get,subresource=scale,result=k8s.io/api/autoscaling/v1.Scale`
	`42`	`+// +genclient:method=UpdateScale,verb=update,subresource=scale,input=k8s.io/api/autoscaling/v1.Scale,result=k8s.io/api/autoscaling/v1.Scale`
`41`	`43`	`// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object`
`42`	`44`
`43`	`45`	`// ArangoSchedulerDeployment wraps apps. ArangoSchedulerDeployment with profile details`