Add license type for managed integration runtime

Author: HaoQian-MS

src/ResourceManager/DataFactories/Commands.DataFactoryV2/Constants.cs

@@ -108,6 +108,8 @@ internal static class Constants
 
         public const string HelpIntegrationRuntimeMaxParallelExecutionsPerNode = "Maximum parallel execution count per node for a managed dedicated integration runtime.";
 
+        public const string HelpIntegrationRuntimeLicenseType = "The license type that you want to select for the SSIS IR. There are two types: LicenseIncluded or BasePrice. If you are qualified for the Azure Hybrid Use Benefit (AHUB) pricing, please select BasePrice. If not, please select LicenseIncluded.";
+
         public const string HelpIntegrationRuntimeAutoUpdate = "Enable or disable the self-hosted integration runtime auto-update.";
 
         public const string HelpIntegrationRuntimeAutoUpdateTime = "The time of the day for the self-hosted integration runtime auto-update.";
@@ -167,5 +169,9 @@ internal static class Constants
         public const string IntegrationRuntimeAutoUpdateDisabled = "Off";
 
         public const string TriggerName = "TriggerName";
+
+        public const string IntegrationRuntimeLicenseIncluded = "LicenseIncluded";
+
+        public const string IntegrationRuntimeBasePrice = "BasePrice";
     }
 }

src/ResourceManager/DataFactories/Commands.DataFactoryV2/IntegrationRuntimes/SetAzureDataFactoryIntegrationRuntimeCommand.cs

@@ -102,6 +102,16 @@ public class SetAzureDataFactoryIntegrationRuntimeCommand : IntegrationRuntimeCm
             HelpMessage = Constants.HelpIntegrationRuntimeMaxParallelExecutionsPerNode)]
         public int? MaxParallelExecutionsPerNode { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            HelpMessage = Constants.HelpIntegrationRuntimeLicenseType)]
+        [ValidateNotNullOrEmpty]
+        [ValidateSet(
+            Constants.IntegrationRuntimeLicenseIncluded,
+            Constants.IntegrationRuntimeBasePrice,
+            IgnoreCase = true)]
+        public string LicenseType { get; set; }
+
         [Parameter(
             Mandatory = false, HelpMessage = Constants.HelpDontAskConfirmation)]
         public SwitchParameter Force { get; set; }
@@ -338,6 +348,16 @@ private void HandleManagedIntegrationRuntime(ManagedIntegrationRuntime integrati
                 }
             }
 
+            if (!string.IsNullOrWhiteSpace(LicenseType))
+            {
+                if (integrationRuntime.SsisProperties == null)
+                {
+                    integrationRuntime.SsisProperties = new IntegrationRuntimeSsisProperties();
+                }
+
+                integrationRuntime.SsisProperties.LicenseType = LicenseType;
+            }
+
             integrationRuntime.Validate();
         }
     }

src/ResourceManager/DataFactories/Commands.DataFactoryV2/Models/PSManagedIntegrationRuntime.cs

@@ -60,5 +60,7 @@ public PSManagedIntegrationRuntime(
         public string Subnet => ManagedIntegrationRuntime.ComputeProperties?.VNetProperties?.Subnet;
 
         public string State => ManagedIntegrationRuntime.State;
+
+        public string LicenseType => ManagedIntegrationRuntime.SsisProperties?.LicenseType;
     }
 }