AIDependency.psm1
|
<#
# Dependency #> function New-Stopwatch { return [System.Diagnostics.Stopwatch]::StartNew(); } <# .Synopsis Short description .DESCRIPTION Long description .EXAMPLE Example of how to use this cmdlet .EXAMPLE # Measure and report the time taken by a command Measure-Command { Connect-AADGraph -tenant $tenantname -graphVer "1.6" -Credentials $AADCredential } | Send-AIDependency -Name "AADGraph" -CommandName "Connect" -DependencyTypeName "AAD" -DependencyKind Other Another example of how to use this cmdlet #> function Send-AIDependency { [CmdletBinding()] [OutputType([void])] Param ( # TimeSpan as measured by measure-command (Pipeline) [Parameter(Mandatory=$false,ValueFromPipeline=$true)] [System.TimeSpan] $TimeSpan, # Stopwatch [Parameter(Mandatory=$false)] [System.Diagnostics.Stopwatch]$StopWatch, #Dependency Name [Parameter(Mandatory=$true)] [string]$Name = "External Operation", [string]$CommandName = $name, #Type of dependecy [ValidateSet("HTTP", "SQL", "Other")] [string]$DependencyKind = "Other", [string]$DependencyTypeName = $name, #HTTP result code [bool]$Success = $true, #Hide the resultcode parameter as it appears to be defunct in 2.3.0 [Parameter(DontShow)] #Resultcode alligs to HHTP result codes #[ValidateRange(0,999)] [int]$ResultCode = 200, #The timestamp for the event; defaults to current date/time $Timestamp = (Get-Date), #The AppInsights Client object to use. [Parameter(Mandatory=$false)] [Microsoft.ApplicationInsights.TelemetryClient] $Client , #Directly flush the AI events to the service [switch] $Flush ) Begin { #Check for a specified AI client if ($Client -eq $null) { If ( ($Global:AISingleton ) -AND ( $Global:AISingleton.Client ) ) { #Use Current Client $Client = $Global:AISingleton.Client } } } Process { #no need to do anything if there is no client if ($Client -eq $null) { Write-Verbose 'No AI Client found' return } #check if a timespan has been provided if ( $StopWatch -eq $null -and $TimeSpan -eq $null ) { Write-Warning "No time provided for dependency" } #get a dependecy object $TelDependency= [Microsoft.ApplicationInsights.DataContracts.DependencyTelemetry]::new() # not HTTP or SQL so Other $TelDependency.DependencyKind = $DependencyKind # UP to Now. $TelDependency.Timestamp = $timestamp #Add information $TelDependency.Name = $Name $TelDependency.CommandName = $CommandName $TelDependency.DependencyTypeName = $DependencyTypeName Try { $TelDependency.ResultCode = $ResultCode } catch { #Resultcode is apperantly removed from AI 2.3.0 If ($ResultCode -ne 200) { Write-Warning "Resultcode cannot be reported" } } $TelDependency.Success = $Success if ($TimeSpan ) { $TelDependency.Duration = $TimeSpan } else { $TelDependency.Duration = $StopWatch.Elapsed } #Send it $Client.TrackDependency($TelDependency) if ($Flush) { $client.Flush() } } } |