public override void ExecuteCmdlet() { if (ShouldProcess(Instance, Resources.ExportingLogFromAnalysisServicesServer)) { var context = AsAzureClientSession.Instance.Profile.Context; AsAzureClientSession.Instance.Login(context, null); string accessToken = this.TokenCacheItemProvider.GetTokenFromTokenCache( AsAzureClientSession.TokenCache, context.Account.UniqueId); Uri logfileBaseUri = new Uri(string.Format("{0}{1}{2}", Uri.UriSchemeHttps, Uri.SchemeDelimiter, context.Environment.Name)); UriBuilder resolvedUriBuilder = new UriBuilder(logfileBaseUri); resolvedUriBuilder.Host = ClusterResolve(logfileBaseUri, accessToken, serverName); var logfileEndpoint = string.Format( (string)context.Environment.Endpoints[AsAzureEnvironment.AsRolloutEndpoints.LogfileEndpointFormat], serverName); this.AsAzureHttpClient.resetHttpClient(); using (HttpResponseMessage message = AsAzureHttpClient.CallGetAsync( resolvedUriBuilder.Uri, logfileEndpoint, accessToken).ConfigureAwait(false).GetAwaiter().GetResult()) { message.EnsureSuccessStatusCode(); string actionWarning = string.Format(CultureInfo.CurrentCulture, Resources.ExportingLogOverwriteWarning, this.OutputPath); if (AzureSession.Instance.DataStore.FileExists(this.OutputPath) && !this.Force.IsPresent && !ShouldContinue(actionWarning, Resources.Confirm)) { return; } AzureSession.Instance.DataStore.WriteFile(this.OutputPath, message.Content.ReadAsStringAsync().Result); } } }
/// <summary> /// /// </summary> /// <param name="databaseName">Database name</param> /// <param name="accessToken">Access token</param> /// <param name="pollingUrl">URL for polling</param> /// <param name="pollingInterval">Polling interval set by the post response</param> /// <param name="maxNumberOfAttempts">Max number of attempts for each poll before the attempt is declared a failure</param> /// <returns></returns> private async Task <ScaleOutServerDatabaseSyncResult> PollSyncStatusWithRetryAsync(string databaseName, string accessToken, Uri pollingUrl, TimeSpan pollingInterval, int maxNumberOfAttempts = 3) { return(await Task.Run(async() => { ScaleOutServerDatabaseSyncResult response = null; var syncCompleted = false; var retryCount = 0; while (!syncCompleted && retryCount < maxNumberOfAttempts) { // Wait for specified polling interval other than retries. if (retryCount == 0) { await Task.Delay(pollingInterval); } else { await Task.Delay(DefaultRetryIntervalForPolling); } this.AsAzureHttpClient.resetHttpClient(); using (HttpResponseMessage message = await AsAzureHttpClient.CallGetAsync( pollingUrl, string.Empty, accessToken, correlationId)) { bool shouldRetry = false; if (message.IsSuccessStatusCode && message.Content != null) { var responseString = await message.Content.ReadAsStringAsync(); response = JsonConvert.DeserializeObject <ScaleOutServerDatabaseSyncResult>(responseString); if (response != null) { var state = response.SyncState; if (state == DatabaseSyncState.Completed || state == DatabaseSyncState.Failed) { syncCompleted = true; } else { pollingUrl = message.Headers.Location ?? pollingUrl; pollingInterval = message.Headers.RetryAfter.Delta ?? pollingInterval; } } else { shouldRetry = true; } } else { shouldRetry = true; } if (shouldRetry) { retryCount++; response = new ScaleOutServerDatabaseSyncResult() { Database = databaseName, SyncState = DatabaseSyncState.Invalid }; response.Details = string.Format( "Http Error code: {0}. Message: {1}", message.StatusCode.ToString(), message.Content != null ? await message.Content.ReadAsStringAsync() : string.Empty); if (message.StatusCode >= (HttpStatusCode)400 && message.StatusCode <= (HttpStatusCode)499) { break; } } else { retryCount = 0; } } } return response; })); }
/// <summary> /// /// </summary> /// <param name="databaseName">Database name</param> /// <param name="accessToken">Access token</param> /// <param name="pollingUrl">URL for polling</param> /// <param name="pollingInterval">Polling interval set by the post response</param> /// <param name="maxNumberOfAttempts">Max number of attempts for each poll before the attempt is declared a failure</param> /// <returns></returns> private async Task <ScaleOutServerDatabaseSyncResult> PollSyncStatusWithRetryAsync(string databaseName, string accessToken, Uri pollingUrl, TimeSpan pollingInterval, int maxNumberOfAttempts = 3) { return(await Task.Run(async() => { ScaleOutServerDatabaseSyncResult response = null; var syncCompleted = false; do { var retryCount = 0; while (retryCount < maxNumberOfAttempts) { // Wait for specified polling interval other than retries. if (retryCount == 0) { // WriteInformation(new InformationRecord(string.Format("Synchronize database {0}. Attempt #{1}. Waiting for {2} seconds to get sync results...", databaseName, retryCount, pollingInterval.TotalSeconds), string.Empty)); await Task.Delay(pollingInterval); } else { await Task.Delay(DefaultRetryIntervalForPolling); } this.AsAzureHttpClient.resetHttpClient(); using (HttpResponseMessage message = await AsAzureHttpClient.CallGetAsync( pollingUrl, string.Empty, accessToken, correlationId)) { syncCompleted = !message.StatusCode.Equals(HttpStatusCode.SeeOther); if (syncCompleted) { if (message.IsSuccessStatusCode) { var responseString = await message.Content.ReadAsStringAsync(); response = JsonConvert.DeserializeObject <ScaleOutServerDatabaseSyncResult>(responseString); break; } else { retryCount++; if (response == null) { response = new ScaleOutServerDatabaseSyncResult() { Database = databaseName, SyncState = DatabaseSyncState.Invalid }; response.Details = string.Format( "Http Error code: {0}. {1}", message.StatusCode.ToString(), message.Content != null ? await message.Content.ReadAsStringAsync() : string.Empty); } if (message.StatusCode >= (HttpStatusCode)400 && message.StatusCode <= (HttpStatusCode)499) { break; } } } else { pollingUrl = message.Headers.Location; pollingInterval = message.Headers.RetryAfter.Delta ?? pollingInterval; } } } }while (!syncCompleted); return response; })); }