public override void ExecuteCmdlet()
{
ExecutionBlock(() =>
{
base.ExecuteCmdlet();
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(VaultId);
string vaultName = resourceIdentifier.ResourceName;
string resourceGroupName = resourceIdentifier.ResourceGroupName;
string targetVaultName = "";
string targetResourceGroupName = "";
if (!OriginalWorkloadRestore.IsPresent && !AlternateWorkloadRestore.IsPresent && !RestoreAsFiles.IsPresent)
{
throw new ArgumentException(string.Format(Resources.AzureWorkloadRestoreLocationException));
}
AzureWorkloadRecoveryConfig azureWorkloadRecoveryConfig = GetConfigObject();
azureWorkloadRecoveryConfig.SourceResourceId = Item != null ? Item.SourceResourceId : GetResourceId();
DateTime currentTime = DateTime.Now;
TimeSpan timeSpan = DateTime.UtcNow - new DateTime(1970, 1, 1);
int offset = (int)timeSpan.TotalSeconds;
string targetDb = "";
if (AlternateWorkloadRestore.IsPresent || RestoreAsFiles.IsPresent)
{
if (TargetContainer == null)
{
throw new ArgumentNullException("TargetContainer", Resources.TargetContainerRequiredException);
}
azureWorkloadRecoveryConfig.TargetVirtualMachineId = (TargetContainer as AzureVmWorkloadContainer).SourceResourceId;
}
if (TargetItem != null)
{
// getting Target vault/RG for TargetPhysicalPath in CRR
Dictionary <UriEnums, string> keyValueDict = HelperUtils.ParseUri(TargetItem.Id);
targetResourceGroupName = HelperUtils.GetResourceGroupNameFromId(keyValueDict, TargetItem.Id);
targetVaultName = HelperUtils.GetVaultNameFromId(keyValueDict, TargetItem.Id);
// check if the TragetItem provided is of workload type
if (!string.Equals(((AzureWorkloadProtectableItem)TargetItem).ProtectableItemType,
ProtectableItemType.SQLInstance.ToString()))
{
throw new ArgumentException(string.Format(Resources.AzureWorkloadRestoreProtectableItemException));
}
}
if (ParameterSetName == RpParameterSet)
{
Dictionary <UriEnums, string> keyValueDict = HelperUtils.ParseUri(RecoveryPoint.Id);
string itemUri = HelperUtils.GetProtectedItemUri(keyValueDict, RecoveryPoint.Id);
targetDb = itemUri.Split(new string[] { ";" }, StringSplitOptions.None)[2];
}
if (OriginalWorkloadRestore.IsPresent)
{
azureWorkloadRecoveryConfig.RestoredDBName = Item != null ?
((AzureWorkloadSQLDatabaseProtectedItem)Item).FriendlyName : targetDb;
azureWorkloadRecoveryConfig.OverwriteWLIfpresent = "No";
azureWorkloadRecoveryConfig.NoRecoveryMode = "Disabled";
azureWorkloadRecoveryConfig.ContainerId = Item != null ?
GetContainerId(Item.Id) : GetContainerId(GetItemId(RecoveryPoint.Id));
}
else if (AlternateWorkloadRestore.IsPresent && Item == null)
{
azureWorkloadRecoveryConfig.RestoredDBName =
GetRestoredDBName(RecoveryPoint.ItemName, currentTime);
azureWorkloadRecoveryConfig.OverwriteWLIfpresent = "No";
azureWorkloadRecoveryConfig.NoRecoveryMode = "Disabled";
List <SQLDataDirectoryMapping> targetPhysicalPath = new List <SQLDataDirectoryMapping>();
//Get target workload item
ODataQuery <BMSWorkloadItemQueryObject> queryParams = null;
string backupManagementType = ServiceClientModel.BackupManagementType.AzureWorkload;
queryParams = new ODataQuery <BMSWorkloadItemQueryObject>(
q => q.WorkloadItemType == WorkloadItemType.SQLInstance &&
q.BackupManagementType == backupManagementType);
List <WorkloadItemResource> itemResponses;
if (targetVaultName != "" && targetResourceGroupName != "")
{
itemResponses = ServiceClientAdapter.ListWorkloadItem(
TargetItem.ContainerName,
queryParams,
vaultName: targetVaultName,
resourceGroupName: targetResourceGroupName);
}
else
{
itemResponses = ServiceClientAdapter.ListWorkloadItem(
TargetItem.ContainerName,
queryParams,
vaultName: vaultName,
resourceGroupName: resourceGroupName);
}
foreach (var itemResponse in itemResponses)
{
if (string.Compare(((AzureVmWorkloadSQLInstanceWorkloadItem)itemResponse.Properties).FriendlyName,
((AzureWorkloadProtectableItem)TargetItem).FriendlyName) == 0 &&
string.Compare(((AzureVmWorkloadSQLInstanceWorkloadItem)itemResponse.Properties).ServerName,
((AzureWorkloadProtectableItem)TargetItem).ServerName) == 0)
{
string itemId = GetItemId(RecoveryPoint.Id);
IList <SQLDataDirectory> dataDirectoryPaths = GetRpDetails(vaultName, resourceGroupName);
foreach (var dataDirectoryPath in dataDirectoryPaths)
{
targetPhysicalPath.Add(new SQLDataDirectoryMapping()
{
MappingType = dataDirectoryPath.Type,
SourceLogicalName = dataDirectoryPath.LogicalName,
SourcePath = dataDirectoryPath.Path,
TargetPath = GetTargetPath(dataDirectoryPath.Path, dataDirectoryPath.LogicalName, dataDirectoryPath.Type,
((AzureVmWorkloadSQLInstanceWorkloadItem)itemResponse.Properties).DataDirectoryPaths
as List <SQLDataDirectory>, offset)
});
}
break;
}
}
azureWorkloadRecoveryConfig.targetPhysicalPath = targetPhysicalPath;
azureWorkloadRecoveryConfig.ContainerId = GetContainerId(TargetItem.Id);
}
else if (Item != null && TargetItem != null)
{
azureWorkloadRecoveryConfig.RestoredDBName =
GetRestoredDBName(Item.Name, currentTime);
azureWorkloadRecoveryConfig.OverwriteWLIfpresent = "No";
azureWorkloadRecoveryConfig.NoRecoveryMode = "Disabled";
List <SQLDataDirectoryMapping> targetPhysicalPath = new List <SQLDataDirectoryMapping>();
//Get target workload item
ODataQuery <BMSWorkloadItemQueryObject> queryParams = null;
string backupManagementType = ServiceClientModel.BackupManagementType.AzureWorkload;
queryParams = new ODataQuery <BMSWorkloadItemQueryObject>(
q => q.WorkloadItemType == WorkloadItemType.SQLInstance &&
q.BackupManagementType == backupManagementType);
List <WorkloadItemResource> itemResponses;
if (targetVaultName != "" && targetResourceGroupName != "")
{
itemResponses = ServiceClientAdapter.ListWorkloadItem(
TargetItem.ContainerName,
queryParams,
vaultName: targetVaultName,
resourceGroupName: targetResourceGroupName);
}
else
{
itemResponses = ServiceClientAdapter.ListWorkloadItem(
TargetItem.ContainerName,
queryParams,
vaultName: vaultName,
resourceGroupName: resourceGroupName);
}
foreach (var itemResponse in itemResponses)
{
if (string.Compare(((AzureVmWorkloadSQLInstanceWorkloadItem)itemResponse.Properties).FriendlyName,
((AzureWorkloadProtectableItem)TargetItem).FriendlyName) == 0 &&
string.Compare(((AzureVmWorkloadSQLInstanceWorkloadItem)itemResponse.Properties).ServerName,
((AzureWorkloadProtectableItem)TargetItem).ServerName) == 0)
{
List <SQLDataDirectory> dataDirectory = GetDataDirectory(vaultName, resourceGroupName, Item.Id, PointInTime);
foreach (var dataDirectoryPath in dataDirectory)
{
targetPhysicalPath.Add(new SQLDataDirectoryMapping()
{
MappingType = dataDirectoryPath.Type,
SourceLogicalName = dataDirectoryPath.LogicalName,
SourcePath = dataDirectoryPath.Path,
TargetPath = GetTargetPath(dataDirectoryPath.Path, dataDirectoryPath.LogicalName, dataDirectoryPath.Type,
((AzureVmWorkloadSQLInstanceWorkloadItem)itemResponse.Properties).DataDirectoryPaths
as List <SQLDataDirectory>, offset)
});
}
break;
}
}
azureWorkloadRecoveryConfig.targetPhysicalPath = targetPhysicalPath;
azureWorkloadRecoveryConfig.ContainerId = GetContainerId(TargetItem.Id);
}
else if (RestoreAsFiles.IsPresent)
{
azureWorkloadRecoveryConfig.OverwriteWLIfpresent = "No";
azureWorkloadRecoveryConfig.NoRecoveryMode = "Disabled";
azureWorkloadRecoveryConfig.ContainerId = (TargetContainer as AzureVmWorkloadContainer).Id;
azureWorkloadRecoveryConfig.RestoreRequestType = "Alternate WL Restore";
azureWorkloadRecoveryConfig.RecoveryMode = "FileRecovery";
azureWorkloadRecoveryConfig.FilePath = FilePath;
azureWorkloadRecoveryConfig.FullRP = FromFull;
}
RecoveryConfigBase baseobj = azureWorkloadRecoveryConfig;
WriteObject(baseobj);
});
}
