PowerShell循环中If-Else导致vSphere VM标签被重复覆盖

问题描述 #

首次编写PowerShell脚本，使用PowerCLI cmdlet为vSphere中的VM创建并分配标签：读取VM列表到$vms2tag，读取JSON文件获取标签数据，通过判断标签分类、标签是否存在来创建或分配标签。脚本运行时标签可临时正确应用，但执行完毕后所有VM的标签均被最后一个VM的标签覆盖。

原代码如下：

$myJson = Get-Content 'C:\For-Powershell.json'| Out-String | ConvertFrom-Json
$vms2tag = Get-Content 'C:\Output-VM.txt'

foreach ($vm in $vms2tag) {
    For ($j=0; $j -lt $myJson.Length; $j++) {
        if ($vm -eq $myJson.Server[$j]) {
            Write-Output "Match!"

            # Variables for Application Owner
            $nameAO = [string]$myJson.Application_Owner[$j]
            $descriptionAO = [string]$myJson.Application_Owner[$j]


            # check if Tag Category and/or Tag exist

            if ((Get-TagCategory -Name "app_owner") -eq $null) {
                New-TagCategory -Name "app_owner" -Cardinality "Multiple"
            }
            
            if ((Get-Tag -Category "app_owner" | Set-Tag -Name $nameAO -Description $descriptionAO) -eq $null) {
                $myTagAO = Get-TagCategory -Name "app_owner" | New-Tag -Name $nameAO -Description $descriptionAO
                New-TagAssignment -Tag $myTagAO -Entity $myJson.Server[$j]
            }
            else {
                $myTagAO = Get-Tag -Category "app_owner" | Set-Tag -Name $nameAO -Description $descriptionAO
                New-TagAssignment -Tag $myTagAO -Entity $myJson.Server[$j]
            }
        }
    }
}

原因分析 #

核心错误有两点：

1. 错误使用Set-Tag：Get-Tag -Category "app_owner" | Set-Tag -Name $nameAO -Description $descriptionAO这行代码会获取app_owner分类下的所有标签，并将它们的名称和描述全部修改为当前循环的$nameAO值。每处理一个VM，就会把该分类下所有标签改成当前VM对应的标签名，最终所有VM关联的标签都会被替换成最后一次循环修改后的标签。
2. 标签存在性判断逻辑错误：原代码试图通过修改标签后的结果是否为null来判断标签是否存在，这完全不符合逻辑——修改操作本身会改变已有标签，而非判断存在性。

修复后的代码 #

调整逻辑，先检查指定名称的标签是否存在，避免修改所有标签，同时优化循环效率：

# 读取并转换JSON数据，转为哈希表方便快速查找
$myJson = Get-Content 'C:\For-Powershell.json' | ConvertFrom-Json
$vmTagMap = @{}
foreach ($item in $myJson) {
    $vmTagMap[$item.Server] = $item.Application_Owner
}

$vms2tag = Get-Content 'C:\Output-VM.txt'

# 确保标签分类存在
$tagCategory = Get-TagCategory -Name "app_owner" -ErrorAction SilentlyContinue
if (-not $tagCategory) {
    $tagCategory = New-TagCategory -Name "app_owner" -Cardinality "Multiple"
}

foreach ($vmName in $vms2tag) {
    if ($vmTagMap.ContainsKey($vmName)) {
        Write-Output "匹配到VM: $vmName"
        $tagName = [string]$vmTagMap[$vmName]
        
        # 检查指定名称的标签是否存在
        $targetTag = Get-Tag -Category $tagCategory -Name $tagName -ErrorAction SilentlyContinue
        
        if (-not $targetTag) {
            # 不存在则创建新标签
            $targetTag = New-Tag -Name $tagName -Description $tagName -Category $tagCategory
        }
        
        # 给VM分配标签（避免重复分配可加判断）
        $existingAssignment = Get-TagAssignment -Entity $vmName -Tag $targetTag -ErrorAction SilentlyContinue
        if (-not $existingAssignment) {
            New-TagAssignment -Tag $targetTag -Entity $vmName
        }
    } else {
        Write-Output "未找到VM $vmName 的标签数据"
    }
}

关键优化点 #

• 将JSON数据转为哈希表$vmTagMap，避免嵌套循环，提升查找效率。
• 使用-ErrorAction SilentlyContinue替代$null判断，更符合PowerCLI的最佳实践。
• 仅针对指定名称的标签进行存在性检查，不会修改其他标签。
• 增加重复标签分配的判断，避免多次执行相同操作。

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