[DataFactory]Added PublicNetworkAccess to Update_AzDataFactoryV2 Command (Azure#16087)

* [DataFactory]Added PublicNetworkAccess to Update_AzDataFactoryV2 Command

* update

* update

* update

* remove

* update

* update

* update

Author: Jingshu923

src/DataFactory/DataFactoryV2/Changelog.md

@@ -18,6 +18,7 @@
         - Additional information about change #1
 -->
 ## Upcoming Release
+* Added PublicNetworkAccess to Update_AzDataFactoryV2 Command
 * Updated ADF .Net SDK version to 4.26.0
 
 ## Version 1.15.0

src/DataFactory/DataFactoryV2/DataFactories/UpdateAzureDataFactoryCommand.cs

@@ -23,7 +23,7 @@
 
 namespace Microsoft.Azure.Commands.DataFactoryV2
 {
-    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "DataFactoryV2", DefaultParameterSetName = ParameterSetNames.ByFactoryName, SupportsShouldProcess = true),OutputType(typeof(PSDataFactory))]
+    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "DataFactoryV2", DefaultParameterSetName = ParameterSetNames.ByFactoryName, SupportsShouldProcess = true), OutputType(typeof(PSDataFactory))]
     public class UpdateAzureDataFactoryCommand : DataFactoryBaseCmdlet
     {
         [Parameter(ParameterSetName = ParameterSetNames.ByFactoryName, Position = 0, Mandatory = true,
@@ -42,6 +42,11 @@ public class UpdateAzureDataFactoryCommand : DataFactoryBaseCmdlet
         #endregion
         public string IdentityType { get; set; }
 
+        #region Attributes
+        [Parameter(Mandatory = false, HelpMessage = Constants.HelpPublicNetworkAccess)]
+        #endregion
+        public string PublicNetworkAccess { get; set; }
+
         #region Attributes
         [Parameter(Mandatory = false, HelpMessage = Constants.HelpUserAssignedIdenty)]
         #endregion
@@ -121,9 +126,16 @@ public override void ExecuteCmdlet()
                 encryption = new EncryptionConfiguration(this.EncryptionKeyName, this.EncryptionVaultBaseUrl, this.EncryptionKeyVersion, cmkIdentity);
             }
 
+            string publicNetworkAccess = Management.DataFactory.Models.PublicNetworkAccess.Enabled;
+            if (!string.IsNullOrWhiteSpace(this.PublicNetworkAccess))
+            {
+                publicNetworkAccess = this.PublicNetworkAccess;
+            }
+
             var parameters = new UpdatePSDataFactoryParameters()
             {
                 ResourceGroupName = ResourceGroupName,
+                PublicNetworkAccess = publicNetworkAccess,
                 DataFactoryName = Name,
                 EncryptionConfiguration = encryption,
                 FactoryIdentity = factoryIdentity,

src/DataFactory/DataFactoryV2/Models/DataFactoryClient.DataFactories.cs

@@ -86,7 +86,7 @@ public List<PSDataFactory> ListDataFactoriesBySubscription(DataFactoryFilterOpti
 
             if (response != null)
             {
-                dataFactories.AddRange(response.Select(df => 
+                dataFactories.AddRange(response.Select(df =>
                                                         {
                                                             var parsedResourceId = new ResourceIdentifier(df.Id);
                                                             var ResourceGroupName = parsedResourceId.ResourceGroupName;
@@ -115,7 +115,7 @@ public List<PSDataFactory> FilterPSDataFactories(DataFactoryFilterOptions filter
             }
             else
             {
-                if(filterOptions.ResourceGroupName == null && filterOptions.DataFactoryName != null)
+                if (filterOptions.ResourceGroupName == null && filterOptions.DataFactoryName != null)
                 {
                     throw new Exception("ResourceGroupName name can't be null if factory name is not due to parameter sets. Should never reach this point");
                 }
@@ -173,7 +173,7 @@ public PSDataFactory CreatePSDataFactory(CreatePSDataFactoryParameters parameter
 
         public PSDataFactory UpdatePSDataFactory(UpdatePSDataFactoryParameters parameters)
         {
-            var updateParams = new FactoryUpdateParameters(parameters.Tags?.ToDictionary());
+            var updateParams = new FactoryUpdateParameters(parameters.Tags?.ToDictionary(), publicNetworkAccess: parameters.PublicNetworkAccess);
             return new PSDataFactory(
                 this.DataFactoryManagementClient.Factories.Update(
                     parameters.ResourceGroupName,

src/DataFactory/DataFactoryV2/Models/UpdatePSDataFactoryParameters.cs

@@ -20,5 +20,7 @@ namespace Microsoft.Azure.Commands.DataFactoryV2
     public class UpdatePSDataFactoryParameters : DataFactoryParametersBase
     {
         public Hashtable Tags { get; set; }
+
+        public string PublicNetworkAccess { get; set; }
     }
 }

src/DataFactory/DataFactoryV2/help/Update-AzDataFactoryV2.md

@@ -157,6 +157,20 @@ Accept pipeline input: False
 Accept wildcard characters: False
 ```
 
+### -PublicNetworkAccess
+Whether or not public network access is allowed for the data factory. Options are 'Enabled' or 'Disabled'
+
+```yaml
+Type: System.String
+Parameter Sets: (All)
+Aliases:
+Required: False
+Position: Named
+Default value: Enabled
+Accept pipeline input: False
+Accept wildcard characters: False
+```
+
 ### -InputObject
 The data factory object.