Worker group tags

CHANGELOG.md

@@ -12,6 +12,7 @@ project adheres to [Semantic Versioning](http://semver.org/).
 ##### Added
 
 - Ability to configure force_delete for the worker group ASG (by @stefansedich)
+- Ability to configure worker group ASG tags (by @stefansedich)
 
 ##### Changed
 

examples/eks_test_fixture/main.tf

@@ -39,6 +39,24 @@ locals {
   #   },
   # ]
 
+
+  # the commented out worker group tags below shows an example of how to define
+  # custom tags for the worker groups ASG
+  # worker_group_tags = [
+  #   [
+  #     {
+  #       key   = "k8s.io/cluster-autoscaler/node-template/taint/nvidia.com/gpu"
+  #       value = "gpu:NoSchedule"
+  #     },
+  #   ],
+  #   [
+  #     {
+  #       key   = "k8s.io/cluster-autoscaler/node-template/taint/nvidia.com/gpu"
+  #       value = "gpu:NoSchedule"
+  #     },
+  #   ],
+  # ]
+
   worker_groups = [
     {
       # This will launch an autoscaling group with only On-Demand instances

variables.tf

@@ -101,6 +101,13 @@ variable "workers_group_defaults" {
   default     = {}
 }
 
+variable "worker_group_tags" {
+  description = "A list of lists containing maps defining extra tags to be applied to the worker group ASG."
+  type        = "list"
+
+  default = [[]]
+}
+
 variable "worker_groups_launch_template" {
   description = "A list of maps defining worker group configurations to be defined using AWS Launch Templates. See workers_group_defaults for valid keys."
   type        = "list"
@@ -124,6 +131,13 @@ variable "workers_group_launch_template_defaults" {
   default     = {}
 }
 
+variable "worker_group_launch_template_tags" {
+  description = "A list of lists containing maps defining extra tags to be applied to the worker group template ASG."
+  type        = "list"
+
+  default = [[]]
+}
+
 variable "worker_security_group_id" {
   description = "If provided, all workers will be attached to this security group. If not given, a security group will be created with necessary ingres/egress to work with the EKS cluster."
   default     = ""

workers.tf

@@ -19,7 +19,8 @@ resource "aws_autoscaling_group" "workers" {
       map("key", "kubernetes.io/cluster/${aws_eks_cluster.this.name}", "value", "owned", "propagate_at_launch", true),
       map("key", "k8s.io/cluster-autoscaler/${lookup(var.worker_groups[count.index], "autoscaling_enabled", local.workers_group_defaults["autoscaling_enabled"]) == 1 ? "enabled" : "disabled"  }", "value", "true", "propagate_at_launch", false)
     ),
-    local.asg_tags)
+    local.asg_tags,
+    list(var.worker_group_tags[length(var.worker_group_tags) > count.index ? count.index : "0"]))
   }"]
 
   lifecycle {

workers_launch_template.tf

@@ -45,7 +45,8 @@ resource "aws_autoscaling_group" "workers_launch_template" {
       map("key", "kubernetes.io/cluster/${aws_eks_cluster.this.name}", "value", "owned", "propagate_at_launch", true),
       map("key", "k8s.io/cluster-autoscaler/${lookup(var.worker_groups_launch_template[count.index], "autoscaling_enabled", local.workers_group_launch_template_defaults["autoscaling_enabled"]) == 1 ? "enabled" : "disabled"  }", "value", "true", "propagate_at_launch", false)
     ),
-    local.asg_tags)
+    local.asg_tags,
+    list(var.worker_group_launch_template_tags[length(var.worker_group_launch_template_tags) > count.index ? count.index : "0"]))
   }"]
 
   lifecycle {