Add input option for taints in nodegroups

Author: OlamideOl1

aws/cluster/main.tf

@@ -48,6 +48,7 @@ module "node_groups" {
   role           = module.node_role.instance
   subnets        = values(data.aws_subnet.private)
   tags           = var.tags
+  taints         = each.value.taints
 
   depends_on = [module.node_role]
 }

aws/cluster/modules/eks-node-group/main.tf

@@ -20,6 +20,15 @@ resource "aws_eks_node_group" "this" {
   lifecycle {
     ignore_changes = [scaling_config[0].desired_size]
   }
+
+  dynamic "taint" {
+    for_each = var.taints
+    content {
+      key    = taint.value["key"]
+      value  = taint.value["value"]
+      effect = taint.value["effect"]
+    }
+  }
 }
 
 locals {

aws/cluster/modules/eks-node-group/variables.tf

@@ -39,6 +39,20 @@ variable "subnets" {
   type        = list(object({ id = string, availability_zone = string }))
   description = "Subnets in which the node group should run"
 }
+variable "taints" {
+  type = list(object({
+    key    = string
+    value  = optional(string)
+    effect = string
+  }))
+  description = <<-EOT
+    List of `key`, `value`, `effect` objects representing Kubernetes taints.
+    `effect` must be one of `NO_SCHEDULE`, `NO_EXECUTE`, or `PREFER_NO_SCHEDULE`.
+    `key` and `effect` are required, `value` may be null.
+    EOT
+  default     = []
+  nullable    = false
+}
 
 variable "tags" {
   type        = map(string)

aws/cluster/variables.tf

@@ -33,9 +33,13 @@ variable "node_groups" {
     instance_types = list(string),
     max_size       = number
     min_size       = number
+    taints         = optional(list(object({
+      key    = string
+      value  = optional(string)
+      effect = string
+    })), [])
   }))
 }
-
 variable "tags" {
   type        = map(string)
   description = "Tags to be applied to all created resources"