From 7219e8ccf2a7690e4f6cf9051cc6503ead1f51be Mon Sep 17 00:00:00 2001
From: Ranjan Shrestha <ranjan.shrestha.np@gmail.com>
Date: Mon, 22 Jul 2024 10:55:43 +0545
Subject: [PATCH] updated the config params for scaling tasks;

---
 main.tf        | 16 ++++++++++++++++
 prod.tfvars    | 12 +++++++++++-
 staging.tfvars | 10 ++++++++++
 variables.tf   | 10 ++++++++++
 4 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/main.tf b/main.tf
index 8ca5eae..04d7145 100644
--- a/main.tf
+++ b/main.tf
@@ -209,6 +209,10 @@ module "topicmodel" {
 
   # ecs task count
   app_count = var.topicmodeling_task_count
+
+  # ecs tasks max and min
+  topicmodel_scaling_max_capacity = var.topicmodel_scaling_max_capacity
+  topicmodel_scaling_min_capacity = var.topicmodel_scaling_min_capacity
 }
 
 module "ngrams" {
@@ -467,6 +471,10 @@ module "summarization_v3" {
 
   # ecs task count
   app_count = var.summarization_v3_task_count
+
+  # ecs tasks max and min
+  summarization_v3_scaling_max_capacity = var.summarization_v3_scaling_max_capacity
+  summarization_v3_scaling_min_capacity = var.summarization_v3_scaling_min_capacity
 }
 
 module "cloudmap" {
@@ -525,6 +533,10 @@ module "deepex" {
   # ecs task count
   app_count = var.text_extraction_task_count
 
+  # ecs tasks max and min
+  textextraction_scaling_max_capacity = var.textextraction_scaling_max_capacity
+  textextraction_scaling_min_capacity = var.textextraction_scaling_min_capacity
+
   # efs
   efs_volume_id = module.efilesystem.efs_volume_id
 
@@ -586,6 +598,10 @@ module "entryextraction" {
 
   # ecs task count
   app_count = var.entry_extraction_task_count
+
+  # ecs tasks max and min
+  entryextraction_scaling_max_capacity = var.entryextraction_scaling_max_capacity
+  entryextraction_scaling_min_capacity = var.entryextraction_scaling_min_capacity
 }
 
 module "reliability" {
diff --git a/prod.tfvars b/prod.tfvars
index 698e561..4474c2f 100644
--- a/prod.tfvars
+++ b/prod.tfvars
@@ -69,11 +69,21 @@ cloudflare_proxy_srv_fargate_memory = "1024"
 # ecs tasks count
 text_extraction_task_count      = 2
 entry_extraction_task_count     = 2
-summarization_v3_task_count     = 2
+summarization_v3_task_count     = 1
 geolocations_task_count         = 1
 topicmodeling_task_count        = 2
 cloudflare_proxy_srv_task_count = 1
 
+# ecs tasks max and min
+textextraction_scaling_max_capacity   = 5
+textextraction_scaling_min_capacity   = 2
+topicmodel_scaling_max_capacity       = 5
+topicmodel_scaling_min_capacity       = 2
+entryextraction_scaling_max_capacity  = 5
+entryextraction_scaling_min_capacity  = 2
+summarization_v3_scaling_max_capacity = 5
+summarization_v3_scaling_min_capacity = 1
+
 # model info
 classification_model_id      = "classification_model"
 classification_model_version = "1.0.0"
diff --git a/staging.tfvars b/staging.tfvars
index db57dc6..f3fea68 100644
--- a/staging.tfvars
+++ b/staging.tfvars
@@ -74,6 +74,16 @@ geolocations_task_count         = 1
 topicmodeling_task_count        = 1
 cloudflare_proxy_srv_task_count = 1
 
+# ecs tasks max and min
+textextraction_scaling_max_capacity   = 5
+textextraction_scaling_min_capacity   = 1
+topicmodel_scaling_max_capacity       = 5
+topicmodel_scaling_min_capacity       = 1
+entryextraction_scaling_max_capacity  = 5
+entryextraction_scaling_min_capacity  = 1
+summarization_v3_scaling_max_capacity = 5
+summarization_v3_scaling_min_capacity = 1
+
 # model info
 classification_model_id      = "classification_model"
 classification_model_version = "1.0.0"
diff --git a/variables.tf b/variables.tf
index 508ec1d..f11aa96 100644
--- a/variables.tf
+++ b/variables.tf
@@ -99,3 +99,13 @@ variable "summarization_v3_task_count" {}
 variable "geolocations_task_count" {}
 variable "topicmodeling_task_count" {}
 variable "cloudflare_proxy_srv_task_count" {}
+
+# ecs task max and min
+variable "textextraction_scaling_max_capacity" {}
+variable "textextraction_scaling_min_capacity" {}
+variable "topicmodel_scaling_max_capacity" {}
+variable "topicmodel_scaling_min_capacity" {}
+variable "entryextraction_scaling_max_capacity" {}
+variable "entryextraction_scaling_min_capacity" {}
+variable "summarization_v3_scaling_max_capacity" {}
+variable "summarization_v3_scaling_min_capacity" {}
\ No newline at end of file