com.stripe.model.PagingIterator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of stripe-java Show documentation

Stripe Java Bindings

There is a newer version: 28.1.0-beta.3

package com.stripe.model;

import com.stripe.net.*;
import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.NoSuchElementException;

public class PagingIterator extends ApiResource implements Iterator {
  private final String url;

  private final Type pageType;

  private StripeCollectionInterface currentCollection;
  private Iterator currentDataIterator;

  private String lastId;

  PagingIterator(
      final StripeCollectionInterface stripeCollection,
      StripeResponseGetter responseGetter,
      Type pageType) {
    this.url = stripeCollection.getUrl();

    this.pageType = pageType;

    this.currentCollection = stripeCollection;
    this.currentDataIterator = stripeCollection.getData().iterator();
    setResponseGetter(responseGetter);
  }

  @Override
  public boolean hasNext() {
    return currentDataIterator.hasNext() || currentCollection.getHasMore();
  }

  @Override
  public T next() {
    // if we've run out of data on the current page, try to fetch another
    // one
    if (!currentDataIterator.hasNext() && currentCollection.getHasMore()) {
      try {
        Map params = new HashMap<>();

        // copy all the parameters from the initial request
        Map initialParams = currentCollection.getRequestParams();
        if (initialParams != null) {
          params.putAll(initialParams);
        }

        // then put our new page start in
        params.put("starting_after", lastId);

        this.currentCollection = list(params, currentCollection.getRequestOptions());

        this.currentDataIterator = currentCollection.getData().iterator();
      } catch (final Exception e) {
        throw new RuntimeException("Unable to lazy-load stripe objects", e);
      }
    }

    if (currentDataIterator.hasNext()) {
      final T next = currentDataIterator.next();
      this.lastId = next.getId();
      return next;
    }

    throw new NoSuchElementException();
  }

  @Override
  public void remove() {
    throw new UnsupportedOperationException();
  }

  private StripeCollectionInterface list(
      final Map params, final RequestOptions options) throws Exception {
    ApiRequest request = new ApiRequest(BaseAddress.API, RequestMethod.GET, url, params, options);
    return getResponseGetter().request(request, pageType);
  }
}