Added PackageAction Set for Update-AzSynapseSparkPool to support removing and adding packages in one action

src/Synapse/Synapse/ChangeLog.md

@@ -20,6 +20,7 @@
 
 ## Upcoming Release
 * Updated Azure.Core to 1.31.0.
+* Added PackageAction `Set` for `Update-AzSynapseSparkPool` to support removing and adding packages in one action
 
 ## Version 2.3.0
 * Upgraded Azure.Analytics.Synapse.Artifacts to 1.0.0-preview.17

src/Synapse/Synapse/Commands/ManagementCommands/SparkPool/UpdateAzureSynapseSparkPool.cs

@@ -267,9 +267,9 @@ public override void ExecuteCmdlet()
 
             if (this.IsParameterBound(c => c.PackageAction) && this.IsParameterBound(c => c.Package))
             {
-                if (this.PackageAction == SynapseConstants.PackageActionType.Add)
+                if (this.PackageAction == SynapseConstants.PackageActionType.Add || this.PackageAction == SynapseConstants.PackageActionType.Set)
                 {
-                    if (existingSparkPool.CustomLibraries == null)
+                    if (existingSparkPool.CustomLibraries == null || this.PackageAction == SynapseConstants.PackageActionType.Set)
                     {
                         existingSparkPool.CustomLibraries = new List<LibraryInfo>();
                     }

src/Synapse/Synapse/Models/SynapseConstants.cs

@@ -332,7 +332,8 @@ public class RepositoryType
         public enum PackageActionType
         {
             Add,
-            Remove
+            Remove,
+            Set
         }
         public enum UserAssignedManagedIdentityActionType
         {

src/Synapse/Synapse/help/Update-AzSynapseSparkPool.md

@@ -169,28 +169,42 @@ The first command retrieves an Apache Spark pool in Azure Synapse Analytics. The
 
 ### Example 14
 ```powershell
+$workspace_packages = Get-AzSynapseWorkspacePackage -WorkspaceName ContosoWorkspace
+
+$pool = Get-AzSynapseSparkPool -ResourceGroupName ContosoResourceGroup -WorkspaceName ContosoWorkspace -Name ContosoSparkPool
+$library_names = $pool.WorkspacePackages | Where-Object {$_.name -notlike "new_package-*"} | ForEach-Object {$_.name}
+$library_names += "new_package-2.0-py3-none-any.whl"
+
+$new_pool_packages = @($workspace_packages | Where-Object {$_.name -in $library_names})
+Update-AzSynapseSparkPool -ResourceGroupName ContosoResourceGroup -WorkspaceName ContosoWorkspace -Name ContosoSparkPool -PackageAction Set -Package $new_pool_packages
+```
+
+The first command retrieves the packages available in the workspace. The second command group retrieves the spark pool to get the packages currently linked to this pool and removes all versions of the package starting with `new_package-` from the retrieved list. The new version of the package is then added to this list. In the third group of commands the package list, containing only package names, is tranformed into a list of workspace packages by filtering the list of available workspace_packages accordingly and is then linked to the spark pool.
+
+### Example 15
+```powershell
 $config = Get-AzSynapseSparkConfiguration -WorkspaceName ContosoWorkspace -Name ContosoSparkConfig1
 Update-AzSynapseSparkPool -WorkspaceName ContosoWorkspace -Name ContosoSparkPool -Tag @{"key" = "value"} -NodeCount 5 -NodeSize Medium -SparkConfiguration $configs
 ```
 
 This command updates an Apache Spark pool in Azure Synapse Analytics and specify a Spark configuration for the Spark pool.
 
-### Example 15
+### Example 16
 ```powershell
 Update-AzSynapseSparkPool -WorkspaceName ContosoWorkspace -Name ContosoSparkPool -NodeSize small -ForceApplySetting
 ```
 
 This command updates an Apache Spark pool in Azure Synapse Analytics, set NodeSize to small for the spark pool and force stop any running jobs in the Spark pool to apply this new setting.
 
-### Example 16
+### Example 17
 ```powershell
 $pool = Get-AzSynapseSparkPool -ResourceGroupName ContosoResourceGroup -WorkspaceName ContosoWorkspace -Name ContosoSparkPool
 $pool | Update-AzSynapseSparkPool -PackageAction Remove -Package $pool.WorkspacePackages -ForceApplySetting
 ```
 
 The first command retrieves an Apache Spark pool in Azure Synapse Analytics. The second command removes all workspace packages that are linked to that Apache Spark pool and force stop any running jobs in the Spark pool to apply this new setting.
 
-### Example 17
+### Example 18
 ```powershell
 Update-AzSynapseSparkPool -WorkspaceName ContosoWorkspace -Name ContosoSparkPool -EnableIsolatedCompute $true -NodeSize XXXLarge
 ```
@@ -480,7 +494,7 @@ Package action must be specified when you add or remove a workspace package from
 Type: Microsoft.Azure.Commands.Synapse.Models.SynapseConstants+PackageActionType
 Parameter Sets: (All)
 Aliases:
-Accepted values: Add, Remove
+Accepted values: Add, Remove, Set
 
 Required: False
 Position: Named