PowerShell Retry Wrapper

Retry.ps1

function WithRetry {
    param (
        [Parameter(Mandatory = $true, ValueFromPipeline = $true)]
        [ValidateNotNull()]
        [ScriptBlock] $Command,

        [Parameter(Mandatory = $false)]
        [int]$RetryCount = 3
    )
    Write-Host "Retry Injection Scope"  
    $i = 0
    do {
        try {
            return Invoke-Command $Command
        }
        catch {
            Write-Host "Retry #$((++$i))" -fore Red
        }
    } until ($i -ge $RetryCount)
}

function Invoke-FlakyCommand {
    param (
        [Parameter(Mandatory = $true, ValueFromPipeline = $true)]
        [int]$Number
    )
    # Retries the command on transient errors.
    WithRetry {
        if (Get-Random -InputObject $true, $false) {
            throw "Oooops"
        }
        return $Number
    }
}

(1..5) | ForEach-Object { $_ | Invoke-FlakyCommand }