Az.PolicyInsights

2.0.0

PolicyInsights.Autorest/custom/GeneratedOverrides/GetAzPolicyEvent_List7.cs

using Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Runtime;
using Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Runtime.PowerShell;
using System;
 
namespace Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Cmdlets
{
    public partial class GetAzPolicyEvent_List7
    {
        /// <summary>
        /// Holds amount returned from call(s) to check against Top parameter value.
        /// </summary>
        private long totalResponseCount = 0;
 
        /// <summary>
        /// <c>overrideOnOk</c> will be called before the regular onOk has been processed, allowing customization of what happens
        /// on that response. Implement this method in a partial class to enable this behavior
        /// </summary>
        /// <param name="responseMessage">the raw response message as an global::System.Net.Http.HttpResponseMessage.</param>
        /// <param name="response">the body result as a <see cref="Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Models.IPolicyEventsQueryResults">Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Models.IPolicyEventsQueryResults</see>
        /// from the remote call</param>
        /// <param name="returnNow">/// Determines if the rest of the onOk method should be processed, or if the method should return
        /// immediately (set to true to skip further processing )</param>
        partial void overrideOnOk(global::System.Net.Http.HttpResponseMessage responseMessage, global::System.Threading.Tasks.Task<Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Models.IPolicyEventsQueryResults> response, ref global::System.Threading.Tasks.Task<bool> returnNow)
        {
            // implementing this override in order to:
            // - prevent the nextLink from being called when we've reached the "Top" amount
            // - change the request method from GET to POST on subsquent paging calls, as the API expects a POST request for this cmdlet
 
            // set this immediately so the generated paging client call is never made - the get request is incorrect, it needs to be a post request
            returnNow = global::System.Threading.Tasks.Task<bool>.FromResult(true);
 
            var result = response.GetAwaiter().GetResult();
            // response should be returning an array of some kind. +Pageable
            // pageable / value / @odata.nextLink
            if (null != result.Value)
            {
                if (0 == _responseSize && 1 == result.Value.Count)
                {
                    _firstResponse = result.Value[0];
                    _responseSize = 1;
                }
                else
                {
                    if (1 == _responseSize)
                    {
                        // Flush buffer
                        WriteObject(_firstResponse.AddMultipleTypeNameIntoPSObject());
                    }
                    var values = new System.Collections.Generic.List<System.Management.Automation.PSObject>();
                    foreach (var value in result.Value)
                    {
                        values.Add(value.AddMultipleTypeNameIntoPSObject());
                    }
                    WriteObject(values, true);
                    _responseSize = 2;
                }
            }
            else
            {
                return;
            }
 
            // if there is a Top value and it has been reached, then we don't need to call the nextLink
            if(this.InvocationInformation.BoundParameters.ContainsKey("Top"))
            {
                int responseCount = result.Value.Count;
                this.totalResponseCount += responseCount;
                if(this.totalResponseCount >= this.Top)
                {
                    // set nextLink to null so the method doesn't get caught in a loop
                    _nextLink = null;
                    return;
                }
                _nextLink = result.OdataNextLink;
            }
            else
            {
                _nextLink = result.OdataNextLink;
            }
 
            if (_isFirst)
            {
                _isFirst = false;
                while (!String.IsNullOrEmpty(_nextLink))
                {
                    if (responseMessage.RequestMessage is System.Net.Http.HttpRequestMessage requestMessage)
                    {
                        requestMessage = requestMessage.Clone(new global::System.Uri(_nextLink), Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Runtime.Method.Post);
                        ((Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Runtime.IEventListener)this).Signal(Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Runtime.Events.FollowingNextLink).Wait();
                        if (((Microsoft.Azure.PowerShell.Cmdlets.PolicyInsights.Runtime.IEventListener)this).Token.IsCancellationRequested) { return; }
                        this.Client.PolicyEventsListQueryResultsForResourceGroupLevelPolicyAssignment_Call(requestMessage, onOk, onDefault, this, Pipeline).Wait();
                    }
                }
            }
        }
    }
}