AzureAS.AutoScheduler.ps1

param(     
    [string] $resourceGroupName = "BITabularModel",
    [string] $serverName = "guardiantabular",
    [string] $azureProfilePath  = "",
    [string] $azureRunAsConnectionName = "AzureRunAsConnection",
    [string] $configStr = '
[
    {
	    WeekDays:[1,2,3,4,5]	
	    ,StartTime: "06:00:00"
	    ,StopTime: "18:59:59"
	    ,Sku: "S1"
	}
    ,
    {
	    WeekDays:[1,2,3,4,5]	
	    ,StartTime: "19:00:00"
	    ,StopTime: "23:59:59"
	    ,Sku: "S0"
	}
    ,
    {
	    WeekDays:[6, 0]	
	    ,StartTime: "00:00:00"
	    ,StopTime: "23:59:59"
	    ,Sku: "S0"
	}    
]
'
)

$VerbosePreference = "Continue"
$ErrorActionPreference = "Stop"

Import-Module "AzureRM.AnalysisServices" 

Write-Verbose "Logging in to Azure..."

# Load the profile from local file
if (-not [string]::IsNullOrEmpty($azureProfilePath))
{    
    Import-AzureRmContext -Path $azureProfilePath | Out-Null
}
# Load the profile from Azure Automation RunAS connection
elseif (-not [string]::IsNullOrEmpty($azureRunAsConnectionName))
{
    $runAsConnectionProfile = Get-AutomationConnection -Name $azureRunAsConnectionName      

    Add-AzureRmAccount -ServicePrincipal -TenantId $runAsConnectionProfile.TenantId `
        -ApplicationId $runAsConnectionProfile.ApplicationId -CertificateThumbprint $runAsConnectionProfile.CertificateThumbprint | Out-Null
}
# Interactive Login
else
{
    Add-AzureRmAccount | Out-Null
}

$stateConfig = $configStr | ConvertFrom-Json

$inputStartTime = Get-Date

$fromTimeZone = "UTC" # Default timezone for Powershell runbooks
$toTimeZone = "Eastern Standard Time"

$startTime = [System.TimeZoneInfo]::ConvertTimeBySystemTimeZoneId($inputStartTime.DateTime, $fromTimeZone, $toTimeZone)

$currentDayOfWeek = [Int]($startTime).DayOfWeek

# Find a match in the config

$dayObjects = $stateConfig | Where-Object {$_.WeekDays -contains $currentDayOfWeek } `
    | Select-Object Sku, @{Name="StartTime"; Expression = {[datetime]::ParseExact($_.StartTime,"HH:mm:ss", [System.Globalization.CultureInfo]::InvariantCulture)}}, @{Name="StopTime"; Expression = {[datetime]::ParseExact($_.StopTime,"HH:mm:ss", [System.Globalization.CultureInfo]::InvariantCulture)}}

# Get the server status

$asServer = Get-AzureRmAnalysisServicesServer -ResourceGroupName $resourceGroupName -Name $serverName

Write-Verbose "Current Azure AS status: $($asServer.State)"

# If not match any day then exit

if($dayObjects -ne $null){

    # Can't treat several objects for same time-frame, if there's more than one, pick first

    $matchingObject = $dayObjects | Where-Object { ($startTime -ge $_.StartTime) -and ($startTime -lt $_.StopTime) } | Select-Object -First 1

    if($matchingObject -ne $null)
    {
        Write-Verbose "Is in active hours"

        # if Paused resume

        if($asServer.State -eq "Paused")
        {
            $asServer | Resume-AzureRmAnalysisServicesServer -Verbose
        }        
    
        # Change the SKU if needed
    
        if($asServer.Sku.Name -ne $matchingObject.Sku){

            Write-Verbose "Updating AAS server from $($asServer.Sku.Name) to $($matchingObject.Sku)"

            #$asServer | Set-AzureRmAnalysisServicesServer -Sku $matchingObject.Sku
            Set-AzureRmAnalysisServicesServer -Name $asServer.Name -ResourceGroupName $resourceGroupName -Sku $matchingObject.Sku -Verbose
        }
    }
    else {
        Write-Verbose "Is in offline hours"

        if($asServer.State -eq "Succeeded")
        {
            $asServer | Suspend-AzureRmAnalysisServicesServer -Verbose
        } 
    }
}
else
{
    Write-Verbose "No object config for current day of week"
    
    if($asServer.State -eq "Succeeded")
    {
        $asServer | Suspend-AzureRmAnalysisServicesServer -Verbose
    }         

}