/// <summary>
/// Set PE protection.
/// </summary>
private void SetPEReprotect()
{
if (this.ProtectionEntity == null)
{
ProtectionEntityResponse protectionEntityResponse =
RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
this.ProtectionContainerId,
this.ProtectionEntityId);
this.ProtectionEntity = new ASRProtectionEntity(protectionEntityResponse.ProtectionEntity);
}
// Until RR is done active location remains same from where FO was initiated.
if ((this.Direction == PSRecoveryServicesClient.PrimaryToRecovery &&
this.ProtectionEntity.ActiveLocation != PSRecoveryServicesClient.RecoveryLocation) ||
(this.Direction == PSRecoveryServicesClient.RecoveryToPrimary &&
this.ProtectionEntity.ActiveLocation != PSRecoveryServicesClient.PrimaryLocation))
{
throw new ArgumentException("Parameter value is not correct.", "Direction");
}
this.jobResponse = RecoveryServicesClient.StartAzureSiteRecoveryReprotection(
this.ProtectionContainerId,
this.ProtectionEntityId);
this.WriteJob(this.jobResponse.Job);
if (this.WaitForCompletion.IsPresent)
{
this.WaitForJobCompletion(this.jobResponse.Job.ID);
}
}
/// <summary>
/// Queries by Id.
/// </summary>
private void GetById()
{
ProtectionEntityResponse protectionEntityResponse =
RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
this.ProtectionContainerId,
this.Id);
this.WriteProtectionEntity(protectionEntityResponse.ProtectionEntity);
}
/// <summary>
/// ProcessRecord of the command.
/// </summary>
public override void ExecuteCmdlet()
{
switch (this.ParameterSetName)
{
case ASRParameterSets.ByPEObject:
this.Id = this.ProtectionEntity.ID;
this.ProtectionContainerId = this.ProtectionEntity.ProtectionContainerId;
this.targetNameOrId = this.ProtectionEntity.Name;
this.alreadyEnabled = this.ProtectionEntity.Protected;
break;
case ASRParameterSets.ByIDs:
this.targetNameOrId = this.Id;
ProtectionEntityResponse protectionEntityResponse =
RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
this.ProtectionContainerId,
this.Id);
this.alreadyEnabled = protectionEntityResponse.ProtectionEntity.Protected;
this.targetNameOrId = protectionEntityResponse.ProtectionEntity.Name;
break;
}
if (this.alreadyEnabled &&
this.Protection.Equals(Constants.EnableProtection, StringComparison.OrdinalIgnoreCase))
{
throw new ArgumentException(
string.Format(
Properties.Resources.ProtectionEntityAlreadyEnabled,
this.targetNameOrId));
}
else if (!this.alreadyEnabled &&
this.Protection.Equals(Constants.DisableProtection, StringComparison.OrdinalIgnoreCase))
{
throw new ArgumentException(
Properties.Resources.ProtectionEntityAlreadyDisabled,
this.targetNameOrId);
}
this.ConfirmAction(
this.Force.IsPresent || 0 != string.CompareOrdinal(this.Protection, Constants.DisableProtection),
string.Format(Properties.Resources.DisableProtectionWarning, this.targetNameOrId),
string.Format(Properties.Resources.DisableProtectionWhatIfMessage, this.Protection),
this.targetNameOrId,
() =>
{
try
{
if (this.Protection == Constants.EnableProtection)
{
string profileId = string.Empty;
var input = new EnableProtectionInput();
if (this.ProtectionEntity == null)
{
var pe = RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
this.ProtectionContainerId,
this.Id);
this.ProtectionEntity = new ASRProtectionEntity(pe.ProtectionEntity);
}
// Get the replciation provider from profile object otherwise assume its E2E.
// Let the call go without profileId set.
string replicationProvider = null;
if (this.ProtectionProfile != null)
{
profileId = this.ProtectionProfile.ID;
replicationProvider = this.ProtectionProfile.ReplicationProvider;
}
else
{
string pcId = this.ProtectionContainerId ?? this.ProtectionEntity.ProtectionContainerId;
var pc = RecoveryServicesClient.GetAzureSiteRecoveryProtectionContainer(
pcId);
// PC will have all profiles associated with same replciation providers only.
replicationProvider =
pc.ProtectionContainer.AvailableProtectionProfiles.Count < 1 ?
null :
pc.ProtectionContainer.AvailableProtectionProfiles[0].ReplicationProvider;
if (replicationProvider != Constants.HyperVReplica)
{
throw new Exception("Please provide the protection profile object. It can be chosen from available protection profiles of the protection container.");
}
}
if (this.ParameterSetName == ASRParameterSets.ByIDs)
{
this.ValidateUsageById(replicationProvider, "Id");
}
if (replicationProvider == Constants.HyperVReplicaAzure)
{
input.ProtectionProfileId = this.ProtectionProfile.ID;
AzureEnableProtectionInput azureInput = new AzureEnableProtectionInput();
azureInput.HvHostVmId = this.ProtectionEntity.FabricObjectId;
azureInput.VmName = this.ProtectionEntity.Name;
azureInput.OSType = this.OS;
if (string.IsNullOrWhiteSpace(this.OS))
{
azureInput.OSType = this.ProtectionEntity.OS;
}
if (string.IsNullOrWhiteSpace(this.OSDiskName))
{
azureInput.VHDId = this.ProtectionEntity.OSDiskId;
}
else
{
foreach (var disk in this.ProtectionEntity.Disks)
{
if (disk.Name == this.OSDiskName)
{
azureInput.VHDId = disk.Id;
break;
}
}
}
input.ReplicationProviderInput = DataContractUtils.Serialize <AzureEnableProtectionInput>(azureInput);
}
else if (string.IsNullOrWhiteSpace(profileId))
{
input = null;
}
else
{
input.ReplicationProviderInput = string.Empty;
input.ProtectionProfileId = profileId;
}
this.jobResponse =
RecoveryServicesClient.EnableProtection(
this.ProtectionContainerId,
this.Id,
input);
}
else
{
this.jobResponse =
RecoveryServicesClient.DisbleProtection(
this.ProtectionContainerId,
this.Id);
}
this.WriteJob(this.jobResponse.Job);
if (this.WaitForCompletion.IsPresent)
{
this.WaitForJobCompletion(this.jobResponse.Job.ID);
}
}
catch (Exception exception)
{
this.HandleException(exception);
}
});
}