public override void ExecuteCmdlet()
{
if (ShouldProcess(Instance, Resources.RestartingAnalysisServicesServer))
{
var context = AsAzureClientSession.Instance.Profile.Context;
AsAzureClientSession.Instance.Login(context, null);
string accessToken = this.TokenCacheItemProvider.GetTokenFromTokenCache(AsAzureClientSession.TokenCache, context.Account.UniqueId);
Uri restartBaseUri = new Uri(string.Format("{0}{1}{2}", Uri.UriSchemeHttps, Uri.SchemeDelimiter, context.Environment.Name));
var restartEndpoint = string.Format((string)context.Environment.Endpoints[AsAzureEnvironment.AsRolloutEndpoints.RestartEndpointFormat], serverName);
using (HttpResponseMessage message = AsAzureHttpClient.CallPostAsync(
restartBaseUri,
restartEndpoint,
accessToken).Result)
{
message.EnsureSuccessStatusCode();
if (PassThru.IsPresent)
{
WriteObject(true);
}
}
}
}
private string ClusterResolve(Uri clusterUri, string accessToken, string serverName)
{
const string resolveEndpoint = "/webapi/clusterResolve";
var content = new StringContent($"ServerName={serverName}");
content.Headers.ContentType = MediaTypeHeaderValue.Parse("application/x-www-form-urlencoded");
using (var message = AsAzureHttpClient.CallPostAsync(
clusterUri,
resolveEndpoint,
accessToken,
content).Result)
{
message.EnsureSuccessStatusCode();
var rawResult = message.Content.ReadAsStringAsync().Result;
var jsonResult = JObject.Parse(rawResult);
return(jsonResult["clusterFQDN"].ToString());
}
}
/// <summary>
/// Resolves the cluster to which the request needs to be sent for the current environment
/// </summary>
/// <param name="context"></param>
/// <param name="serverName"></param>
/// <returns></returns>
private ClusterResolutionResult ClusterResolve(AsAzureContext context, string serverName)
{
Uri clusterResolveBaseUri = new Uri(string.Format("{0}{1}{2}", Uri.UriSchemeHttps, Uri.SchemeDelimiter, context.Environment.Name));
UriBuilder resolvedUriBuilder = new UriBuilder(clusterResolveBaseUri);
string rolloutAccessToken = this.TokenCacheItemProvider.GetTokenFromTokenCache(AsAzureClientSession.TokenCache, context.Account.UniqueId, context.Environment.Name);
var resolveEndpoint = "/webapi/clusterResolve";
var content = new StringContent($"ServerName={serverName}");
content.Headers.ContentType = MediaTypeHeaderValue.Parse("application/x-www-form-urlencoded");
this.AsAzureHttpClient.resetHttpClient();
using (HttpResponseMessage message = AsAzureHttpClient.CallPostAsync(
clusterResolveBaseUri,
resolveEndpoint,
rolloutAccessToken,
content).Result)
{
message.EnsureSuccessStatusCode();
var rawResult = message.Content.ReadAsStringAsync().Result;
ClusterResolutionResult result = JsonConvert.DeserializeObject <ClusterResolutionResult>(rawResult);
return(result);
}
}
/// <summary>
/// Worker Method for the synchronize request.
/// </summary>
/// <param name="context">The AS azure context</param>
/// <param name="syncBaseUri">Base Uri for sync</param>
/// <param name="databaseName">Database name</param>
/// <param name="accessToken">Access token</param>
/// <param name="maxNumberOfAttempts">Max number of retries for get command</param>
/// <returns></returns>
private async Task <ScaleOutServerDatabaseSyncDetails> SynchronizeDatabaseAsync(
AsAzureContext context,
Uri syncBaseUri,
string databaseName,
string accessToken)
{
Tuple <Uri, RetryConditionHeaderValue> pollingUrlAndRetryAfter = new Tuple <Uri, RetryConditionHeaderValue>(null, null);
ScaleOutServerDatabaseSyncDetails syncResult = null;
return(await Task.Run(async() =>
{
try
{
var synchronize = string.Format((string)context.Environment.Endpoints[AsAzureEnvironment.AsRolloutEndpoints.SyncEndpoint], this.serverName, databaseName);
this.AsAzureHttpClient.resetHttpClient();
using (var message = await AsAzureHttpClient.CallPostAsync(
syncBaseUri,
synchronize,
accessToken,
correlationId,
null))
{
this.syncRequestRootActivityId = message.Headers.Contains(RootActivityIdHeaderName) ? message.Headers.GetValues(RootActivityIdHeaderName).FirstOrDefault() : string.Empty;
this.syncRequestTimeStamp = message.Headers.Contains(CurrentUtcDateHeaderName) ? message.Headers.GetValues(CurrentUtcDateHeaderName).FirstOrDefault() : string.Empty;
message.EnsureSuccessStatusCode();
if (message.StatusCode != HttpStatusCode.Accepted)
{
var timestampNow = DateTime.Now;
syncResult = new ScaleOutServerDatabaseSyncDetails
{
CorrelationId = correlationId.ToString(),
Database = databaseName,
SyncState = DatabaseSyncState.Completed,
Details = string.Format("Http status code: {0}. Nothing readonly instances found to replicate databases.", message.StatusCode),
UpdatedAt = timestampNow,
StartedAt = timestampNow
};
return syncResult;
}
pollingUrlAndRetryAfter = new Tuple <Uri, RetryConditionHeaderValue>(message.Headers.Location, message.Headers.RetryAfter);
}
}
catch (Exception e)
{
var timestampNow = DateTime.Now;
// Return sync details with exception message as details
return new ScaleOutServerDatabaseSyncDetails
{
CorrelationId = correlationId.ToString(),
Database = databaseName,
SyncState = DatabaseSyncState.Invalid,
Details = Resources.PostSyncRequestFailureMessage.FormatInvariant(
this.clusterResolveResult.CoreServerName,
this.syncRequestRootActivityId,
this.syncRequestTimeStamp,
string.Format(e.Message)),
UpdatedAt = timestampNow,
StartedAt = timestampNow
};
}
Uri pollingUrl = pollingUrlAndRetryAfter.Item1;
var retryAfter = pollingUrlAndRetryAfter.Item2;
try
{
ScaleOutServerDatabaseSyncResult result = await this.PollSyncStatusWithRetryAsync(
databaseName,
accessToken,
pollingUrl,
retryAfter.Delta ?? DefaultPollingInterval);
syncResult = ScaleOutServerDatabaseSyncDetails.FromResult(result, correlationId.ToString());
}
catch (Exception e)
{
var timestampNow = DateTime.Now;
// Append exception message to sync details and return
syncResult = new ScaleOutServerDatabaseSyncDetails
{
CorrelationId = correlationId.ToString(),
Database = databaseName,
SyncState = DatabaseSyncState.Invalid,
Details = Resources.SyncASPollStatusFailureMessage.FormatInvariant(
serverName,
string.Empty,
timestampNow.ToString(CultureInfo.InvariantCulture),
string.Format(e.StackTrace)),
UpdatedAt = timestampNow,
StartedAt = timestampNow
};
}
return syncResult;
}));
}