feat(cdk-docker-cluster): autoscaling thresholds and cooldowns

codedazur · Sep 19, 2024 · 5a3a979 · 5a3a979
1 parent f33889f
commit 5a3a979
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 8 deletions.
diff --git a/.changeset/heavy-fishes-judge.md b/.changeset/heavy-fishes-judge.md
@@ -0,0 +1,5 @@
+---
+"@codedazur/cdk-docker-cluster": minor
+---
+
+Autoscaling thresholds and cooldowns can now be configured.
diff --git a/packages/cdk-docker-cluster/src/constructs/DockerCluster.ts b/packages/cdk-docker-cluster/src/constructs/DockerCluster.ts
@@ -2,7 +2,7 @@ import {
   SiteDistribution,
   SiteDistributionProps,
 } from "@codedazur/cdk-site-distribution";
-import { App } from "aws-cdk-lib";
+import { App, Duration } from "aws-cdk-lib";
 import {
   AllowedMethods,
   OriginProtocolPolicy,
@@ -25,12 +25,7 @@ export interface DockerClusterProps {
   source: string | SourceProps | ContainerImage;
   service?: {
     port?: number;
-    tasks?:
-      | number
-      | {
-          minimum: number;
-          maximum: number;
-        };
+    tasks?: number | AutoScalingConfig;
     cpu?: ApplicationLoadBalancedFargateServiceProps["cpu"];
     memory?: ApplicationLoadBalancedFargateServiceProps["memoryLimitMiB"];
     environment?: Record<string, string>;
@@ -39,6 +34,41 @@ export interface DockerClusterProps {
   distribution?: Omit<SiteDistributionProps, "origin">;
 }
 
+interface AutoScalingConfig {
+  /**
+   * The minimum number of tasks to scale out to.
+   */
+  minimum: number;
+  /**
+   * The maximum number of tasks to scale out to.
+   */
+  maximum: number;
+  threshold?: {
+    /**
+     * The memory utilization percentage above which the service will scale out.
+     * @default 75
+     */
+    memory?: number;
+    /**
+     * The CPU utilization percentage above which the service will scale out.
+     * @default 75
+     */
+    cpu?: number;
+  };
+  cooldown?: {
+    /**
+     * The cooldown period for scaling in.
+     * @default Duration.seconds(300)
+     */
+    in?: Duration;
+    /**
+     * The cooldown period for scaling out.
+     * @default Duration.seconds(300)
+     */
+    out?: Duration;
+  };
+}
+
 interface SourceProps {
   directory: string;
   exclude?: string[];
@@ -119,10 +149,24 @@ export class DockerCluster extends Construct {
     });
 
     if (typeof this.props.service?.tasks === "object") {
-      service.service.autoScaleTaskCount({
+      const autoScaling = service.service.autoScaleTaskCount({
         minCapacity: this.props.service?.tasks.minimum,
         maxCapacity: this.props.service?.tasks.maximum,
       });
+
+      autoScaling.scaleOnMemoryUtilization("MemoryScaling", {
+        targetUtilizationPercent:
+          this.props.service?.tasks.threshold?.memory ?? 75,
+        scaleInCooldown: this.props.service?.tasks.cooldown?.in,
+        scaleOutCooldown: this.props.service?.tasks.cooldown?.out,
+      });
+
+      autoScaling.scaleOnCpuUtilization("CpuScaling", {
+        targetUtilizationPercent:
+          this.props.service?.tasks.threshold?.cpu ?? 75,
+        scaleInCooldown: this.props.service?.tasks.cooldown?.in,
+        scaleOutCooldown: this.props.service?.tasks.cooldown?.out,
+      });
     }
 
     return service;