/// <summary>
/// Initializes a new instance of the <see cref="ASRVaultUpgradeResponse" /> class with required
/// parameters.
/// </summary>
/// <param name="resourceUpgradeDetails">Resource upgrade details.</param>
/// <param name="operationResult">Vault upgrade operation result.</param>
/// <param name="operationStatus">Vault upgrade operation status.</param>
/// <param name="message">Upgrade status.</param>
public ASRVaultUpgradeResponse(ResourceUpgradeDetails resourceUpgradeDetails, string operationResult, string operationStatus, string message)
{
this.OperationId = resourceUpgradeDetails.OperationId;
this.StartTimeUtc = resourceUpgradeDetails.StartTimeUtc;
this.OperationResult = operationResult;
this.OperationStatus = operationStatus;
this.UpgradeStatus = message;
}
/// <summary>
/// ProcessRecord of the command.
/// </summary>
public override void ExecuteCmdlet()
{
CheckVaultUpgradePrerequisitesResponse result =
this.ValidateVaultUpgradePrerequisites();
if (result == CheckVaultUpgradePrerequisitesResponse.Failed)
{
return;
}
bool actionConfirmed = false;
ResourceUpgradeDetails response = null;
string message = result == CheckVaultUpgradePrerequisitesResponse.Succeeded ?
Properties.Resources.PrerequisitesCheckPassed :
Properties.Resources.PrerequisitesCheckReturnedWarnings;
try
{
if (this.ShouldProcess(this.VaultName, Properties.Resources.StartVaultUpgradeWhatIfMessage) &&
(this.Force.IsPresent || this.ShouldContinue(string.Format(message), string.Empty)))
{
response = this.RecoveryServicesClient.StartVaultUpgrade(
this.VaultName,
this.Location,
this.ResourceType,
this.TargetResourceGroupName,
this.Profile.Context.Subscription.Id.ToString());
actionConfirmed = true;
}
if (actionConfirmed)
{
this.WaitForJobCompletion(response);
}
}
catch (Exception exception)
{
this.HandleVaultUpgradeException(exception);
}
}
/// <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>
/// Start Resource Upgrade.
/// </summary>
/// <param name='resourceUpgradeInput'>
/// Required. Input required for resource upgradation.
/// </param>
/// <param name='customRequestHeaders'>
/// Optional. Request header parameters.
/// </param>
/// <param name='cancellationToken'>
/// Cancellation token.
/// </param>
/// <returns>
/// The definition of a resource upgrade output object.
/// </returns>
public async Task <ResourceUpgradeDetails> UpgradeResourceAsync(ResourceUpgradeInput resourceUpgradeInput, CustomRequestHeaders customRequestHeaders, CancellationToken cancellationToken)
{
// Validate
if (resourceUpgradeInput == null)
{
throw new ArgumentNullException("resourceUpgradeInput");
}
if (resourceUpgradeInput.NewResourcePath == null)
{
throw new ArgumentNullException("resourceUpgradeInput.NewResourcePath");
}
// Tracing
bool shouldTrace = TracingAdapter.IsEnabled;
string invocationId = null;
if (shouldTrace)
{
invocationId = TracingAdapter.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("resourceUpgradeInput", resourceUpgradeInput);
tracingParameters.Add("customRequestHeaders", customRequestHeaders);
TracingAdapter.Enter(invocationId, this, "UpgradeResourceAsync", 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 + "/upgrade";
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.Post;
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);
// Serialize Request
string requestContent = null;
XDocument requestDoc = new XDocument();
XElement resourceUpgradeInputElement = new XElement(XName.Get("ResourceUpgradeInput", "http://schemas.microsoft.com/windowsazure"));
requestDoc.Add(resourceUpgradeInputElement);
XElement newResourcePathElement = new XElement(XName.Get("NewResourcePath", "http://schemas.microsoft.com/windowsazure"));
newResourcePathElement.Value = resourceUpgradeInput.NewResourcePath;
resourceUpgradeInputElement.Add(newResourcePathElement);
requestContent = requestDoc.ToString();
httpRequest.Content = new StringContent(requestContent, Encoding.UTF8);
httpRequest.Content.Headers.ContentType = MediaTypeHeaderValue.Parse("application/xml");
// 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, requestContent, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false));
if (shouldTrace)
{
TracingAdapter.Error(invocationId, ex);
}
throw ex;
}
// Create Result
ResourceUpgradeDetails result = null;
// Deserialize Response
if (statusCode == HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
result = new ResourceUpgradeDetails();
XDocument responseDoc = XDocument.Parse(responseContent);
XElement resourceUpgradeDetailsElement = responseDoc.Element(XName.Get("ResourceUpgradeDetails", "http://schemas.microsoft.com/windowsazure"));
if (resourceUpgradeDetailsElement != null)
{
XElement operationIdElement = resourceUpgradeDetailsElement.Element(XName.Get("OperationId", "http://schemas.microsoft.com/windowsazure"));
if (operationIdElement != null)
{
string operationIdInstance = operationIdElement.Value;
result.OperationId = operationIdInstance;
}
XElement startTimeUtcElement = resourceUpgradeDetailsElement.Element(XName.Get("StartTimeUtc", "http://schemas.microsoft.com/windowsazure"));
if (startTimeUtcElement != null)
{
DateTime startTimeUtcInstance = DateTime.Parse(startTimeUtcElement.Value, CultureInfo.InvariantCulture, DateTimeStyles.AdjustToUniversal).ToLocalTime();
result.StartTimeUtc = startTimeUtcInstance;
}
XElement lastUpdatedTimeUtcElement = resourceUpgradeDetailsElement.Element(XName.Get("LastUpdatedTimeUtc", "http://schemas.microsoft.com/windowsazure"));
if (lastUpdatedTimeUtcElement != null)
{
DateTime lastUpdatedTimeUtcInstance = DateTime.Parse(lastUpdatedTimeUtcElement.Value, CultureInfo.InvariantCulture, DateTimeStyles.AdjustToUniversal).ToLocalTime();
result.LastUpdatedTimeUtc = lastUpdatedTimeUtcInstance;
}
XElement endTimeUtcElement = resourceUpgradeDetailsElement.Element(XName.Get("EndTimeUtc", "http://schemas.microsoft.com/windowsazure"));
if (endTimeUtcElement != null)
{
DateTime endTimeUtcInstance = DateTime.Parse(endTimeUtcElement.Value, CultureInfo.InvariantCulture, DateTimeStyles.AdjustToUniversal).ToLocalTime();
result.EndTimeUtc = endTimeUtcInstance;
}
XElement statusElement = resourceUpgradeDetailsElement.Element(XName.Get("Status", "http://schemas.microsoft.com/windowsazure"));
if (statusElement != null)
{
string statusInstance = statusElement.Value;
result.Status = statusInstance;
}
XElement messageElement = resourceUpgradeDetailsElement.Element(XName.Get("Message", "http://schemas.microsoft.com/windowsazure"));
if (messageElement != null)
{
string messageInstance = messageElement.Value;
result.Message = messageInstance;
}
XElement triggerTypeElement = resourceUpgradeDetailsElement.Element(XName.Get("TriggerType", "http://schemas.microsoft.com/windowsazure"));
if (triggerTypeElement != null)
{
string triggerTypeInstance = triggerTypeElement.Value;
result.TriggerType = triggerTypeInstance;
}
XElement upgradedResourceIdElement = resourceUpgradeDetailsElement.Element(XName.Get("UpgradedResourceId", "http://schemas.microsoft.com/windowsazure"));
if (upgradedResourceIdElement != null)
{
string upgradedResourceIdInstance = upgradedResourceIdElement.Value;
result.UpgradedResourceId = upgradedResourceIdInstance;
}
}
}
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();
}
}
}
