feat (ocean/aws) : added ShouldScaleDownNonServiceTasks parameter (#…

…227)
spotinst · Feb 10, 2023 · 63bd932 · 63bd932
1 parent 386127a
commit 63bd932
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 7 deletions.
diff --git a/examples/service/ocean/providers/aws/createECS/main.go b/examples/service/ocean/providers/aws/createECS/main.go
@@ -42,6 +42,21 @@ func main() {
 					AvailabilityVsCost: spotinst.String("balanced"),
 				},
 			},
+			AutoScaler: &aws.ECSAutoScaler{
+				IsEnabled:    spotinst.Bool(false),
+				IsAutoConfig: spotinst.Bool(false),
+				Cooldown:     spotinst.Int(300),
+
+				Down: &aws.ECSAutoScalerDown{
+					MaxScaleDownPercentage: spotinst.Float64(20),
+				},
+				ShouldScaleDownNonServiceTasks: spotinst.Bool(true),
+
+				ResourceLimits: &aws.ECSAutoScalerResourceLimits{
+					MaxVCPU:      spotinst.Int(1024),
+					MaxMemoryGiB: spotinst.Int(20),
+				},
+			},
 			Compute: &aws.ECSCompute{
 				InstanceTypes: &aws.ECSInstanceTypes{
 					Filters: &aws.ECSFilters{

diff --git a/service/ocean/providers/aws/cluster_ecs.go b/service/ocean/providers/aws/cluster_ecs.go
@@ -216,13 +216,14 @@ type ECSIAMInstanceProfile struct {
 }
 
 type ECSAutoScaler struct {
-	IsEnabled              *bool                        `json:"isEnabled,omitempty"`
-	IsAutoConfig           *bool                        `json:"isAutoConfig,omitempty"`
-	Cooldown               *int                         `json:"cooldown,omitempty"`
-	Headroom               *ECSAutoScalerHeadroom       `json:"headroom,omitempty"`
-	ResourceLimits         *ECSAutoScalerResourceLimits `json:"resourceLimits,omitempty"`
-	Down                   *ECSAutoScalerDown           `json:"down,omitempty"`
-	AutoHeadroomPercentage *int                         `json:"autoHeadroomPercentage,omitempty"`
+	IsEnabled                      *bool                        `json:"isEnabled,omitempty"`
+	IsAutoConfig                   *bool                        `json:"isAutoConfig,omitempty"`
+	Cooldown                       *int                         `json:"cooldown,omitempty"`
+	Headroom                       *ECSAutoScalerHeadroom       `json:"headroom,omitempty"`
+	ResourceLimits                 *ECSAutoScalerResourceLimits `json:"resourceLimits,omitempty"`
+	Down                           *ECSAutoScalerDown           `json:"down,omitempty"`
+	AutoHeadroomPercentage         *int                         `json:"autoHeadroomPercentage,omitempty"`
+	ShouldScaleDownNonServiceTasks *bool                        `json:"shouldScaleDownNonServiceTasks,omitempty"`
 
 	forceSendFields []string
 	nullFields      []string
@@ -1104,6 +1105,13 @@ func (o *ECSAutoScaler) SetAutoHeadroomPercentage(v *int) *ECSAutoScaler {
 	return o
 }
 
+func (o *ECSAutoScaler) SetShouldScaleDownNonServiceTasks(v *bool) *ECSAutoScaler {
+	if o.ShouldScaleDownNonServiceTasks = v; o.ShouldScaleDownNonServiceTasks == nil {
+		o.nullFields = append(o.nullFields, "ShouldScaleDownNonServiceTasks")
+	}
+	return o
+}
+
 // endregion
 
 // region AutoScalerHeadroom