Merge pull request #8 from AsrOneSdk/A2APolicyEnumerationCreationAndAssociation

A2A Policy enumeration, creation, association

Author: sriramvu

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Models/PSConstants.cs

@@ -152,6 +152,11 @@ public static class Constants
         /// </summary>
         public const string HyperVReplicaAzure = "HyperVReplicaAzure";
 
+        /// <summary>
+        /// Represents AzureToAzure string constant.
+        /// </summary>
+        public const string AzureToAzure = "AzureToAzure";
+
         /// <summary>
         /// Represents HyperVReplicaAzureReplicationDetails string constant.
         /// </summary>
@@ -356,6 +361,16 @@ public static class Constants
         /// JSON field: InstanceType
         /// </summary>
         public const string InstanceType = "InstanceType";
+
+        /// <summary>
+        /// Enable.
+        /// </summary>
+        public const string Enable = "Enable";
+
+        /// <summary>
+        /// Disable.
+        /// </summary>
+        public const string Disable = "Disable";
     }
 
      /// <summary>

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Models/PSObjects.cs

@@ -666,6 +666,26 @@ public ASRPolicy(Policy policy)
 
                 this.ReplicationProviderSettings = replicationProviderSettings;
             }
+            else if (policy.Properties.ProviderSpecificDetails.InstanceType == Constants.AzureToAzure)
+            {
+                A2APolicyDetails details =
+                    (A2APolicyDetails)policy.Properties.ProviderSpecificDetails;
+
+                ASRAzureToAzurePolicyDetails replicationProviderSettings =
+                    new ASRAzureToAzurePolicyDetails();
+
+                replicationProviderSettings.AppConsistentFrequencyInMinutes =
+                    details.AppConsistentFrequencyInMinutes;
+                replicationProviderSettings.CrashConsistentFrequencyInMinutes =
+                    details.CrashConsistentFrequencyInMinutes;
+                replicationProviderSettings.MultiVmSyncStatus =
+                    details.MultiVmSyncStatus;
+                replicationProviderSettings.RecoveryPointHistory = details.RecoveryPointHistory;
+                replicationProviderSettings.RecoveryPointThresholdInMinutes =
+                    details.RecoveryPointThresholdInMinutes;
+
+                this.ReplicationProviderSettings = replicationProviderSettings;
+            }
         }
 
         #region Properties
@@ -790,6 +810,32 @@ public class ASRHyperVReplicaAzurePolicyDetails : ASRPolicyProviderSettingsDetai
         public int ReplicationFrequencyInSeconds { get; set; }
     }
 
+    /// <summary>
+    /// ASR AzureToAzure policy details.
+    /// </summary>
+    public class ASRAzureToAzurePolicyDetails : ASRPolicyProviderSettingsDetails
+    {
+        // Summary:
+        //     Optional.
+        public int AppConsistentFrequencyInMinutes { get; set; }
+        //
+        // Summary:
+        //     Optional.
+        public int CrashConsistentFrequencyInMinutes { get; set; }
+        //
+        // Summary:
+        //     Optional.
+        public string MultiVmSyncStatus { get; set; }
+        //
+        // Summary:
+        //     Optional.
+        public int RecoveryPointHistory { get; set; }
+        //
+        // Summary:
+        //     Optional.
+        public int RecoveryPointThresholdInMinutes { get; set; }
+    }
+
     /// <summary>
     /// ASR Customer Storage Account.
     /// </summary>

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Models/PSParameterSets.cs

@@ -229,6 +229,11 @@ internal static class ASRParameterSets
         /// </summary>
         internal const string EnterpriseToAzure = "EnterpriseToAzure";
 
+        /// <summary>
+        /// To define Azure to Azure parameter set.
+        /// </summary>
+        internal const string AzureToAzure = "AzureToAzure";
+
         /// <summary>
         /// Mapping between HyperV Site to Azure.
         /// </summary>

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Policy/NewAzureSiteRecoveryPolicy.cs

@@ -40,18 +40,21 @@ public class NewAzureSiteRecoveryPolicy : SiteRecoveryCmdletBase
         /// </summary>
         [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true)]
         [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure, Mandatory = true)]
         public string Name { get; set; }
 
         /// <summary>
         /// Gets or sets Replication Provider of the Policy.
         /// </summary>
         [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToEnterprise, Mandatory = true)]
         [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure, Mandatory = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure, Mandatory = true)]
         [ValidateNotNullOrEmpty]
         [ValidateSet(
             Constants.HyperVReplica2012R2,
             Constants.HyperVReplica2012,
-            Constants.HyperVReplicaAzure)]
+            Constants.HyperVReplicaAzure,
+            Constants.AzureToAzure)]
         public string ReplicationProvider { get; set; }
 
         /// <summary>
@@ -147,6 +150,41 @@ public class NewAzureSiteRecoveryPolicy : SiteRecoveryCmdletBase
         [Parameter(ParameterSetName = ASRParameterSets.EnterpriseToAzure)]
         public SwitchParameter Encrypt { get; set; }
 
