Increase AMI builder capacity

main.tf

@@ -46,10 +46,11 @@ locals {
     ? "nixos/*-${var.system}"
     : var.base_nixos_ami_name_pattern
   )
-  base_nixos_ami_id = var.base_nixos_ami_id == null ? data.aws_ami.nixos[0].id : var.base_nixos_ami_id
-  resource_name     = "${var.name}-builder-${local.aws_architecture}-${substr(var.input_hash, 0, 12)}"
-  nix_device_id     = replace(aws_ebs_volume.nix.id, "-", "")
-  nix_device        = "/dev/disk/by-id/nvme-Amazon_Elastic_Block_Store_${local.nix_device_id}"
+  builder_instance_type = var.builder_instance_type == null ? (var.system == "aarch64-linux" ? "c7g.2xlarge" : "c7i.2xlarge") : var.builder_instance_type
+  base_nixos_ami_id     = var.base_nixos_ami_id == null ? data.aws_ami.nixos[0].id : var.base_nixos_ami_id
+  resource_name         = "${var.name}-builder-${local.aws_architecture}-${substr(var.input_hash, 0, 12)}"
+  nix_device_id         = replace(aws_ebs_volume.nix.id, "-", "")
+  nix_device            = "/dev/disk/by-id/nvme-Amazon_Elastic_Block_Store_${local.nix_device_id}"
   tags = {
     Name          = local.resource_name
     Project       = var.name
@@ -114,6 +115,8 @@ resource "aws_ebs_volume" "nix" {
   availability_zone = data.aws_subnet.selected.availability_zone
   size              = var.nix_volume_size_gb
   type              = "gp3"
+  iops              = var.nix_volume_iops
+  throughput        = var.nix_volume_throughput
 
   tags = merge(local.tags, {
     Name = "${local.resource_name}-nix"
@@ -122,11 +125,12 @@ resource "aws_ebs_volume" "nix" {
 
 resource "aws_instance" "builder" {
   ami                         = local.base_nixos_ami_id
-  instance_type               = var.builder_instance_type
+  instance_type               = local.builder_instance_type
   subnet_id                   = local.subnet_id
   vpc_security_group_ids      = [aws_security_group.builder.id]
   iam_instance_profile        = aws_iam_instance_profile.builder.name
   associate_public_ip_address = true
+  ebs_optimized               = true
   key_name                    = var.ssh_public_key == null ? null : aws_key_pair.admin[0].key_name
   user_data_replace_on_change = true
 
@@ -143,6 +147,8 @@ resource "aws_instance" "builder" {
   root_block_device {
     volume_size = var.root_volume_size_gb
     volume_type = "gp3"
+    iops        = var.root_volume_iops
+    throughput  = var.root_volume_throughput
   }
 
   tags = local.tags