Public/Enable-AvdInsightsApplicationGroup.ps1
function Enable-AvdInsightsApplicationGroup { <# .SYNOPSIS Enables the AVD Diagnostics and will send it to a new LogAnalytics workspace .DESCRIPTION The function will enable AVD diagnostics for a application group. It will create a new Log Analytics workspace if no existing workspace is provided. .PARAMETER HostPoolName Enter the name of the hostpool you want to enable start vm on connnect. .PARAMETER ResourceGroupName Enter the name of the resourcegroup where the hostpool resides in. .PARAMETER Id Enter the application group's resource ID. .PARAMETER ApplicationGroupName Enter the application group's name. .PARAMETER ApplicationResourceGroup Enter the application group's resource group. .PARAMETER LASku Enter the name of the Log Analytics SKU .PARAMETER LAWorkspace Enter the name of the Log Analytics Workspace .PARAMETER LaResourceGroupName Enter the name of the Log Analyics Workspace resource group .PARAMETER AdditionalCategories The categories you like extra to save in Log Analytics, beside the mandatory categories for AVD Insights. .PARAMETER RetentionInDays How long should the data be saved .PARAMETER DiagnosticsName Enter the diagnostics display name .PARAMETER AutoCreate Use this switch to auto create a Log Analtyics Workspace .EXAMPLE Enable-AvdInsightsApplicationGroup -HostPoolName avd-hostpool-001 -ResourceGroupName rg-avd-001 -LAWorkspace 'la-avd-workspace' -LaResourceGroupName 'rg-la-01' -DiagnosticsName "AvdInsights" .EXAMPLE Enable-AvdInsightsApplicationGroup -ApplicationGroupName avd-appgroup-01 -ApplicationResourceGroup rg-avd-001 -LAWorkspace 'la-avd-workspace' -LaResourceGroupName 'rg-la-01' -DiagnosticsName "AvdInsights" .EXAMPLE Enable-AvdInsightsApplicationGroup -ApplicationGroupName avd-appgroup-01 -ApplicationResourceGroup rg-avd-001 -LAWorkspace 'la-avd-workspace' -LaResourceGroupName 'rg-la-01' -LAWorkspace 'la-avd-workspace' -LaResourceGroupName 'la-rg' -LaLocation 'westeurope' -RetentionInDays 30 -AutoCreate -DiagnosticsName "AvdInsights" .EXAMPLE Enable-AvdInsightsApplicationGroup -Id /subscription/../applicationgroup/groupname -LAWorkspace 'la-avd-workspace' -LaResourceGroupName 'la-rg' -LaLocation 'westeurope' -RetentionInDays 30 -AutoCreate -DiagnosticsName "AvdInsights" #> [CmdletBinding(DefaultParameterSetName = 'Id')] param ( [parameter(Mandatory, ParameterSetName = 'HostpoolLevel')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [ValidateNotNullOrEmpty()] [string]$HostpoolName, [parameter(Mandatory, ParameterSetName = 'HostpoolLevel')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [ValidateNotNullOrEmpty()] [string]$ResourceGroupName, [parameter(Mandatory, ParameterSetName = 'Id', ValueFromPipelineByPropertyName)] [parameter(Mandatory, ParameterSetName = 'Create-Id')] [ValidateNotNullOrEmpty()] [object]$Id, [parameter(Mandatory, ParameterSetName = 'SingleLevel')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [ValidateNotNullOrEmpty()] [string]$ApplicationGroupName, [parameter(Mandatory, ParameterSetName = 'SingleLevel')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [ValidateNotNullOrEmpty()] [string]$ApplicationResourceGroup, [parameter(Mandatory, ParameterSetName = 'HostpoolLevel')] [parameter(Mandatory, ParameterSetName = 'SingleLevel')] [parameter(Mandatory, ParameterSetName = 'Id')] [parameter(Mandatory, ParameterSetName = 'Create-Id')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [string]$LAWorkspace, [parameter(ParameterSetName = 'Create-Id')] [parameter(ParameterSetName = 'Create-Friendly')] [parameter(ParameterSetName = 'Create-SingleLevel')] [ValidateSet("CapacityReservation", "Free", "LACluster", "PerGB2018", "PerNode", "Premium", "Standalone", "Standard")] [string]$LASku = "Standard", [parameter(Mandatory, ParameterSetName = 'HostpoolLevel')] [parameter(Mandatory, ParameterSetName = 'SingleLevel')] [parameter(Mandatory, ParameterSetName = 'Id')] [parameter(Mandatory, ParameterSetName = 'Create-Id')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [string]$LaResourceGroupName, [parameter(Mandatory, ParameterSetName = 'Create-Id')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [string]$LaLocation, [parameter(Mandatory, ParameterSetName = 'Create-Id')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [int]$RetentionInDays, [parameter(Mandatory, ParameterSetName = 'HostpoolLevel')] [parameter(Mandatory, ParameterSetName = 'Id')] [parameter(Mandatory, ParameterSetName = 'Create-Id')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [string]$DiagnosticsName, [parameter(Mandatory, ParameterSetName = 'Create-Id')] [parameter(Mandatory, ParameterSetName = 'Create-Friendly')] [parameter(Mandatory, ParameterSetName = 'Create-SingleLevel')] [switch]$AutoCreate ) Begin { Write-Verbose "[Enable-AvdInsightsApplicationGroup] - Start enabling AVD Insights for application group" AuthenticationCheck $token = GetAuthToken -resource $global:AzureApiUrl } Process { switch ($PsCmdlet.ParameterSetName) { HostpoolLevel { $parameters = @{ HostPoolName = $HostpoolName ResourceGroupName = $ResourceGroupName } $hostpoolId = Get-AvdHostPool @parameters | Select-Object Id $applicationGroup = Get-AvdApplicationGroup -ResourceGroupName $ResourceGroupName | Where-Object { $hostpoolId.id -in $_.properties.hostpoolArmPath } } SingleLevel { $appGroupParameters = @{ ApplicationGroupName = $ApplicationGroupName ResourceGroupName = $ApplicationResourceGroup } $applicationGroup = Get-AvdApplicationGroup @appGroupParameters } default { Write-Verbose "[Enable-AvdInsightsApplicationGroup] - Got the applicationgroup's resource ID. Thank you for that!" } } Write-Verbose "[Enable-AvdInsightsApplicationGroup] - Looking for workspace" $workspaceId = "/subscriptions/{0}/resourceGroups/{1}/providers/Microsoft.OperationalInsights/workspaces/{2}" -f $global:subscriptionId, $LaResourceGroupName, $LAWorkspace Write-Verbose $workspaceId $laws = (Get-Resource -ResourceId $workspaceId -Verbose).value if ($null -eq $laws) { try { if ($AutoCreate.IsPresent) { Write-Warning "[Enable-AvdInsightsApplicationGroup] - No Log Analytics Workspace found! Creating a new workspace" $laws = New-Workspace -Workspace $LAWorkspace -Sku $LASku -ResourceGroupName $LaResourceGroupName -Location $LaLocation } else { Throw "[Enable-AvdInsightsApplicationGroup] - No workspace found! If it is a new workspace, add -AutoCreate in your command, $_" } } catch { Throw $_ } } else { try { Write-Information "[Enable-AvdInsightsApplicationGroup] - Workspace found, configuring diagnostics" -InformationAction Continue $categoryArray = @() $mandatoryCategories = @("Checkpoint", "Error", "Management") $mandatoryCategories | ForEach-Object { $category = @{ Category = $_ Enabled = $true } $categoryArray += ($category) } $diagnosticsBody = @{ Properties = @{ workspaceId = $laws.id logs = @($categoryArray) } } if ($applicationGroup) { Write-Verbose "[Enable-AvdInsightsApplicationGroup] - Grabbing ID's form application groups" $Id = $applicationGroup.id } $Id | ForEach-Object { $parameters = @{ uri = "{0}{1}/providers/microsoft.insights/diagnosticSettings/{2}?api-version={3}" -f $global:AzureApiUrl, $_, $DiagnosticsName, $global:AvdDiagnosticsApiVersion Method = "PUT" Headers = $token Body = $diagnosticsBody | ConvertTo-Json -Depth 4 } Invoke-RestMethod @parameters Write-Verbose "[Enable-AvdInsightsApplicationGroup] - Diagnostics enabled for $_, sending info to $LAWorkspace" } } catch { Throw "[Enable-AvdInsightsApplicationGroup] - $_" } } } } |