internal void OnTelemetrySent(PSCmdlet pscmdlet) { AzurePSQoSEvent qos; if (_telemetry.TryGetValue(MetricHelper.TelemetryId, out qos)) { qos.IsSuccess = (qos.Exception == null); _telemetry.LogEvent(MetricHelper.TelemetryId); _previousEndTime = DateTimeOffset.Now; pscmdlet.WriteDebug(qos.ToString()); } }
internal async Task OnCmdletException(string id, CancellationToken cancellationToken, GetEventData getEventData, SignalDelegate signal, string processRecordId, Exception exception) { var data = EventDataConverter.ConvertFrom(getEventData()); await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Received Exception with message '{data?.Message}'")); AzurePSQoSEvent qos; if (_telemetry.TryGetValue(processRecordId, out qos)) { await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Sending new QosEvent for command '{qos.CommandName}': {qos.ToString()}")); qos.IsSuccess = false; qos.Exception = exception; _telemetry.LogEvent(processRecordId); } }
internal async Task OnCmdletEndProcessing(string id, CancellationToken cancellationToken, GetEventData getEventData, SignalDelegate signal, string processRecordId) { AzurePSQoSEvent qos; if (_telemetry.TryGetValue(processRecordId, out qos)) { qos.IsSuccess = (qos.Exception == null); await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Sending new QosEvent for command '{qos.CommandName}': {qos.ToString()}")); _telemetry.LogEvent(processRecordId); _previousEndTime = DateTimeOffset.Now; } }
/// <summary> /// The cmdlet will call this for every event during the pipeline. /// </summary> /// <param name="id">a <c>string</c> containing the name of the event being raised (well-known events are in <see cref="Microsoft.Azure.Commands.Common.Events"/></param> /// <param name="cancellationToken">a <c>CancellationToken</c> indicating if this request is being cancelled.</param> /// <param name="getEventData">a delegate to call to get the event data for this event</param> /// <param name="signal">a delegate to signal an event from the handler to the cmdlet.</param> /// <param name="invocationInfo">The <see cref="System.Management.Automation.InvocationInfo" /> from the cmdlet</param> /// <param name="parameterSetName">The <see cref="string" /> containing the name of the parameter set for this invocation (if available></param> /// <param name="correlationId">The <see cref="string" /> containing the correlation id for the cmdlet (if available)</param> /// <param name="processRecordId">The <see cref="string" /> containing the correlation id for the individual process record. (if available)</param> /// <param name="exception">The <see cref="System.Exception" /> that is being thrown (if available)</param> public async Task EventListener(string id, CancellationToken cancellationToken, GetEventData getEventData, SignalDelegate signal, InvocationInfo invocationInfo, string parameterSetName, string correlationId, string processRecordId, System.Exception exception) { /// Drain the queue of ADAL events whenever an event is fired DrainMessages(signal, cancellationToken); switch (id) { case Events.BeforeCall: { var data = EventDataConverter.ConvertFrom(getEventData()); // also, we manually use our TypeConverter to return an appropriate type var request = data?.RequestMessage as HttpRequestMessage; if (request != null) { AzurePSQoSEvent qos; if (_telemetryEvents.TryGetValue(processRecordId, out qos)) { IEnumerable <string> headers; if (request.Headers != null && request.Headers.TryGetValues("x-ms-client-request-id", out headers)) { qos.ClientRequestId = headers.FirstOrDefault(); await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Amending QosEvent for command '{qos.CommandName}': {qos.ToString()}")); } } /// Print formatted request message await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent(GeneralUtilities.GetLog(request))); } } break; case Events.CmdletProcessRecordAsyncStart: { var qos = CreateQosEvent(invocationInfo, parameterSetName, correlationId); await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Created new QosEvent for command '{qos.CommandName}': {qos.ToString()}")); _telemetryEvents.Add(processRecordId, qos); } break; case Events.CmdletProcessRecordAsyncEnd: { AzurePSQoSEvent qos; if (_telemetryEvents.TryGetValue(processRecordId, out qos)) { qos.IsSuccess = qos.Exception == null; await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Sending new QosEvent for command '{qos.CommandName}': {qos.ToString()}")); _metricHelper.LogEvent(qos); _telemetryEvents.Remove(processRecordId); } } break; case Events.CmdletException: { var data = EventDataConverter.ConvertFrom(getEventData()); await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Received Exception with message '{data?.Message}'")); AzurePSQoSEvent qos; if (_telemetryEvents.TryGetValue(processRecordId, out qos)) { await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent($"[{id}]: Sending new QosEvent for command '{qos.CommandName}': {qos.ToString()}")); qos.IsSuccess = false; qos.Exception = exception; _metricHelper.LogEvent(qos); _telemetryEvents.Remove(processRecordId); } } break; case Events.ResponseCreated: { var data = EventDataConverter.ConvertFrom(getEventData()); var response = data?.ResponseMessage as HttpResponseMessage; if (response != null) { AzurePSQoSEvent qos; if (_telemetryEvents.TryGetValue(processRecordId, out qos)) { qos.ClientRequestId = response?.Headers?.GetValues("x-ms-request-id").FirstOrDefault(); } /// Print formatted response message await signal(Events.Debug, cancellationToken, () => EventHelper.CreateLogEvent(GeneralUtilities.GetLog(response))); } } break; default: getEventData.Print(signal, cancellationToken, Events.Information, id); break; } }