public override void ExecuteCmdlet()
{
base.ExecuteCmdlet();
ExecuteClientAction(() =>
{
if (ShouldProcess(this.Location, VerbsData.Export))
{
var parameters = new ThrottledRequestsInput();
parameters.GroupByOperationName = this.GroupByOperationName;
parameters.BlobContainerSasUri = this.BlobContainerSasUri;
parameters.FromTime = this.FromTime;
parameters.ToTime = this.ToTime;
parameters.GroupByResourceName = this.GroupByResourceName;
parameters.GroupByThrottlePolicy = this.GroupByThrottlePolicy;
string location = this.Location.Canonicalize();
if (NoWait.IsPresent)
{
var result = LogAnalyticsClient.BeginExportThrottledRequests(parameters, location);
var psObject = new PSLogAnalyticsOperationResult();
ComputeAutomationAutoMapperProfile.Mapper.Map <LogAnalyticsOperationResult, PSLogAnalyticsOperationResult>(result, psObject);
WriteObject(psObject);
}
else
{
var result = LogAnalyticsClient.ExportThrottledRequests(parameters, location);
var psObject = new PSLogAnalyticsOperationResult();
ComputeAutomationAutoMapperProfile.Mapper.Map <LogAnalyticsOperationResult, PSLogAnalyticsOperationResult>(result, psObject);
WriteObject(psObject);
}
}
});
}
protected void ExecuteLogAnalyticExportThrottledRequestsMethod(object[] invokeMethodInputParameters)
{
var parameters = new ThrottledRequestsInput();
var pGroupByOperationName = (bool?)ParseParameter(invokeMethodInputParameters[0]);
parameters.GroupByOperationName = pGroupByOperationName;
var pFromTime = (DateTime)ParseParameter(invokeMethodInputParameters[1]);
parameters.FromTime = pFromTime;
var pGroupByThrottlePolicy = (bool?)ParseParameter(invokeMethodInputParameters[2]);
parameters.GroupByThrottlePolicy = pGroupByThrottlePolicy;
var pBlobContainerSasUri = (string)ParseParameter(invokeMethodInputParameters[3]);
parameters.BlobContainerSasUri = pBlobContainerSasUri;
var pGroupByResourceName = (bool?)ParseParameter(invokeMethodInputParameters[4]);
parameters.GroupByResourceName = pGroupByResourceName;
var pToTime = (DateTime)ParseParameter(invokeMethodInputParameters[5]);
parameters.ToTime = pToTime;
string location = (string)ParseParameter(invokeMethodInputParameters[7]);
var result = LogAnalyticsClient.ExportThrottledRequests(parameters, location);
WriteObject(result);
}
public async Task TestExportingThrottlingLogs()
{
string rg1Name = Recording.GenerateAssetName(TestPrefix);
string storageAccountName = Recording.GenerateAssetName(TestPrefix);
EnsureClientsInitialized(DefaultLocation);
string sasUri = await GetBlobContainerSasUri(rg1Name, storageAccountName);
RequestRateByIntervalInput requestRateByIntervalInput = new RequestRateByIntervalInput(sasUri, Recording.UtcNow.AddDays(-10), Recording.UtcNow.AddDays(-8), IntervalInMins.FiveMins);
var result = await WaitForCompletionAsync(await LogAnalyticsOperations.StartExportRequestRateByIntervalAsync("westcentralus", requestRateByIntervalInput));
//BUG: LogAnalytics API does not return correct result.
//Assert.EndsWith(".csv", result.Properties.Output);
ThrottledRequestsInput throttledRequestsInput = new ThrottledRequestsInput(sasUri, Recording.UtcNow.AddDays(-10), Recording.UtcNow.AddDays(-8))
{
GroupByOperationName = true,
};
var result1 = await WaitForCompletionAsync(await LogAnalyticsOperations.StartExportThrottledRequestsAsync("westcentralus", throttledRequestsInput));
//BUG: LogAnalytics API does not return correct result.
//Assert.EndsWith(".csv", result.Properties.Output);
}
internal HttpMessage CreateExportThrottledRequestsRequest(string location, ThrottledRequestsInput parameters)
{
var message = _pipeline.CreateMessage();
var request = message.Request;
request.Method = RequestMethod.Post;
var uri = new RawRequestUriBuilder();
uri.Reset(endpoint);
uri.AppendPath("/subscriptions/", false);
uri.AppendPath(subscriptionId, true);
uri.AppendPath("/providers/Microsoft.Compute/locations/", false);
uri.AppendPath(location, true);
uri.AppendPath("/logAnalytics/apiAccess/getThrottledRequests", false);
uri.AppendQuery("api-version", "2021-03-01", true);
request.Uri = uri;
request.Headers.Add("Accept", "application/json");
request.Headers.Add("Content-Type", "application/json");
var content = new Utf8JsonRequestContent();
content.JsonWriter.WriteObjectValue(parameters);
request.Content = content;
message.SetProperty("UserAgentOverride", _userAgent);
return(message);
}
public void TestExportingThrottlingLogs()
{
using (MockContext context = MockContext.Start(this.GetType()))
{
string rg1Name = ComputeManagementTestUtilities.GenerateName(TestPrefix);
string storageAccountName = ComputeManagementTestUtilities.GenerateName(TestPrefix);
try
{
EnsureClientsInitialized(context);
string sasUri = GetBlobContainerSasUri(rg1Name, storageAccountName);
RequestRateByIntervalInput requestRateByIntervalInput = new RequestRateByIntervalInput()
{
BlobContainerSasUri = sasUri,
FromTime = DateTime.UtcNow.AddDays(-10),
ToTime = DateTime.UtcNow.AddDays(-8),
IntervalLength = IntervalInMins.FiveMins,
};
var result = m_CrpClient.LogAnalytics.ExportRequestRateByInterval(requestRateByIntervalInput, "westcentralus");
//BUG: LogAnalytics API does not return correct result.
//Assert.EndsWith(".csv", result.Properties.Output);
ThrottledRequestsInput throttledRequestsInput = new ThrottledRequestsInput()
{
BlobContainerSasUri = sasUri,
FromTime = DateTime.UtcNow.AddDays(-10),
ToTime = DateTime.UtcNow.AddDays(-8),
GroupByOperationName = true,
};
result = m_CrpClient.LogAnalytics.ExportThrottledRequests(throttledRequestsInput, "westcentralus");
//BUG: LogAnalytics API does not return correct result.
//Assert.EndsWith(".csv", result.Properties.Output);
ThrottledRequestsInput throttledRequestsInput2 = new ThrottledRequestsInput()
{
BlobContainerSasUri = sasUri,
FromTime = DateTime.UtcNow.AddDays(-10),
ToTime = DateTime.UtcNow.AddDays(-8),
GroupByOperationName = false,
GroupByClientApplicationId = true,
GroupByUserAgent = false,
};
result = m_CrpClient.LogAnalytics.ExportThrottledRequests(throttledRequestsInput2, "eastus2");
}
finally
{
m_ResourcesClient.ResourceGroups.Delete(rg1Name);
}
}
}
protected PSArgument[] CreateLogAnalyticExportThrottledRequestsParameters()
{
ThrottledRequestsInput parameters = new ThrottledRequestsInput();
string location = string.Empty;
return(ConvertFromObjectsToArguments(
new string[] { "Parameters", "Location" },
new object[] { parameters, location }));
}
public Response ExportThrottledRequests(string location, ThrottledRequestsInput parameters, CancellationToken cancellationToken = default)
{
if (location == null)
{
throw new ArgumentNullException(nameof(location));
}
if (parameters == null)
{
throw new ArgumentNullException(nameof(parameters));
}
using var message = CreateExportThrottledRequestsRequest(location, parameters);
_pipeline.Send(message, cancellationToken);
switch (message.Response.Status)
{
case 200:
case 202:
return(message.Response);
default:
throw _clientDiagnostics.CreateRequestFailedException(message.Response);
}
}
public override void ExecuteCmdlet()
{
base.ExecuteCmdlet();
ExecuteClientAction(() =>
{
if (ShouldProcess(this.Location, VerbsData.Export))
{
var parameters = new ThrottledRequestsInput();
parameters.GroupByOperationName = this.GroupByOperationName;
parameters.BlobContainerSasUri = this.BlobContainerSasUri;
parameters.FromTime = this.FromTime;
parameters.ToTime = this.ToTime;
parameters.GroupByResourceName = this.GroupByResourceName;
parameters.GroupByThrottlePolicy = this.GroupByThrottlePolicy;
parameters.GroupByClientApplicationId = this.GroupByApplicationId;
parameters.GroupByUserAgent = this.GroupByUserAgent;
string location = this.Location.Canonicalize();
if (NoWait.IsPresent)
{
var result = LogAnalyticsClient.BeginExportThrottledRequests(parameters, location);
var psObject = new PSLogAnalyticsOperationResult();
ComputeAutomationAutoMapperProfile.Mapper.Map <LogAnalyticsOperationResult, PSLogAnalyticsOperationResult>(result, psObject);
WriteObject(psObject);
}
else
{
var result = LogAnalyticsClient.ExportThrottledRequests(parameters, location);
var psObject = new PSLogAnalyticsOperationResult();
ComputeAutomationAutoMapperProfile.Mapper.Map <LogAnalyticsOperationResult, PSLogAnalyticsOperationResult>(result, psObject);
WriteObject(psObject);
}
WriteWarning("Please go to https://aka.ms/requestRateByInterval to learn more about this cmdlet.");
}
});
}
/// <summary>
/// Export logs that show total throttled Api requests for this subscription in
/// the given time window.
/// </summary>
/// <param name='parameters'>
/// Parameters supplied to the LogAnalytics getThrottledRequests Api.
/// </param>
/// <param name='location'>
/// The location upon which virtual-machine-sizes is queried.
/// </param>
/// <param name='customHeaders'>
/// The headers that will be added to request.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public async Task <AzureOperationResponse <LogAnalyticsOperationResult> > ExportThrottledRequestsWithHttpMessagesAsync(ThrottledRequestsInput parameters, string location, Dictionary <string, List <string> > customHeaders = null, CancellationToken cancellationToken = default(CancellationToken))
{
// Send request
AzureOperationResponse <LogAnalyticsOperationResult> _response = await BeginExportThrottledRequestsWithHttpMessagesAsync(parameters, location, customHeaders, cancellationToken).ConfigureAwait(false);
return(await Client.GetPostOrDeleteOperationResultAsync(_response, customHeaders, cancellationToken).ConfigureAwait(false));
}
/// <summary>
/// Export logs that show total throttled Api requests for this subscription in
/// the given time window.
/// </summary>
/// <param name='parameters'>
/// Parameters supplied to the LogAnalytics getThrottledRequests Api.
/// </param>
/// <param name='location'>
/// The location upon which virtual-machine-sizes is queried.
/// </param>
/// <param name='customHeaders'>
/// Headers that will be added to request.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
/// <exception cref="CloudException">
/// Thrown when the operation returned an invalid status code
/// </exception>
/// <exception cref="SerializationException">
/// Thrown when unable to deserialize the response
/// </exception>
/// <exception cref="ValidationException">
/// Thrown when a required parameter is null
/// </exception>
/// <exception cref="System.ArgumentNullException">
/// Thrown when a required parameter is null
/// </exception>
/// <return>
/// A response object containing the response body and response headers.
/// </return>
public async Task <AzureOperationResponse <LogAnalyticsOperationResult> > BeginExportThrottledRequestsWithHttpMessagesAsync(ThrottledRequestsInput parameters, string location, Dictionary <string, List <string> > customHeaders = null, CancellationToken cancellationToken = default(CancellationToken))
{
if (parameters == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "parameters");
}
if (parameters != null)
{
parameters.Validate();
}
if (location == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "location");
}
if (location != null)
{
if (!System.Text.RegularExpressions.Regex.IsMatch(location, "^[-\\w\\._]+$"))
{
throw new ValidationException(ValidationRules.Pattern, "location", "^[-\\w\\._]+$");
}
}
if (Client.SubscriptionId == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "this.Client.SubscriptionId");
}
string apiVersion = "2021-07-01";
// Tracing
bool _shouldTrace = ServiceClientTracing.IsEnabled;
string _invocationId = null;
if (_shouldTrace)
{
_invocationId = ServiceClientTracing.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("parameters", parameters);
tracingParameters.Add("location", location);
tracingParameters.Add("apiVersion", apiVersion);
tracingParameters.Add("cancellationToken", cancellationToken);
ServiceClientTracing.Enter(_invocationId, this, "BeginExportThrottledRequests", tracingParameters);
}
// Construct URL
var _baseUrl = Client.BaseUri.AbsoluteUri;
var _url = new System.Uri(new System.Uri(_baseUrl + (_baseUrl.EndsWith("/") ? "" : "/")), "subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests").ToString();
_url = _url.Replace("{location}", System.Uri.EscapeDataString(location));
_url = _url.Replace("{subscriptionId}", System.Uri.EscapeDataString(Client.SubscriptionId));
List <string> _queryParameters = new List <string>();
if (apiVersion != null)
{
_queryParameters.Add(string.Format("api-version={0}", System.Uri.EscapeDataString(apiVersion)));
}
if (_queryParameters.Count > 0)
{
_url += (_url.Contains("?") ? "&" : "?") + string.Join("&", _queryParameters);
}
// Create HTTP transport objects
var _httpRequest = new HttpRequestMessage();
HttpResponseMessage _httpResponse = null;
_httpRequest.Method = new HttpMethod("POST");
_httpRequest.RequestUri = new System.Uri(_url);
// Set Headers
if (Client.GenerateClientRequestId != null && Client.GenerateClientRequestId.Value)
{
_httpRequest.Headers.TryAddWithoutValidation("x-ms-client-request-id", System.Guid.NewGuid().ToString());
}
if (Client.AcceptLanguage != null)
{
if (_httpRequest.Headers.Contains("accept-language"))
{
_httpRequest.Headers.Remove("accept-language");
}
_httpRequest.Headers.TryAddWithoutValidation("accept-language", Client.AcceptLanguage);
}
if (customHeaders != null)
{
foreach (var _header in customHeaders)
{
if (_httpRequest.Headers.Contains(_header.Key))
{
_httpRequest.Headers.Remove(_header.Key);
}
_httpRequest.Headers.TryAddWithoutValidation(_header.Key, _header.Value);
}
}
// Serialize Request
string _requestContent = null;
if (parameters != null)
{
_requestContent = Rest.Serialization.SafeJsonConvert.SerializeObject(parameters, Client.SerializationSettings);
_httpRequest.Content = new StringContent(_requestContent, System.Text.Encoding.UTF8);
_httpRequest.Content.Headers.ContentType = System.Net.Http.Headers.MediaTypeHeaderValue.Parse("application/json; charset=utf-8");
}
// Set Credentials
if (Client.Credentials != null)
{
cancellationToken.ThrowIfCancellationRequested();
await Client.Credentials.ProcessHttpRequestAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
}
// Send Request
if (_shouldTrace)
{
ServiceClientTracing.SendRequest(_invocationId, _httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
_httpResponse = await Client.HttpClient.SendAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
if (_shouldTrace)
{
ServiceClientTracing.ReceiveResponse(_invocationId, _httpResponse);
}
HttpStatusCode _statusCode = _httpResponse.StatusCode;
cancellationToken.ThrowIfCancellationRequested();
string _responseContent = null;
if ((int)_statusCode != 200 && (int)_statusCode != 202)
{
var ex = new CloudException(string.Format("Operation returned an invalid status code '{0}'", _statusCode));
try
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
CloudError _errorBody = Rest.Serialization.SafeJsonConvert.DeserializeObject <CloudError>(_responseContent, Client.DeserializationSettings);
if (_errorBody != null)
{
ex = new CloudException(_errorBody.Message);
ex.Body = _errorBody;
}
}
catch (JsonException)
{
// Ignore the exception
}
ex.Request = new HttpRequestMessageWrapper(_httpRequest, _requestContent);
ex.Response = new HttpResponseMessageWrapper(_httpResponse, _responseContent);
if (_httpResponse.Headers.Contains("x-ms-request-id"))
{
ex.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
if (_shouldTrace)
{
ServiceClientTracing.Error(_invocationId, ex);
}
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw ex;
}
// Create Result
var _result = new AzureOperationResponse <LogAnalyticsOperationResult>();
_result.Request = _httpRequest;
_result.Response = _httpResponse;
if (_httpResponse.Headers.Contains("x-ms-request-id"))
{
_result.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
// Deserialize Response
if ((int)_statusCode == 200)
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
try
{
_result.Body = Rest.Serialization.SafeJsonConvert.DeserializeObject <LogAnalyticsOperationResult>(_responseContent, Client.DeserializationSettings);
}
catch (JsonException ex)
{
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw new SerializationException("Unable to deserialize the response.", _responseContent, ex);
}
}
if (_shouldTrace)
{
ServiceClientTracing.Exit(_invocationId, _result);
}
return(_result);
}
/// <summary>
/// Export logs that show total throttled Api requests for this subscription in
/// the given time window.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='parameters'>
/// Parameters supplied to the LogAnalytics getThrottledRequests Api.
/// </param>
/// <param name='location'>
/// The location upon which virtual-machine-sizes is queried.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <LogAnalyticsOperationResultInner> ExportThrottledRequestsAsync(this ILogAnalyticsOperations operations, ThrottledRequestsInput parameters, string location, CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.ExportThrottledRequestsWithHttpMessagesAsync(parameters, location, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
public Response ExportThrottledRequests(string subscriptionId, string location, ThrottledRequestsInput parameters, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(subscriptionId, nameof(subscriptionId));
Argument.AssertNotNullOrEmpty(location, nameof(location));
Argument.AssertNotNull(parameters, nameof(parameters));
using var message = CreateExportThrottledRequestsRequest(subscriptionId, location, parameters);
_pipeline.Send(message, cancellationToken);
switch (message.Response.Status)
{
case 200:
case 202:
return(message.Response);
default:
throw new RequestFailedException(message.Response);
}
}
/// <summary>
/// Export logs that show total throttled Api requests for this subscription in
/// the given time window.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='parameters'>
/// Parameters supplied to the LogAnalytics getThrottledRequests Api.
/// </param>
/// <param name='location'>
/// The location upon which virtual-machine-sizes is queried.
/// </param>
public static LogAnalyticsOperationResult ExportThrottledRequests(this ILogAnalyticsOperations operations, ThrottledRequestsInput parameters, string location)
{
return(operations.ExportThrottledRequestsAsync(parameters, location).GetAwaiter().GetResult());
}
public virtual LogAnalyticsExportThrottledRequestsOperation StartExportThrottledRequests(string location, ThrottledRequestsInput parameters, CancellationToken cancellationToken = default)
{
if (location == null)
{
throw new ArgumentNullException(nameof(location));
}
if (parameters == null)
{
throw new ArgumentNullException(nameof(parameters));
}
using var scope = _clientDiagnostics.CreateScope("LogAnalyticsOperations.StartExportThrottledRequests");
scope.Start();
try
{
var originalResponse = RestClient.ExportThrottledRequests(location, parameters, cancellationToken);
return(new LogAnalyticsExportThrottledRequestsOperation(_clientDiagnostics, _pipeline, RestClient.CreateExportThrottledRequestsRequest(location, parameters).Request, originalResponse));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}