/// <summary>
/// Track resource upgrade operation.
/// </summary>
/// <param name="details">StartVaultUpgrade response.</param>
private void WaitForJobCompletion(ResourceUpgradeDetails details)
{
TrackResourceUpgradeResponse response = null;
DateTime startTime = DateTime.Now;
double taskTimeoutInSeconds = 6000;
double elapsedSeconds = 0;
ProgressRecord record =
new ProgressRecord(
0,
Properties.Resources.VaultUpgradeInProgress,
Properties.Resources.WaitingForCompletion);
do
{
TestMockSupport.Delay(PSRecoveryServicesClient.TimeToSleepBeforeFetchingJobDetailsAgain);
response = this.RecoveryServicesClient.TrackVaultUpgrade(
this.VaultName,
this.Location,
this.ResourceType);
elapsedSeconds = DateTime.Now.Subtract(startTime).TotalSeconds;
record.PercentComplete = (int)elapsedSeconds * 100 / (int)taskTimeoutInSeconds;
this.WriteProgress(record);
}while (response.OperationStatus == Constants.InProgress &&
elapsedSeconds < taskTimeoutInSeconds &&
!this.StopProcessingFlag);
record.RecordType = ProgressRecordType.Completed;
this.WriteProgress(record);
string operationResult = string.Empty;
string operationStatus = string.Empty;
string message = string.Empty;
if (response.OperationStatus == Constants.Completed)
{
operationResult = response.OperationResult;
operationStatus = response.OperationStatus;
if (response.OperationResult == Constants.Succeeded)
{
message =
string.Format(
Properties.Resources.VaultUpgradeSucceded,
this.ResourceType,
this.VaultName);
}
else
{
message =
Properties.Resources.VaultUpgradeNotSucceded;
}
}
else if (this.StopProcessingFlag)
{
operationResult = VaultUpgradeOperationResult.Unavailable.ToString();
operationStatus = VaultUpgradeOperationResult.InProgress.ToString();
message = Properties.Resources.VaultUpgradeTerminated;
}
else
{
operationResult = VaultUpgradeOperationResult.Failed.ToString();
operationStatus = VaultUpgradeOperationResult.TimedOut.ToString();
message = Properties.Resources.VaultUpgradeTimedOut;
}
this.WriteObject(
new ASRVaultUpgradeResponse(details, operationResult, operationStatus, message));
}
/// <summary>
/// Track Resource Upgrade.
/// </summary>
/// <param name='customRequestHeaders'>
/// Optional. Request header parameters.
/// </param>
/// <param name='cancellationToken'>
/// Cancellation token.
/// </param>
/// <returns>
/// The definition of an output object stating the status of resource
/// upgrade.
/// </returns>
public async Task <TrackResourceUpgradeResponse> TrackResourceUpgradeAsync(CustomRequestHeaders customRequestHeaders, CancellationToken cancellationToken)
{
// Validate
// Tracing
bool shouldTrace = TracingAdapter.IsEnabled;
string invocationId = null;
if (shouldTrace)
{
invocationId = TracingAdapter.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("customRequestHeaders", customRequestHeaders);
TracingAdapter.Enter(invocationId, this, "TrackResourceUpgradeAsync", tracingParameters);
}
// Construct URL
string url = "";
url = url + "/";
if (this.Client.Credentials.SubscriptionId != null)
{
url = url + Uri.EscapeDataString(this.Client.Credentials.SubscriptionId);
}
url = url + "/cloudservices/";
url = url + Uri.EscapeDataString(this.Client.CloudServiceName);
url = url + "/resources/";
url = url + Uri.EscapeDataString(this.Client.ResourceNamespace);
url = url + "/~/";
url = url + Uri.EscapeDataString(this.Client.ResourceType);
url = url + "/";
url = url + Uri.EscapeDataString(this.Client.ResourceName);
url = url + "/upgradeOperationResult";
string baseUrl = this.Client.BaseUri.AbsoluteUri;
// Trim '/' character from the end of baseUrl and beginning of url.
if (baseUrl[baseUrl.Length - 1] == '/')
{
baseUrl = baseUrl.Substring(0, baseUrl.Length - 1);
}
if (url[0] == '/')
{
url = url.Substring(1);
}
url = baseUrl + "/" + url;
url = url.Replace(" ", "%20");
// Create HTTP transport objects
HttpRequestMessage httpRequest = null;
try
{
httpRequest = new HttpRequestMessage();
httpRequest.Method = HttpMethod.Get;
httpRequest.RequestUri = new Uri(url);
// Set Headers
httpRequest.Headers.Add("Accept", "application/xml");
httpRequest.Headers.Add("Agent-Authentication", customRequestHeaders.AgentAuthenticationHeader);
httpRequest.Headers.Add("x-ms-client-request-id", customRequestHeaders.ClientRequestId);
httpRequest.Headers.Add("x-ms-version", "2013-03-01");
// Set Credentials
cancellationToken.ThrowIfCancellationRequested();
await this.Client.Credentials.ProcessHttpRequestAsync(httpRequest, cancellationToken).ConfigureAwait(false);
// Send Request
HttpResponseMessage httpResponse = null;
try
{
if (shouldTrace)
{
TracingAdapter.SendRequest(invocationId, httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
httpResponse = await this.Client.HttpClient.SendAsync(httpRequest, cancellationToken).ConfigureAwait(false);
if (shouldTrace)
{
TracingAdapter.ReceiveResponse(invocationId, httpResponse);
}
HttpStatusCode statusCode = httpResponse.StatusCode;
if (statusCode != HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
CloudException ex = CloudException.Create(httpRequest, null, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false));
if (shouldTrace)
{
TracingAdapter.Error(invocationId, ex);
}
throw ex;
}
// Create Result
TrackResourceUpgradeResponse result = null;
// Deserialize Response
if (statusCode == HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
result = new TrackResourceUpgradeResponse();
XDocument responseDoc = XDocument.Parse(responseContent);
XElement operationInfoElement = responseDoc.Element(XName.Get("OperationInfo", "http://schemas.microsoft.com/windowsazure"));
if (operationInfoElement != null)
{
XElement operationStatusElement = operationInfoElement.Element(XName.Get("OperationStatus", "http://schemas.microsoft.com/windowsazure"));
if (operationStatusElement != null)
{
string operationStatusInstance = operationStatusElement.Value;
result.OperationStatus = operationStatusInstance;
}
XElement operationResultElement = operationInfoElement.Element(XName.Get("OperationResult", "http://schemas.microsoft.com/windowsazure"));
if (operationResultElement != null)
{
string operationResultInstance = operationResultElement.Value;
result.OperationResult = operationResultInstance;
}
XElement messageElement = operationInfoElement.Element(XName.Get("Message", "http://schemas.microsoft.com/windowsazure"));
if (messageElement != null)
{
string messageInstance = messageElement.Value;
result.Message = messageInstance;
}
XElement jobStepsSequenceElement = operationInfoElement.Element(XName.Get("JobSteps", "http://schemas.microsoft.com/windowsazure"));
if (jobStepsSequenceElement != null)
{
foreach (XElement jobStepsElement in jobStepsSequenceElement.Elements(XName.Get("JobStep", "http://schemas.microsoft.com/windowsazure")))
{
JobStep jobStepInstance = new JobStep();
result.JobSteps.Add(jobStepInstance);
XElement operationStatusElement2 = jobStepsElement.Element(XName.Get("OperationStatus", "http://schemas.microsoft.com/windowsazure"));
if (operationStatusElement2 != null)
{
string operationStatusInstance2 = operationStatusElement2.Value;
jobStepInstance.OperationStatus = operationStatusInstance2;
}
XElement operationResultElement2 = jobStepsElement.Element(XName.Get("OperationResult", "http://schemas.microsoft.com/windowsazure"));
if (operationResultElement2 != null)
{
string operationResultInstance2 = operationResultElement2.Value;
jobStepInstance.OperationResult = operationResultInstance2;
}
XElement messageElement2 = jobStepsElement.Element(XName.Get("Message", "http://schemas.microsoft.com/windowsazure"));
if (messageElement2 != null)
{
string messageInstance2 = messageElement2.Value;
jobStepInstance.Message = messageInstance2;
}
XElement detailElement = jobStepsElement.Element(XName.Get("Detail", "http://schemas.microsoft.com/windowsazure"));
if (detailElement != null)
{
string detailInstance = detailElement.Value;
jobStepInstance.Detail = detailInstance;
}
}
}
XElement jobsSequenceElement = operationInfoElement.Element(XName.Get("Jobs", "http://schemas.microsoft.com/windowsazure"));
if (jobsSequenceElement != null)
{
foreach (XElement jobsElement in jobsSequenceElement.Elements(XName.Get("Job", "http://schemas.microsoft.com/2003/10/Serialization/Arrays")))
{
result.Jobs.Add(jobsElement.Value);
}
}
XElement errorCodeElement = operationInfoElement.Element(XName.Get("ErrorCode", "http://schemas.microsoft.com/windowsazure"));
if (errorCodeElement != null)
{
string errorCodeInstance = errorCodeElement.Value;
result.ErrorCode = errorCodeInstance;
}
}
}
result.StatusCode = statusCode;
if (httpResponse.Headers.Contains("x-ms-request-id"))
{
result.RequestId = httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
if (shouldTrace)
{
TracingAdapter.Exit(invocationId, result);
}
return(result);
}
finally
{
if (httpResponse != null)
{
httpResponse.Dispose();
}
}
}
finally
{
if (httpRequest != null)
{
httpRequest.Dispose();
}
}
}
