Pass in custom metadata options to the eks launch template

thoughtbot · Sep 25, 2024 · d35aca0 · d35aca0
1 parent a094ef5
commit d35aca0
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 16 deletions.
diff --git a/aws/cluster/main.tf b/aws/cluster/main.tf
@@ -41,19 +41,20 @@ module "node_groups" {
   for_each = var.node_groups
   source   = "./modules/eks-node-group"
 
-  capacity_type   = each.value.capacity_type
-  cluster         = module.eks_cluster.instance
-  instance_types  = each.value.instance_types
-  labels          = var.labels
-  max_size        = each.value.max_size
-  max_unavailable = each.value.max_unavailable
-  min_size        = each.value.min_size
-  name            = each.key
-  namespace       = [module.cluster_name.full]
-  role            = module.node_role.instance
-  subnets         = values(data.aws_subnet.private)
-  tags            = var.tags
-  user_data      = var.user_data
+  capacity_type    = each.value.capacity_type
+  cluster          = module.eks_cluster.instance
+  instance_types   = each.value.instance_types
+  labels           = var.labels
+  max_size         = each.value.max_size
+  max_unavailable  = each.value.max_unavailable
+  metadata_options = var.metadata_options
+  min_size         = each.value.min_size
+  name             = each.key
+  namespace        = [module.cluster_name.full]
+  role             = module.node_role.instance
+  subnets          = values(data.aws_subnet.private)
+  tags             = var.tags
+  user_data        = lookup(var.user_data, each.key, null)
 
   depends_on = [module.node_role]
 }

diff --git a/aws/cluster/modules/eks-node-group/main.tf b/aws/cluster/modules/eks-node-group/main.tf
@@ -9,7 +9,7 @@ resource "aws_eks_node_group" "this" {
   subnet_ids      = [each.value.id]
 
   dynamic "launch_template" {
-    for_each = var.user_data != null ? [aws_launch_template.this[0]] : []
+    for_each = var.metadata_options != {} ? [aws_launch_template.this[0]] : []
 
     content {
       id      = launch_template.value.id
@@ -40,10 +40,22 @@ resource "aws_eks_node_group" "this" {
   }
 }
 
+# resource "aws_launch_template" "this" {
+#   count = var.user_data != null ? 1 : 0
+
+#   user_data = base64encode(var.user_data)
+# }
+
 resource "aws_launch_template" "this" {
-  count = var.user_data != null ? 1 : 0
+  count = var.metadata_options != {} ? 1 : 0
 
-  user_data = base64encode(var.user_data)
+  metadata_options {
+    http_endpoint               = lookup(var.metadata_options, "http_endpoint", "enabled")
+    http_tokens                 = lookup(var.metadata_options, "http_tokens", "optional")
+    http_put_response_hop_limit = lookup(var.metadata_options, "htthttp_put_response_hop_limit", "1")
+    http_protocol_ipv6          = lookup(var.metadata_options, "http_protocol_ipv6", "disabled")
+    instance_metadata_tags      = lookup(var.metadata_options, "instance_metadata_tags", "disabled")
+  }
 }
 
 locals {
@@ -54,4 +66,6 @@ locals {
     var.subnets[*].availability_zone,
     var.subnets
   )
+
+
 }
diff --git a/aws/cluster/modules/eks-node-group/variables.tf b/aws/cluster/modules/eks-node-group/variables.tf
@@ -75,3 +75,9 @@ variable "user_data" {
   description = "Optional user data script for the launch template"
   default     = null # Default to an empty string if no user data is provided
 }
+
+variable "metadata_options" {
+  type        = map(string)
+  description = "Customize the metadata options for the cluster nodes"
+  default     = {}
+}
diff --git a/aws/cluster/variables.tf b/aws/cluster/variables.tf
@@ -67,3 +67,9 @@ variable "user_data" {
   description = "Optional user data script for the launch template"
   default     = {} # Default to an empty string if no user data is provided
 }
+
+variable "metadata_options" {
+  type        = map(string)
+  description = "Customize the metadata options for the cluster nodes"
+  default     = {}
+}