software.amazon.awssdk.services.athena.paginators.ListTagsForResourceIterable Maven / Gradle / Ivy

/*
 * 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.
 */

package software.amazon.awssdk.services.athena.paginators;

import java.util.Collections;
import java.util.Iterator;
import java.util.function.Function;
import software.amazon.awssdk.annotations.Generated;
import software.amazon.awssdk.core.pagination.sync.PaginatedItemsIterable;
import software.amazon.awssdk.core.pagination.sync.PaginatedResponsesIterator;
import software.amazon.awssdk.core.pagination.sync.SdkIterable;
import software.amazon.awssdk.core.pagination.sync.SyncPageFetcher;
import software.amazon.awssdk.core.util.PaginatorUtils;
import software.amazon.awssdk.services.athena.AthenaClient;
import software.amazon.awssdk.services.athena.model.ListTagsForResourceRequest;
import software.amazon.awssdk.services.athena.model.ListTagsForResourceResponse;
import software.amazon.awssdk.services.athena.model.Tag;

/**
 * 
 * Represents the output for the
 * {@link software.amazon.awssdk.services.athena.AthenaClient#listTagsForResourcePaginator(software.amazon.awssdk.services.athena.model.ListTagsForResourceRequest)}
 * operation which is a paginated operation. This class is an iterable of
 * {@link software.amazon.awssdk.services.athena.model.ListTagsForResourceResponse} that can be used to iterate through
 * all the response pages of the operation.
 * 
 * 
 * When the operation is called, an instance of this class is returned. At this point, no service calls are made yet and
 * so there is no guarantee that the request is valid. As you iterate through the iterable, SDK will start lazily
 * loading response pages by making service calls until there are no pages left or your iteration stops. If there are
 * errors in your request, you will see the failures only after you start iterating through the iterable.
 * 
 *
 * 
 * The following are few ways to iterate through the response pages:
 * 
 * 1) Using a Stream
 * 
 *  * {@code
 * software.amazon.awssdk.services.athena.paginators.ListTagsForResourceIterable responses = client.listTagsForResourcePaginator(request);
 * responses.stream().forEach(....);
 * }
 * 
 *
 * 2) Using For loop
 * 
 *  * {
 *     @code
 *     software.amazon.awssdk.services.athena.paginators.ListTagsForResourceIterable responses = client
 *             .listTagsForResourcePaginator(request);
 *     for (software.amazon.awssdk.services.athena.model.ListTagsForResourceResponse response : responses) {
 *         // do something;
 *     }
 * }
 * 
 *
 * 3) Use iterator directly
 * 
 *  * {@code
 * software.amazon.awssdk.services.athena.paginators.ListTagsForResourceIterable responses = client.listTagsForResourcePaginator(request);
 * responses.iterator().forEachRemaining(....);
 * }
 * 
 * 
 * Please notice that the configuration of MaxResults won't limit the number of results you get with the paginator.
 * It only limits the number of results in each page.
 * 
 * 
 * Note: If you prefer to have control on service calls, use the
 * {@link #listTagsForResource(software.amazon.awssdk.services.athena.model.ListTagsForResourceRequest)} operation.
 * 
 */
@Generated("software.amazon.awssdk:codegen")
public class ListTagsForResourceIterable implements SdkIterable {
    private final AthenaClient client;

    private final ListTagsForResourceRequest firstRequest;

    private final SyncPageFetcher nextPageFetcher;

    public ListTagsForResourceIterable(AthenaClient client, ListTagsForResourceRequest firstRequest) {
        this.client = client;
        this.firstRequest = firstRequest;
        this.nextPageFetcher = new ListTagsForResourceResponseFetcher();
    }

    @Override
    public Iterator iterator() {
        return PaginatedResponsesIterator.builder().nextPageFetcher(nextPageFetcher).build();
    }

    /**
     * Returns an iterable to iterate through the paginated {@link ListTagsForResourceResponse#tags()} member. The
     * returned iterable is used to iterate through the results across all response pages and not a single page.
     *
     * This method is useful if you are interested in iterating over the paginated member in the response pages instead
     * of the top level pages. Similar to iteration over pages, this method internally makes service calls to get the
     * next list of results until the iteration stops or there are no more results.
     */
    public final SdkIterable tags() {
        Function> getIterator = response -> {
            if (response != null && response.tags() != null) {
                return response.tags().iterator();
            }
            return Collections.emptyIterator();
        };
        return PaginatedItemsIterable. builder().pagesIterable(this)
                .itemIteratorFunction(getIterator).build();
    }

    private class ListTagsForResourceResponseFetcher implements SyncPageFetcher {
        @Override
        public boolean hasNextPage(ListTagsForResourceResponse previousPage) {
            return PaginatorUtils.isOutputTokenAvailable(previousPage.nextToken());
        }

        @Override
        public ListTagsForResourceResponse nextPage(ListTagsForResourceResponse previousPage) {
            if (previousPage == null) {
                return client.listTagsForResource(firstRequest);
            }
            return client.listTagsForResource(firstRequest.toBuilder().nextToken(previousPage.nextToken()).build());
        }
    }
}