mirror of
https://github.com/actions/runner-images.git
synced 2025-12-20 06:35:47 +00:00
86adee10d0
* scheduling condition added and changed VMs specs for macos * typo * changed conditions order * fixed an error * added spec parameters that use pipeline variables values Co-authored-by: Leonid Lapshin <originalnoe-nazvanie@yandex.ru>
111 lines
3.1 KiB
PowerShell
111 lines
3.1 KiB
PowerShell
<#
|
|
.SYNOPSIS
|
|
|
|
This script selects local datastore based on the following rules:
|
|
|
|
- Name starts with ds-local-Datastore
|
|
- Datastore FreespaceGB > 400 Gb
|
|
- VM count on the datastore < 2
|
|
|
|
.PARAMETER VIServer
|
|
vCenter address (Example "10.0.1.16")
|
|
|
|
.PARAMETER VIUserName
|
|
vCenter username (Example "Administrator")
|
|
|
|
.PARAMETER VIPassword
|
|
vCenter password (Example "12345678")
|
|
#>
|
|
|
|
|
|
[CmdletBinding()]
|
|
param(
|
|
[Parameter(Mandatory)]
|
|
[ValidateNotNullOrEmpty()]
|
|
[string]$VMName,
|
|
|
|
[Parameter(Mandatory)]
|
|
[ValidateNotNullOrEmpty()]
|
|
[string]$VIServer,
|
|
|
|
[Parameter(Mandatory)]
|
|
[ValidateNotNullOrEmpty()]
|
|
[string]$VIUserName,
|
|
|
|
[Parameter(Mandatory)]
|
|
[ValidateNotNullOrEmpty()]
|
|
[string]$VIPassword,
|
|
|
|
[string]$TagCategory = "Busy"
|
|
)
|
|
|
|
# Import helpers module
|
|
Import-Module $PSScriptRoot\helpers.psm1 -DisableNameChecking
|
|
|
|
function Select-DataStore {
|
|
param (
|
|
[string]$VMName,
|
|
[string]$TagCategory,
|
|
[string]$TemplateDatastore = "ds-local-Datastore-*",
|
|
[int]$ThresholdInGb = 400,
|
|
[int]$VMCount = 2,
|
|
[int]$Retries = 5
|
|
)
|
|
|
|
# 1. Name starts with ds-local-Datastore
|
|
# 2. FreespaceGB > 400 Gb
|
|
# 3. Choose a datastore with the minimal VM count < 2
|
|
|
|
Write-Host "Start Datastore selection process..."
|
|
$allDatastores = Get-Datastore -Name $templateDatastore | Where-Object { $_.State -eq "Available" }
|
|
$buildDatastore = $allDatastores `
|
|
| Where-Object { $_.FreeSpaceGB -ge $thresholdInGb } `
|
|
| Where-Object {
|
|
$vmOnDatastore = @((Get-ChildItem -Path $_.DatastoreBrowserPath).Name -notmatch "^\.").Count
|
|
$vmOnDatastore -lt $vmCount } `
|
|
| Group-Object -Property { $vmOnDatastore } `
|
|
| Select-Object -First 1 -ExpandProperty Group `
|
|
| Get-Random `
|
|
| Select-Object -ExpandProperty Name
|
|
|
|
$tag = Get-Tag -Category $TagCategory -Name $VMName -ErrorAction Ignore
|
|
if (-not $tag)
|
|
{
|
|
$tag = New-Tag -Name $VMName -Category $TagCategory
|
|
}
|
|
|
|
New-TagAssignment -Tag $tag -Entity $buildDatastore | Out-Null
|
|
|
|
# Wait for 60 seconds to check if any other tags are assigned to the same datastore
|
|
Start-Sleep -Seconds 60
|
|
# Take only first 2 tags, all the others will go to the next round
|
|
$tagAssignments = (Get-TagAssignment -Entity $buildDatastore).Tag.Name | Select-Object -First 2
|
|
$isAllow = $tagAssignments -contains $VMName
|
|
|
|
if ($isAllow)
|
|
{
|
|
Write-Host "Datastore selected successfully"
|
|
Write-Host "##vso[task.setvariable variable=buildDatastore;issecret=true]$buildDatastore"
|
|
return
|
|
}
|
|
|
|
# Remove the tag if datastore wasn't selected
|
|
Remove-Tag $tag -Confirm:$false
|
|
|
|
$retries--
|
|
if ($retries -le 0)
|
|
{
|
|
Write-Host "##vso[task.LogIssue type=error;]No datastores found for the condition"
|
|
exit 1
|
|
}
|
|
|
|
Write-Host "Datastore select failed, $retries left"
|
|
Select-DataStore -VMName $VMName -TagCategory $TagCategory -Retries $retries
|
|
}
|
|
|
|
# Connection to a vCenter Server system
|
|
Connect-VCServer -VIServer $VIServer -VIUserName $VIUserName -VIPassword $VIPassword
|
|
|
|
# Get a target datastore for current deployment
|
|
Select-DataStore -VMName $VMName -TagCategory $TagCategory
|