Parsing Skip token for next call (#21600)

* Parsing Skip token for next call * Change log
Azure · Apr 20, 2023 · 4e7f42b · 4e7f42b
1 parent 2d6db97
commit 4e7f42b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/Billing/Billing/ChangeLog.md b/src/Billing/Billing/ChangeLog.md
@@ -18,6 +18,7 @@
         - Additional information about change #1
 -->
 ## Upcoming Release
+* Fixed skip token for Consumption PriceSheet cmdlet
 
 ## Version 2.0.1
 * Fixed pagination for `Get-AzConsumptionPriceSheet` cmdlet

diff --git a/src/Billing/Consumption/Cmdlets/GetAzureRmConsumptionPriceSheet.cs b/src/Billing/Consumption/Cmdlets/GetAzureRmConsumptionPriceSheet.cs
@@ -68,6 +68,17 @@ public override void ExecuteCmdlet()
                             expand, skipToken, numberToFetch);
                         UpdateResult(result, priceSheet);
                         nextLink = priceSheet?.NextLink;
+
+                        if (!string.IsNullOrWhiteSpace(nextLink))
+                        {
+                            string queryString = new Uri(nextLink).Query;
+                            var queryDictionary = System.Web.HttpUtility.ParseQueryString(queryString);
+
+                            if (queryDictionary.AllKeys.Contains("skipToken"))
+                            {
+                                skipToken = queryDictionary["skipToken"];
+                            }
+                        }
                     } while (!this.Top.HasValue && !string.IsNullOrWhiteSpace(nextLink));                    
                 }
                 else
@@ -77,6 +88,17 @@ public override void ExecuteCmdlet()
                         priceSheet = ConsumptionManagementClient.PriceSheet.Get(expand, skipToken, numberToFetch);
                         UpdateResult(result, priceSheet);
                         nextLink = priceSheet?.NextLink;
+
+                        if (!string.IsNullOrWhiteSpace(nextLink))
+                        {
+                            string queryString = new Uri(nextLink).Query;
+                            var queryDictionary = System.Web.HttpUtility.ParseQueryString(queryString);
+
+                            if (queryDictionary.AllKeys.Contains("skipToken"))
+                            {
+                                skipToken = queryDictionary["skipToken"];
+                            }
+                        }
                     } while (!this.Top.HasValue && !string.IsNullOrWhiteSpace(nextLink));
                 }
             }