adding additional capabilities and identity blocks to Linux VM scale set

Author: kumarvna

main.tf

@@ -271,6 +271,13 @@ resource "azurerm_linux_virtual_machine_scale_set" "linux_vmss" {
     write_accelerator_enabled = var.enable_os_disk_write_accelerator
   }
 
+  dynamic "additional_capabilities" {
+    for_each = var.enable_ultra_ssd_data_disk_storage_support ? [1] : []
+    content {
+      ultra_ssd_enabled = var.enable_ultra_ssd_data_disk_storage_support
+    }
+  }
+
   dynamic "data_disk" {
     for_each = var.additional_data_disks
     content {
@@ -333,6 +340,14 @@ resource "azurerm_linux_virtual_machine_scale_set" "linux_vmss" {
     }
   }
 
+  dynamic "identity" {
+    for_each = var.managed_identity_type != null ? [1] : []
+    content {
+      type         = var.managed_identity_type
+      identity_ids = var.managed_identity_type == "UserAssigned" || var.managed_identity_type == "SystemAssigned, UserAssigned" ? var.managed_identity_ids : null
+    }
+  }
+
   dynamic "boot_diagnostics" {
     for_each = var.enable_boot_diagnostics ? [1] : []
     content {

variables.tf

@@ -356,6 +356,11 @@ variable "enable_os_disk_write_accelerator" {
   default     = false
 }
 
+variable "enable_ultra_ssd_data_disk_storage_support" {
+  description = "Should the capacity to enable Data Disks of the UltraSSD_LRS storage account type be supported on this Virtual Machine"
+  default     = false
+}
+
 variable "additional_data_disks" {
   description = "Adding additional disks capacity to add each instance (GB)"
   type        = list(number)
@@ -408,7 +413,15 @@ variable "enable_automatic_instance_repair" {
   default     = false
 }
 
+variable "managed_identity_type" {
+  description = "The type of Managed Identity which should be assigned to the Linux Virtual Machine Scale Set. Possible values are `SystemAssigned`, `UserAssigned` and `SystemAssigned, UserAssigned`"
+  default     = null
+}
 
+variable "managed_identity_ids" {
+  description = " A list of User Managed Identity ID's which should be assigned to the Linux Virtual Machine Scale Set."
+  default     = null
+}
 
 variable "enable_boot_diagnostics" {
   description = "Should the boot diagnostics enabled?"