/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 * 
 *  http://aws.amazon.com/apache2.0
 * 
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

/*
 * Do not modify this file. This file is generated from the timestream-query-2018-11-01.normal.json service model.
 */

using System;
using System.Collections.Generic;
using System.Text;
using System.Collections;
using System.Threading;
using System.Threading.Tasks;
using Amazon.Runtime;
 
namespace Amazon.TimestreamQuery.Model
{
    /// <summary>
    /// Base class for Query paginators.
    /// </summary>
    internal sealed partial class QueryPaginator : IPaginator<QueryResponse>, IQueryPaginator
    {
        private readonly IAmazonTimestreamQuery _client;
        private readonly QueryRequest _request;
        private int _isPaginatorInUse = 0;
        
        /// <summary>
        /// Enumerable containing all full responses for the operation
        /// </summary>
        public IPaginatedEnumerable<QueryResponse> Responses => new PaginatedResponse<QueryResponse>(this);

        /// <summary>
        /// Enumerable containing all of the Rows
        /// </summary>
        public IPaginatedEnumerable<Row> Rows => 
            new PaginatedResultKeyResponse<QueryResponse, Row>(this, (i) => i.Rows);

        internal QueryPaginator(IAmazonTimestreamQuery client, QueryRequest request)
        {
            this._client = client;
            this._request = request;
        }
#if BCL
        IEnumerable<QueryResponse> IPaginator<QueryResponse>.Paginate()
        {
            if (Interlocked.Exchange(ref _isPaginatorInUse, 1) != 0)
            {
                throw new System.InvalidOperationException("Paginator has already been consumed and cannot be reused. Please create a new instance.");
            }
            PaginatorUtils.SetUserAgentAdditionOnRequest(_request);
            var nextToken = _request.NextToken;
            QueryResponse response;
            do
            {
                _request.NextToken = nextToken;
                response = _client.Query(_request);
                nextToken = response.NextToken;
                yield return response;
            }
            while (!string.IsNullOrEmpty(nextToken));
        }
#endif
#if AWS_ASYNC_ENUMERABLES_API
        async IAsyncEnumerable<QueryResponse> IPaginator<QueryResponse>.PaginateAsync(CancellationToken cancellationToken = default)
        {
            if (Interlocked.Exchange(ref _isPaginatorInUse, 1) != 0)
            {
                throw new System.InvalidOperationException("Paginator has already been consumed and cannot be reused. Please create a new instance.");
            }
            PaginatorUtils.SetUserAgentAdditionOnRequest(_request);
            var nextToken = _request.NextToken;
            QueryResponse response;
            do
            {
                _request.NextToken = nextToken;
                response = await _client.QueryAsync(_request, cancellationToken).ConfigureAwait(false);
                nextToken = response.NextToken;
                cancellationToken.ThrowIfCancellationRequested();
                yield return response;
            }
            while (!string.IsNullOrEmpty(nextToken));
        }
#endif
    }
}