Merge pull request #12 from pixelfusion/feat/scaling

feat: add autoscaling to fargate
pixelfusion · May 13, 2024 · 052361d · 052361d
2 parents ac359f2 + 43262c2
commit 052361d
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 4 deletions.
diff --git a/dist/fargate.d.ts b/dist/fargate.d.ts
@@ -14,6 +14,8 @@ export interface TaskConfiguration {
     healthCheckGracePeriod?: cdk.Duration;
     cpu?: number;
     desiredCount?: number;
+    autoScalingCpuTarget?: number;
+    maxCount?: number;
     environment?: Record<string, string>;
     secrets?: Record<string, string>;
 }

diff --git a/dist/fargate.js b/dist/fargate.js
@@ -61,6 +61,7 @@ class FargateService extends cdk.NestedStack {
         const image = imageVersion === DEFAULT_VERSION
             ? ecs.ContainerImage.fromRegistry(DEFAULT_IMAGE)
             : ecs.ContainerImage.fromEcrRepository(repository, imageVersion);
+        const desiredCount = taskConfiguration?.desiredCount || 1;
         this.service = new ecs_patterns.ApplicationLoadBalancedFargateService(this, stack.getResourceID('AdminService'), {
             assignPublicIp: true,
             cluster: cluster,
@@ -69,7 +70,7 @@ class FargateService extends cdk.NestedStack {
             memoryLimitMiB: taskConfiguration?.memoryLimitMiB || 512,
             healthCheckGracePeriod: taskConfiguration?.healthCheckGracePeriod || cdk.Duration.seconds(60),
             cpu: taskConfiguration?.cpu || 256,
-            desiredCount: taskConfiguration?.desiredCount || 1,
+            desiredCount,
             taskImageOptions: {
                 image,
                 environment: taskConfiguration?.environment || {},
@@ -80,6 +81,17 @@ class FargateService extends cdk.NestedStack {
                 onePerAz: true,
             },
         });
+        // Setup AutoScaling policy
+        if (taskConfiguration.autoScalingCpuTarget) {
+            // Default max capacity to double desired unless specified
+            const maxCapacity = taskConfiguration?.maxCount || desiredCount * 2;
+            const scaling = this.service.service.autoScaleTaskCount({ maxCapacity });
+            scaling.scaleOnCpuUtilization(stack.getResourceID('CpuScaling'), {
+                targetUtilizationPercent: taskConfiguration.autoScalingCpuTarget,
+                scaleInCooldown: cdk.Duration.seconds(60),
+                scaleOutCooldown: cdk.Duration.seconds(60),
+            });
+        }
         // Hack to fix subnets issue
         // https://github.com/aws/aws-cdk/issues/5892#issuecomment-701993883
         const cfnLoadBalancer = this.service.loadBalancer.node

diff --git a/package-lock.json b/package-lock.json
diff --git a/src/fargate.ts b/src/fargate.ts
@@ -26,6 +26,8 @@ export interface TaskConfiguration {
   healthCheckGracePeriod?: cdk.Duration
   cpu?: number
   desiredCount?: number
+  autoScalingCpuTarget?: number
+  maxCount?: number
   environment?: Record<string, string>
   secrets?: Record<string, string>
 }
@@ -100,6 +102,7 @@ export class FargateService extends cdk.NestedStack {
         ? ecs.ContainerImage.fromRegistry(DEFAULT_IMAGE)
         : ecs.ContainerImage.fromEcrRepository(repository, imageVersion)
 
+    const desiredCount = taskConfiguration?.desiredCount || 1
     this.service = new ecs_patterns.ApplicationLoadBalancedFargateService(
       this,
       stack.getResourceID('AdminService'),
@@ -112,7 +115,7 @@ export class FargateService extends cdk.NestedStack {
         healthCheckGracePeriod:
           taskConfiguration?.healthCheckGracePeriod || cdk.Duration.seconds(60),
         cpu: taskConfiguration?.cpu || 256,
-        desiredCount: taskConfiguration?.desiredCount || 1,
+        desiredCount,
         taskImageOptions: {
           image,
           environment: taskConfiguration?.environment || {},
@@ -125,6 +128,18 @@ export class FargateService extends cdk.NestedStack {
       },
     )
 
+    // Setup AutoScaling policy
+    if (taskConfiguration.autoScalingCpuTarget) {
+      // Default max capacity to double desired unless specified
+      const maxCapacity = taskConfiguration?.maxCount || desiredCount * 2
+      const scaling = this.service.service.autoScaleTaskCount({ maxCapacity })
+      scaling.scaleOnCpuUtilization(stack.getResourceID('CpuScaling'), {
+        targetUtilizationPercent: taskConfiguration.autoScalingCpuTarget,
+        scaleInCooldown: cdk.Duration.seconds(60),
+        scaleOutCooldown: cdk.Duration.seconds(60),
+      })
+    }
+
     // Hack to fix subnets issue
     // https://github.com/aws/aws-cdk/issues/5892#issuecomment-701993883
     const cfnLoadBalancer = this.service.loadBalancer.node