Spot capacity (#14)

* added spot capacity * improved weight calculation * fix devide by 0 * fix 100% issue * fix calculation * fix logic in calc * moved locals * fix calculation * fix calc * fix cal * add graviton to capacity providers
mergermarket · May 20, 2024 · f9dcf64 · f9dcf64
1 parent 9f473c2
commit f9dcf64
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/main.tf b/main.tf
@@ -22,16 +22,30 @@ module "ecs_update_monitor" {
 }
 
 locals {
+  p = var.spot_capacity_percentage <= 50 ? var.spot_capacity_percentage : 100 - var.spot_capacity_percentage
+  lower_weight = ceil(local.p / 100)
+  higher_weight = local.lower_weight == 0 ? 1 : (floor(local.lower_weight / (local.p / 100)) - local.lower_weight)
+  spot_weight = var.spot_capacity_percentage <= 50 ? local.lower_weight : local.higher_weight
+  ondemand_weight = var.spot_capacity_percentage <= 50 ? local.higher_weight : local.lower_weight
+
   capacity_providers = var.image_build_details["buildx"] == "true" && can(regexall("^arm64", var.image_build_details["platforms"])) ? [
     {
       capacity_provider = "${var.ecs_cluster}-native-scaling-graviton"
-      weight            = 1
+      weight            = local.ondemand_weight
     },
+    {
+      capacity_provider = "${var.ecs_cluster}-native-scaling-graviton-spot" 
+      weight            = local.spot_weight
+    }
   ] : [
     {
       capacity_provider = "${var.ecs_cluster}-native-scaling"
-      weight            = 1
-    },    
+      weight            = local.ondemand_weight
+    },
+    {
+      capacity_provider = "${var.ecs_cluster}-native-scaling-spot" 
+      weight            = local.spot_weight
+    }
   ]
 }
 

diff --git a/variables.tf b/variables.tf
@@ -269,4 +269,10 @@ variable "image_build_details" {
     "buildx" = "false",
     "platforms" = ""
   }
+}
+
+variable "spot_capacity_percentage" {
+  default = 33
+  type = number
+  description = "Percentage of tasks to run on spot instances"
 }