AzureAS.AutoScheduler.ps1

param(     
    [string] $resourceGroupName = "SSASAzureTest",
    [string] $serverName = "rrdvsssassales",
    [string] $azureProfilePath  = "",
    [string] $azureRunAsConnectionName = "",
    [string] $configStr = "
[
    {
	    WeekDays:[1,2,3,4,5]	
	    ,StartTime: ""08:00:00""
	    ,StopTime: ""17:59:59""
	    ,Sku: ""S1""
	}
    ,
    {
	    WeekDays:[1,2,3,4,5]	
	    ,StartTime: ""18:00:00""
	    ,StopTime: ""23:59:59""
	    ,Sku: ""S0""
	}
    ,
    {
	    WeekDays:[6, 0]	
	    ,StartTime: ""08: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

$startTime = Get-Date

$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
    }         

}