Add cpu and memory to variables (#32)

Co-authored-by: Yan Pan <[email protected]>

Author: yanp

modules/metadata-service/README.md

@@ -24,6 +24,8 @@ If the `access_list_cidr_blocks` variable is set, only traffic originating from
 | <a name="input_iam_partition"></a> [iam\_partition](#input\_iam\_partition) | IAM Partition (Select aws-us-gov for AWS GovCloud, otherwise leave as is) | `string` | `"aws"` | no |
 | <a name="input_is_gov"></a> [is\_gov](#input\_is\_gov) | Set to true if IAM partition is 'aws-us-gov' | `bool` | `false` | no |
 | <a name="input_metadata_service_container_image"></a> [metadata\_service\_container\_image](#input\_metadata\_service\_container\_image) | Container image for metadata service | `string` | `""` | no |
+| <a name="input_metadata_service_cpu"></a> [metadata\_service\_cpu](#input\_metadata\_service\_cpu) | ECS task CPU unit for metadata service | `number` | `512` | no |
+| <a name="input_metadata_service_memory"></a> [metadata\_service\_memory](#input\_metadata\_service\_memory) | ECS task memory in MiB for metadata service | `number` | `1024` | no |
 | <a name="input_metaflow_vpc_id"></a> [metaflow\_vpc\_id](#input\_metaflow\_vpc\_id) | ID of the Metaflow VPC this SageMaker notebook instance is to be deployed in | `string` | n/a | yes |
 | <a name="input_rds_master_instance_endpoint"></a> [rds\_master\_instance\_endpoint](#input\_rds\_master\_instance\_endpoint) | The database connection endpoint in address:port format | `string` | n/a | yes |
 | <a name="input_resource_prefix"></a> [resource\_prefix](#input\_resource\_prefix) | Prefix given to all AWS resources to differentiate between applications | `string` | n/a | yes |

modules/metadata-service/ecs.tf

@@ -19,8 +19,8 @@ resource "aws_ecs_task_definition" "this" {
     "name": "${var.resource_prefix}service${var.resource_suffix}",
     "image": "${var.metadata_service_container_image}",
     "essential": true,
-    "cpu": 512,
-    "memory": 1024,
+    "cpu": var.metadata_service_cpu,
+    "memory": var.metadata_service_memory,
     "portMappings": [
       {
         "containerPort": 8080,

modules/metadata-service/variables.tf

@@ -47,6 +47,18 @@ variable "metadata_service_container_image" {
   description = "Container image for metadata service"
 }
 
+variable "metadata_service_cpu" {
+  type        = number
+  default     = 512
+  description = "ECS task CPU unit for metadata service"
+}
+
+variable "metadata_service_memory" {
+  type        = number
+  default     = 1024
+  description = "ECS task memory in MiB for metadata service"
+}
+
 variable "metaflow_vpc_id" {
   type        = string
   description = "ID of the Metaflow VPC this SageMaker notebook instance is to be deployed in"