Terraform如何定义变量让用户选择使用现有azurerm_storage_account或新建

核心通过Terraform的count元参数控制资源/数据源的启用状态，配合条件判断即可实现选择新建或使用现有存储账户的需求，具体实现步骤如下：

步骤1：补充必要变量 #

除了你已经定义的choice变量外，需要额外补充现有存储账户的关联参数，设置为可选即可，用户选择新建时无需填写：

variable "choice" {
  description = "允许用户选择使用现有存储账户或新建存储账户，可选值为create/use_existing"
  type        = string
  default     = "create"
  # 可选增加参数合法性校验
  validation {
    condition     = contains(["create", "use_existing"], var.choice)
    error_message = "choice参数仅允许填入create或use_existing。"
  }
}

variable "existing_storage_account_name" {
  description = "现有存储账户名称，仅当choice为use_existing时需要填写"
  type        = string
  default     = ""
  # 可选增加必填校验
  validation {
    condition     = var.choice == "use_existing" ? length(var.existing_storage_account_name) > 0 : true
    error_message = "当choice参数为use_existing时，必须填写existing_storage_account_name参数。"
  }
}

variable "existing_storage_account_resource_group_name" {
  description = "现有存储账户所属资源组名称，仅当choice为use_existing时需要填写"
  type        = string
  default     = ""
  validation {
    condition     = var.choice == "use_existing" ? length(var.existing_storage_account_resource_group_name) > 0 : true
    error_message = "当choice参数为use_existing时，必须填写existing_storage_account_resource_group_name参数。"
  }
}

步骤2：控制新建存储账户的创建逻辑 #

给azurerm_storage_account资源增加count参数，仅当choice为create时才创建该资源：

resource "azurerm_storage_account" "example" {
  count = var.choice == "create" ? 1 : 0

  name                     = "storageaccountname" # 此处也可替换为自定义变量，按需调整即可
  resource_group_name      = azurerm_resource_group.example.name
  location                 = azurerm_resource_group.example.location
  account_tier             = "Standard"
  account_replication_type = "GRS"

  tags = {
    environment = "staging"
  }
}

步骤3：添加现有存储账户的数据源 #

新增azurerm_storage_account数据源，同样用count控制启用状态，仅当choice为use_existing时才拉取现有存储账户的信息：

data "azurerm_storage_account" "existing" {
  count = var.choice == "use_existing" ? 1 : 0

  name                = var.existing_storage_account_name
  resource_group_name = var.existing_storage_account_resource_group_name
}

步骤4：统一引用存储账户属性 #

后续需要使用存储账户属性时，通过条件判断统一取值即可，无需写两套逻辑：

# 示例：通过本地值统一封装存储账户的常用属性，方便全局引用
locals {
  storage_account = var.choice == "create" ? azurerm_storage_account.example[0] : data.azurerm_storage_account.existing[0]
}

# 后续引用示例
output "storage_account_id" {
  value = local.storage_account.id
}

output "storage_account_connection_string" {
  value = local.storage_account.primary_connection_string
}

内容的提问来源于stack exchange，提问作者BlueJapan