feat: Add support for workspace configuration (#25)

Co-authored-by: Bryant Biggs <[email protected]>

Author: Sudokamikaze

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/antonbabenko/pre-commit-terraform
-    rev: v1.103.0
+    rev: v1.104.0
     hooks:
       - id: terraform_fmt
       - id: terraform_docs

README.md

@@ -58,13 +58,13 @@ Examples codified under the [`examples`](https://github.com/terraform-aws-module
 | Name | Version |
 |------|---------|
 | <a name="requirement_terraform"></a> [terraform](#requirement\_terraform) | >= 1.5.7 |
-| <a name="requirement_aws"></a> [aws](#requirement\_aws) | >= 6.0 |
+| <a name="requirement_aws"></a> [aws](#requirement\_aws) | >= 6.22 |
 
 ## Providers
 
 | Name | Version |
 |------|---------|
-| <a name="provider_aws"></a> [aws](#provider\_aws) | >= 6.0 |
+| <a name="provider_aws"></a> [aws](#provider\_aws) | >= 6.22 |
 
 ## Modules
 
@@ -78,6 +78,7 @@ No modules.
 | [aws_prometheus_alert_manager_definition.this](https://registry.terraform.io/providers/hashicorp/aws/latest/docs/resources/prometheus_alert_manager_definition) | resource |
 | [aws_prometheus_rule_group_namespace.this](https://registry.terraform.io/providers/hashicorp/aws/latest/docs/resources/prometheus_rule_group_namespace) | resource |
 | [aws_prometheus_workspace.this](https://registry.terraform.io/providers/hashicorp/aws/latest/docs/resources/prometheus_workspace) | resource |
+| [aws_prometheus_workspace_configuration.this](https://registry.terraform.io/providers/hashicorp/aws/latest/docs/resources/prometheus_workspace_configuration) | resource |
 
 ## Inputs
 
@@ -93,8 +94,10 @@ No modules.
 | <a name="input_create_alert_manager"></a> [create\_alert\_manager](#input\_create\_alert\_manager) | Controls whether an Alert Manager definition is created along with the AMP workspace | `bool` | `true` | no |
 | <a name="input_create_workspace"></a> [create\_workspace](#input\_create\_workspace) | Determines whether a workspace will be created or to use an existing workspace | `bool` | `true` | no |
 | <a name="input_kms_key_arn"></a> [kms\_key\_arn](#input\_kms\_key\_arn) | The ARN of the KMS Key to for encryption at rest | `string` | `null` | no |
+| <a name="input_limits_per_label_set"></a> [limits\_per\_label\_set](#input\_limits\_per\_label\_set) | Configuration block for setting limits on metrics with specific label sets | <pre>list(object({<br/>    label_set = map(string)<br/>    limits = object({<br/>      max_series = number<br/>    })<br/>  }))</pre> | `null` | no |
 | <a name="input_logging_configuration"></a> [logging\_configuration](#input\_logging\_configuration) | The logging configuration of the prometheus workspace. | <pre>object({<br/>    create_log_group      = optional(bool, true)<br/>    logging_configuration = optional(string)<br/>  })</pre> | `null` | no |
 | <a name="input_region"></a> [region](#input\_region) | Region where the resource(s) will be managed. Defaults to the Region set in the provider configuration | `string` | `null` | no |
+| <a name="input_retention_period_in_days"></a> [retention\_period\_in\_days](#input\_retention\_period\_in\_days) | Number of days to retain metric data in the workspace | `number` | `null` | no |
 | <a name="input_rule_group_namespaces"></a> [rule\_group\_namespaces](#input\_rule\_group\_namespaces) | A map of one or more rule group namespace definitions | <pre>map(object({<br/>    name = string<br/>    data = string<br/>  }))</pre> | `null` | no |
 | <a name="input_tags"></a> [tags](#input\_tags) | A map of tags to add to all resources | `map(string)` | `{}` | no |
 | <a name="input_workspace_alias"></a> [workspace\_alias](#input\_workspace\_alias) | The alias of the prometheus workspace. See more in the [AWS Docs](https://docs.aws.amazon.com/prometheus/latest/userguide/AMP-onboard-create-workspace.html) | `string` | `null` | no |

examples/complete/README.md

@@ -24,7 +24,7 @@ Note that this example may create resources which will incur monetary charges on
 | Name | Version |
 |------|---------|
 | <a name="requirement_terraform"></a> [terraform](#requirement\_terraform) | >= 1.5.7 |
-| <a name="requirement_aws"></a> [aws](#requirement\_aws) | >= 6.0 |
+| <a name="requirement_aws"></a> [aws](#requirement\_aws) | >= 6.22 |
 
 ## Providers
 

examples/complete/main.tf

@@ -21,6 +21,27 @@ module "prometheus" {
     # log_group_arn = "${aws_cloudwatch_log_group.this.arn}:*"
   }
 
+  retention_period_in_days = 60
+
+  limits_per_label_set = [
+    {
+      label_set = {
+        "env" = "dev"
+      }
+      limits = {
+        max_series = 100000
+      }
+    },
+    {
+      label_set = {
+        "env" = "prod"
+      }
+      limits = {
+        max_series = 400000
+      }
+    }
+  ]
+
   create_alert_manager     = true
   alert_manager_definition = <<-EOT
   alertmanager_config: |

examples/complete/versions.tf

@@ -4,7 +4,7 @@ terraform {
   required_providers {
     aws = {
       source  = "hashicorp/aws"
-      version = ">= 6.0"
+      version = ">= 6.22"
     }
   }
 }

main.tf

@@ -25,6 +25,35 @@ resource "aws_prometheus_workspace" "this" {
   tags = var.tags
 }
 
+################################################################################
+# Workspace Configuration
+################################################################################
+
+resource "aws_prometheus_workspace_configuration" "this" {
+  count = var.create && var.create_workspace ? 1 : 0
+
+  region = var.region
+
+  retention_period_in_days = var.retention_period_in_days
+  workspace_id             = local.workspace_id
+
+  dynamic "limits_per_label_set" {
+    for_each = var.limits_per_label_set
+
+    content {
+      label_set = limits_per_label_set.value.label_set
+
+      dynamic "limits" {
+        for_each = limits_per_label_set.value.limits
+
+        content {
+          max_series = limits.value.max_series
+        }
+      }
+    }
+  }
+}
+
 ################################################################################
 # Cloudwatch Log Group
 ################################################################################

variables.tf

@@ -53,6 +53,27 @@ variable "kms_key_arn" {
   default     = null
 }
 
+################################################################################
+# Workspace Configuration
+################################################################################
+
+variable "retention_period_in_days" {
+  description = "Number of days to retain metric data in the workspace"
+  type        = number
+  default     = null
+}
+
+variable "limits_per_label_set" {
+  description = "Configuration block for setting limits on metrics with specific label sets"
+  type = list(object({
+    label_set = map(string)
+    limits = object({
+      max_series = number
+    })
+  }))
+  default = null
+}
+
 ################################################################################
 # CloudWatch Log Group
 ################################################################################

versions.tf

@@ -4,7 +4,7 @@ terraform {
   required_providers {
     aws = {
       source  = "hashicorp/aws"
-      version = ">= 6.0"
+      version = ">= 6.22"
     }
   }
 }