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

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: 26.8.0-beta.1

package com.stripe.model;

import com.stripe.Stripe;
import com.stripe.net.ApiResource;
import com.stripe.net.RequestOptions;
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;

  @SuppressWarnings("rawtypes")
  private final Class collectionType;

  private StripeCollectionInterface currentCollection;
  private Iterator currentDataIterator;

  private String lastId;

  PagingIterator(final StripeCollectionInterface stripeCollection) {
    this.url = Stripe.getApiBase() + stripeCollection.getUrl();

    this.collectionType = stripeCollection.getClass();

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

  @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();
  }

  @SuppressWarnings("unchecked")
  private StripeCollectionInterface list(
      final Map params, final RequestOptions options) throws Exception {
    return ApiResource.requestCollection(url, params, collectionType, options);
  }
}