Merge pull request #14 from pixelfusion/feat/scaling

feat: add ability to set min capacity
pixelfusion · May 16, 2024 · f9c72ad · f9c72ad
2 parents fc0785d + 0156910
commit f9c72ad
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/dist/fargate.d.ts b/dist/fargate.d.ts
@@ -16,6 +16,7 @@ export interface TaskConfiguration {
     desiredCount?: number;
     autoScalingCpuTarget?: number;
     maxCount?: number;
+    minCount?: number;
     environment?: Record<string, string>;
     secrets?: Record<string, string>;
 }

diff --git a/dist/fargate.js b/dist/fargate.js
@@ -85,7 +85,11 @@ class FargateService extends cdk.NestedStack {
         if (taskConfiguration.autoScalingCpuTarget) {
             // Default max capacity to double desired unless specified
             const maxCapacity = taskConfiguration?.maxCount || desiredCount * 2;
-            const scaling = this.service.service.autoScaleTaskCount({ maxCapacity });
+            const minCapacity = taskConfiguration?.minCount || desiredCount;
+            const scaling = this.service.service.autoScaleTaskCount({
+                maxCapacity,
+                minCapacity,
+            });
             scaling.scaleOnCpuUtilization(stack.getResourceID('CpuScaling'), {
                 targetUtilizationPercent: taskConfiguration.autoScalingCpuTarget,
                 scaleInCooldown: cdk.Duration.seconds(60),

diff --git a/src/fargate.ts b/src/fargate.ts
@@ -28,6 +28,7 @@ export interface TaskConfiguration {
   desiredCount?: number
   autoScalingCpuTarget?: number
   maxCount?: number
+  minCount?: number
   environment?: Record<string, string>
   secrets?: Record<string, string>
 }
@@ -132,7 +133,11 @@ export class FargateService extends cdk.NestedStack {
     if (taskConfiguration.autoScalingCpuTarget) {
       // Default max capacity to double desired unless specified
       const maxCapacity = taskConfiguration?.maxCount || desiredCount * 2
-      const scaling = this.service.service.autoScaleTaskCount({ maxCapacity })
+      const minCapacity = taskConfiguration?.minCount || desiredCount
+      const scaling = this.service.service.autoScaleTaskCount({
+        maxCapacity,
+        minCapacity,
+      })
       scaling.scaleOnCpuUtilization(stack.getResourceID('CpuScaling'), {
         targetUtilizationPercent: taskConfiguration.autoScalingCpuTarget,
         scaleInCooldown: cdk.Duration.seconds(60),