Allow for custom cooldown for steps (#221)

* Differenciating scale in/out cooldowns

Author: JohnPreston

ecs_composex/ecs/ecs_scaling.py

@@ -129,12 +129,18 @@ def generate_alarm_scaling_out_policy(
         StepScalingPolicyConfiguration=StepScalingPolicyConfiguration(
             AdjustmentType="ExactCapacity",
             StepAdjustments=step_adjustments,
+            Cooldown=60
+            if not keyisset("scale_out_cooldown", scaling_def)
+            or not (isinstance(scaling_def["scale_out_cooldown"], int))
+            else scaling_def["scale_out_cooldown"],
         ),
     )
     return policy
 
 
-def reset_to_zero_policy(service_name, service_template, scaling_source=None):
+def reset_to_zero_policy(
+    service_name, service_template, scaling_def, scaling_source=None
+):
     """
 
     :return:
@@ -153,11 +159,15 @@ def reset_to_zero_policy(service_name, service_template, scaling_source=None):
         ServiceNamespace="ecs",
         StepScalingPolicyConfiguration=StepScalingPolicyConfiguration(
             AdjustmentType="ExactCapacity",
+            Cooldown=60
+            if not keyisset("scale_in_cooldown", scaling_def)
+            or not (isinstance(scaling_def["scale_in_cooldown"], int))
+            else scaling_def["scale_in_cooldown"],
             StepAdjustments=[
                 StepAdjustment(
                     MetricIntervalUpperBound=0,
                     ScalingAdjustment=0,
-                )
+                ),
             ],
         ),
     )

ecs_composex/sqs/sqs_ecs.py

@@ -126,6 +126,7 @@ def handle_service_scaling(
     scaling_in_policy = reset_to_zero_policy(
         service_family,
         service_stack.stack_template,
+        service_def["scaling"],
         scaling_source=resource.logical_name,
     )
     Alarm(