Public/Enable-AvdDiagnostics.ps1
|
function Enable-AvdDiagnostics { <# .SYNOPSIS Enables the AVD Diagnostics and will send it to a new LogAnalytics workspace .DESCRIPTION .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 Workspace Enter the name(s) of the AVD workspaces .PARAMETER LAWorkspace Enter the name of the Log Analytics Workspace .EXAMPLE Enable-AvdDiagnostics -HostPoolName avd-hostpool-001 -ResourceGroupName rg-avd-001 -AvdWorkspace avd-workpace-001 #> [CmdletBinding(DefaultParameterSetName = 'Existing')] param ( [parameter(Mandatory)] [parameter(ParameterSetName = 'Initial')] [parameter(ParameterSetName = 'Existing')] [ValidateNotNullOrEmpty()] [string]$HostpoolName, [parameter(Mandatory)] [parameter(ParameterSetName = 'Initial')] [parameter(ParameterSetName = 'Existing')] [ValidateNotNullOrEmpty()] [string]$ResourceGroupName, [parameter()] [parameter(ParameterSetName = 'Initial')] [parameter(ParameterSetName = 'Existing')] [string]$AvdWorkspace, [parameter(ParameterSetName = 'Initial')] [parameter(ParameterSetName = 'Existing')] [string]$LAWorkspace, [parameter(ParameterSetName = 'Initial')] [string]$LASku = "Standard", [parameter(ParameterSetName = 'Initial')] [parameter(ParameterSetName = 'Existing')] [string]$LaResourceGroupName, [parameter(ParameterSetName = 'Initial')] [string]$diagnosticsName = "AVD-Diagnostics", [parameter(ParameterSetName = 'Initial')] [parameter(ParameterSetName = 'Existing')] [ValidateSet("Checkpoint", "Error", "Management", "Connection", "HostRegistration", "AgentHealthStatus")] [array]$Categories, [parameter(ParameterSetName = 'Initial')] [int]$RetentionInDays ) Begin { AuthenticationCheck } Process { switch ($PsCmdlet.ParameterSetName) { Initial { if ($null -eq $LAWorkspace) { $LAWorkspace = "log-analytics-avd-" + (Get-Random -Maximum 99999) Write-Verbose "No Log Analytics Workspace provided, creating a new one." Write-Verbose "Workspace name: $LAWorkspace" } $Body = @{ location = $LaLocation properties = @{ retentionInDays = $RetentionInDays sku = @{ name = $LASku } } } $url = "https://management.azure.com/subscriptions/" + $script:subscriptionId + "/resourceGroups/" + $LaResourceGroupName + "/providers/Microsoft.OperationalInsights/workspaces/" + $LAWorkspace + "?api-version=2020-08-01" $loganalyticsParameters = @{ URI = $url Method = "PUT" Body = $Body | ConvertTo-Json Headers = $token } $laws = Invoke-RestMethod @loganalyticsParameters } Existing { $url = "https://management.azure.com/subscriptions/" + $script:subscriptionId + "/resourceGroups/" + $LaResourceGroupName + "/providers/Microsoft.OperationalInsights/workspaces/" + $LAWorkspace + "?api-version=2020-08-01" $loganalyticsParameters = @{ URI = $url Method = "GET" Headers = $token } $laws = Invoke-RestMethod @loganalyticsParameters } } $parameters = @{ HostPoolName = $HostpoolName ResourceGroupName = $ResourceGroupName } $Hostpool = Get-AzWvdHostPool @parameters $categoryArray = @() $Categories | foreach { $category = @{ Category = $_ Enabled = $true } $categoryArray += ($category) } $diagnosticsBody = @{ Properties = @{ workspaceId = $Laws.id logs = $categoryArray } } $parameters = @{ uri = "https://management.azure.com/$($Hostpool.Id)/providers/microsoft.insights/diagnosticSettings/$($diagnosticsName)?api-version=2017-05-01-preview" Method = "PUT" Headers = $token Body = $diagnosticsBody | ConvertTo-Json -Depth 4 } $results = Invoke-RestMethod @parameters return $results Write-Verbose "Diagnostics enabled for $HostpoolName, sending info to $LAWorkspace" } } |