+        /* TODO: SriramVu:
+         * Revise the mandatory inputs for A2A prameterset and use defaults if req.
+         * Check out the possible allowed values (if any) and use ValidateSet.
+         */
+
+        /// <summary>
+        /// Gets or sets Recovery point threshold in minutes.
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure, Mandatory = true)]
+        public int RecoveryPointThresholdInMinutes { get; set; }
+
+        /// <summary>
+        /// Gets or sets Recovery point history.
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure, Mandatory = true)]
+        public int RecoveryPointHistory { get; set; }
+
+        /// <summary>
+        /// Gets or sets Crash consistent frequency in minutes.
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure, Mandatory = true)]
+        public int CrashConsistentFrequencyInMinutes { get; set; }
+
+        /// <summary>
+        /// Gets or sets App consistent frequency in minutes.
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure, Mandatory = true)]
+        public int AppConsistentFrequencyInMinutes { get; set; }
+
+        /// <summary>
+        /// Gets or sets EnableMultiVmSync parameter. On passing, MultiVmSync will be enabled.
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.AzureToAzure)]
+        public SwitchParameter EnableMultiVmSync { get; set; }
+
         #endregion Parameters
 
         /// <summary>
@@ -164,6 +202,9 @@ public override void ExecuteSiteRecoveryCmdlet()
                 case ASRParameterSets.EnterpriseToAzure:
                     this.EnterpriseToAzurePolicyObject();
                     break;
+                case ASRParameterSets.AzureToAzure:
+                    this.CreateAzureToAzurePolicy();
+                    break;
             }
         }
 
@@ -294,5 +335,52 @@ private void EnterpriseToAzurePolicyObject()
 
             WriteObject(new ASRJob(jobResponse.Job));
         }
+
+        /// <summary>
+        /// Creates an A2A Policy.
+        /// </summary>
+        private void CreateAzureToAzurePolicy()
+        {
+            if (string.Compare(
+                this.ReplicationProvider,
+                Constants.AzureToAzure,
+                StringComparison.OrdinalIgnoreCase) != 0)
+            {
+                throw new InvalidOperationException(
+                    string.Format(
+                    Properties.Resources.IncorrectReplicationProvider,
+                    this.ReplicationProvider));
+            }
+
+            var a2aPolicyCreationInput = new A2APolicyCreationInput()
+            {
+                AppConsistentFrequencyInMinutes = this.AppConsistentFrequencyInMinutes,
+                CrashConsistentFrequencyInMinutes = this.CrashConsistentFrequencyInMinutes,
+                MultiVmSyncStatus = this.EnableMultiVmSync ? Constants.Enable: Constants.Disable,
+                RecoveryPointHistory = this.RecoveryPointHistory,
+                RecoveryPointThresholdInMinutes = this.RecoveryPointThresholdInMinutes
+            };
+
+            var createPolicyInputProperties = new CreatePolicyInputProperties()
+            {
+                ProviderSpecificInput = a2aPolicyCreationInput
+            };
+
+            var createPolicyInput = new CreatePolicyInput()
+            {
+                Properties = createPolicyInputProperties
+            };
+
+            LongRunningOperationResponse response =
+                RecoveryServicesClient.CreatePolicy(this.Name, createPolicyInput);
+
+            string jobId = PSRecoveryServicesClient.GetJobIdFromReponseLocation(response.Location);
+
+            JobResponse jobResponse =
+                RecoveryServicesClient
+                .GetAzureSiteRecoveryJobDetails(jobId);
+
+            WriteObject(new ASRJob(jobResponse.Job));
+        }
     }
 }

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/ProtectionContainerMapping/NewAzureRmSiteRecoveryProtectionContainerMapping.cs

@@ -80,6 +80,9 @@ public override void ExecuteSiteRecoveryCmdlet()
                 case ASRParameterSets.EnterpriseToEnterprise:
                     this.EnterpriseToEnterpriseAssociation();
                     break;
+                ////case ASRParameterSets.AzureToAzure:
+                ////    this.EnterpriseToEnterpriseAssociation();
+                ////    break;
             }
         }
 
@@ -88,12 +91,18 @@ public override void ExecuteSiteRecoveryCmdlet()
         /// </summary>
         private void EnterpriseToEnterpriseAssociation()
         {
+            // TODO: SriramVu: Same parameter set can be used for both E2E and A2A - update 
+            // methods and validations.
+
             if (string.Compare(
                 this.Policy.ReplicationProvider,
                 Constants.HyperVReplica2012,
                 StringComparison.OrdinalIgnoreCase) != 0 && string.Compare(
                 this.Policy.ReplicationProvider,
                 Constants.HyperVReplica2012R2,
+                StringComparison.OrdinalIgnoreCase) != 0 && string.Compare(
+                this.Policy.ReplicationProvider,
+                "A2A",
                 StringComparison.OrdinalIgnoreCase) != 0)
             {
                 throw new InvalidOperationException(
@@ -125,6 +134,25 @@ private void EnterpriseToAzureAssociation()
             Associate(Constants.AzureContainer);
         }
 
+        /// <summary>
+        /// Associates Policy with Azure based protection containers
+        /// </summary>
+        private void AzureToAzureAssociation()
+        {
+            if (string.Compare(
+                this.Policy.ReplicationProvider,
+                Constants.AzureToAzure,
+                StringComparison.OrdinalIgnoreCase) != 0)
+            {
+                throw new InvalidOperationException(
+                    string.Format(
+                    Properties.Resources.IncorrectReplicationProvider,
+                    this.Policy.ReplicationProvider));
+            }
+
+            Associate(this.RecoveryProtectionContainer.ID);
+        }
+
         /// <summary>
         /// Helper to configure cloud
         /// </summary>