twitter4jads.BaseAdsListResponseIterable Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of twitter4j-ads Show documentation
The newest version!
package twitter4jads;

import com.google.gson.Gson;
import org.apache.commons.lang3.StringUtils;
import twitter4jads.internal.http.HttpParameter;
import twitter4jads.internal.http.HttpResponse;
import twitter4jads.internal.models4j.RateLimitStatus;
import twitter4jads.internal.models4j.TwitterException;
import twitter4jads.models.ads.TwitterRuntimeException;
import twitter4jads.util.TwitterAdHttpUtils;
import twitter4jads.util.TwitterAdUtil;

import java.io.IOException;
import java.lang.reflect.Type;
import java.util.*;

/**
 * User: poly
 * Date: 07/02/14
 * Time: 11:56 AM
 */
public class BaseAdsListResponseIterable implements Iterable> {
    private final TwitterAdsClient twitterAdsClient;
    private BaseAdsListResponse data;
    private final String baseUrl;
    private final Type type;
    private String nextCursor;
    private final List baseParameters;
    private boolean isCostBasedRateLimit;

    public BaseAdsListResponseIterable(TwitterAdsClient twitterAdsClient, String baseUrl, List baseParameters, Type type,
                                       HttpResponse response, boolean isCostBasedRateLimit) throws IOException, TwitterException {
        this.isCostBasedRateLimit = isCostBasedRateLimit;
        TwitterAdUtil.ensureNotNull(twitterAdsClient, "Twitter Ads API");
        TwitterAdUtil.ensureNotNull(response, "Twitter Ads Response");
        this.twitterAdsClient = twitterAdsClient;
        this.baseUrl = baseUrl;
        this.baseParameters = baseParameters == null ? Collections.emptyList() : Collections.unmodifiableList(baseParameters);
        Gson gson = new Gson();

        this.data = gson.fromJson(response.asString(), type);
        if (this.data == null) {
            this.data = new BaseAdsListResponse<>();
        }

        this.type = type;
        RateLimitStatus rateLimitStatus;
        if (isCostBasedRateLimit) {
            rateLimitStatus = TwitterAdHttpUtils.createFromResponseHeaderForCostBasedRateLimit(response);
        } else {
            rateLimitStatus = TwitterAdHttpUtils.createFromResponseHeader(response);
        }
        data.setRateLimitStatus(rateLimitStatus);
        if (TwitterAdUtil.isNotEmpty(data.getData()) && data.getNextCursor() != null) {
            nextCursor = data.getNextCursor();
        }
    }

    @Override
    public BaseAdsListResponseIterator iterator() {
        return new BaseAdsListResponseIterator<>(this);
    }

    public static class BaseAdsListResponseIterator implements Iterator> {
        private BaseAdsListResponseIterable baseAdsListResponseIterable;
        private boolean initialResponse = true;

        private BaseAdsListResponseIterator(BaseAdsListResponseIterable baseAdsListResponseIterable) {
            this.baseAdsListResponseIterable = baseAdsListResponseIterable;
        }

        @Override
        public boolean hasNext() {
            return initialResponse || baseAdsListResponseIterable.hasNext();
        }

        @Override
        public BaseAdsListResponse next() {
            if (initialResponse) {
                initialResponse = false;
                return baseAdsListResponseIterable.getData();
            }
            if (!hasNext()) {
                throw new NoSuchElementException("No more data to fetch");
            }
            try {
                baseAdsListResponseIterable = baseAdsListResponseIterable.fetchNextResponse();
            } catch (IOException | TwitterException e) {
                throw new TwitterRuntimeException(e, new TwitterException("Could not fetch more data."));
            }
            return baseAdsListResponseIterable.getData();
        }

        @Override
        public void remove() {
            // NOT SUPPORTED OPERATION
        }

        public String getNextCursor() {
            return baseAdsListResponseIterable.getNextCursor();
        }
    }

    private BaseAdsListResponseIterable fetchNextResponse() throws IOException, TwitterException {
        List parameters = TwitterAdUtil.createMutableList(baseParameters);
        parameters = removeParamIfExist(parameters, "cursor");
        parameters.add(new HttpParameter("cursor", getNextCursor()));
        HttpResponse httpResponse = twitterAdsClient.get(baseUrl, parameters.toArray(new HttpParameter[parameters.size()]));
        return new BaseAdsListResponseIterable<>(twitterAdsClient, baseUrl, baseParameters, type, httpResponse, isCostBasedRateLimit);
    }

    private List removeParamIfExist(List parameters, String param) {
        if (parameters.isEmpty() || param == null) {
            return parameters;
        }

        List toReturn = new ArrayList();
        for (HttpParameter parameter : parameters) {
            if (parameter != null && param.equalsIgnoreCase(parameter.getName())) {
                continue;
            }
            toReturn.add(parameter);
        }
        return toReturn;
    }

    public boolean hasData() {
        return data != null && data.getData() != null && !data.getData().isEmpty();
    }

    private BaseAdsListResponse getData() {
        return data == null ? new BaseAdsListResponse() : data;
    }

    private boolean hasNext() {
        return StringUtils.isNotBlank(getNextCursor());
    }

    public String getNextCursor() {
        return nextCursor;
    }

}