Fix bug: Get-AzAks doesn't get all clusters (#12319)

* Fix bug: Get-AzAks doesn't get all clusters

* Update ChangeLog.md

Co-authored-by: Yabo Hu <[email protected]>

Author: erich-wang

src/Aks/Aks/ChangeLog.md

@@ -18,6 +18,7 @@
         - Additional information about change #1
 -->
 ## Upcoming Release
+* Fixed bug Get-AzAks doesn't get all clusters (#12296)
 
 ## Version 1.1.2
 

src/Aks/Aks/Commands/GetAzureRmAks.cs

@@ -13,13 +13,13 @@
 // ----------------------------------------------------------------------------------
 
 using System;
-using System.Collections.Generic;
 using System.Linq;
 using System.Management.Automation;
-using Microsoft.Azure.Management.ContainerService;
+
 using Microsoft.Azure.Commands.Aks.Models;
 using Microsoft.Azure.Commands.Aks.Properties;
 using Microsoft.Azure.Commands.ResourceManager.Common.ArgumentCompleters;
+using Microsoft.Azure.Management.ContainerService;
 using Microsoft.Azure.Management.Internal.Resources.Utilities.Models;
 
 namespace Microsoft.Azure.Commands.Aks
@@ -86,10 +86,13 @@ public override void ExecuteCmdlet()
                         WriteObject(PSMapper.Instance.Map<PSKubernetesCluster>(idCluster), true);
                         break;
                     case ResourceGroupParameterSet:
-                        var kubeClusters = string.IsNullOrEmpty(ResourceGroupName)
-                            ? Client.ManagedClusters.List()
-                            : Client.ManagedClusters.ListByResourceGroup(ResourceGroupName);
-                        WriteObject(kubeClusters.Select(PSMapper.Instance.Map<PSKubernetesCluster>), true);
+                        var kubeClusterList = string.IsNullOrEmpty(ResourceGroupName)
+                                    ? ListPaged(() => Client.ManagedClusters.List(),
+                                        nextPageLink => Client.ManagedClusters.ListNext(nextPageLink))
+                                    : ListPaged(() => Client.ManagedClusters.ListByResourceGroup(ResourceGroupName),
+                                        nextPageLink => Client.ManagedClusters.ListNext(nextPageLink));
+
+                        WriteObject(kubeClusterList.Select(PSMapper.Instance.Map<PSKubernetesCluster>), true);
                         break;
                     default:
                         throw new ArgumentException(Resources.ParameterSetError);

src/Aks/Aks/Commands/GetAzureRmAksNodePool.cs

@@ -14,11 +14,11 @@
 
 using System.Linq;
 using System.Management.Automation;
+
 using Microsoft.Azure.Commands.Aks.Models;
 using Microsoft.Azure.Commands.ResourceManager.Common.ArgumentCompleters;
 using Microsoft.Azure.Management.ContainerService;
 using Microsoft.Azure.Management.Internal.Resources.Utilities.Models;
-using Microsoft.WindowsAzure.Commands.Utilities.Common;
 
 namespace Microsoft.Azure.Commands.Aks.Commands
 {
@@ -77,7 +77,8 @@ public override void ExecuteCmdlet()
                 }
                 if (string.IsNullOrEmpty(Name))
                 {
-                    var pools = Client.AgentPools.List(ResourceGroupName, ClusterName);
+                    var pools = ListPaged(() => Client.AgentPools.List(ResourceGroupName, ClusterName),
+                        nextPageLink => Client.AgentPools.ListNext(nextPageLink));
                     WriteObject(pools.Select(PSMapper.Instance.Map<PSNodePool>), true);
                 }
                 else

src/Aks/Aks/Commands/KubeCmdletBase.cs

@@ -13,8 +13,10 @@
 // ----------------------------------------------------------------------------------
 
 using System;
+using System.Collections.Generic;
 using System.IO;
 using System.Management.Automation;
+
 using Microsoft.Azure.Commands.Aks.Properties;
 using Microsoft.Azure.Commands.Common.Authentication;
 using Microsoft.Azure.Commands.Common.Authentication.Abstractions;
@@ -24,6 +26,7 @@
 using Microsoft.Azure.Management.ContainerService;
 using Microsoft.Azure.Management.Internal.Resources;
 using Microsoft.Rest;
+
 using CloudException = Microsoft.Rest.Azure.CloudException;
 
 namespace Microsoft.Azure.Commands.Aks
@@ -83,5 +86,23 @@ private T BuildClient<T>(string endpoint = null, Func<T, T> postBuild = null) wh
             ".azure");
 
         protected string AcsSpFilePath => Path.Combine(AzConfigDir, "acsServicePrincipal.json");
+
+        protected static IList<T> ListPaged<T>(
+            Func<Rest.Azure.IPage<T>> listFirstPage,
+            Func<string, Rest.Azure.IPage<T>> listNextPage)
+        {
+            var resultsList = new List<T>();
+
+            var pagedResponse = listFirstPage();
+            resultsList.AddRange(pagedResponse);
+
+            while (!string.IsNullOrEmpty(pagedResponse.NextPageLink))
+            {
+                pagedResponse = listNextPage(pagedResponse.NextPageLink);
+                resultsList.AddRange(pagedResponse);
+            }
+
+            return resultsList;
+        }
     }
 }