io.blockfrost.sdk.impl.BaseService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of blockfrost-java Show documentation
Java SDK for the Blockfrost.io API
There is a newer version: 0.1.3
package io.blockfrost.sdk.impl;

import com.fasterxml.jackson.databind.ObjectMapper;
import io.blockfrost.sdk.api.exception.APIException;
import io.blockfrost.sdk.api.model.ResponseError;
import io.blockfrost.sdk.api.util.NetworkHelper;
import io.blockfrost.sdk.impl.util.RateLimitHelper;
import io.github.resilience4j.retrofit.RateLimiterCallAdapter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import retrofit2.Response;
import retrofit2.Retrofit;
import retrofit2.converter.jackson.JacksonConverterFactory;

import java.io.IOException;
import java.util.List;
import java.util.concurrent.CompletableFuture;

public class BaseService {

    private final static Logger LOG = LoggerFactory.getLogger(BaseService.class);
    private final static ObjectMapper OBJECT_MAPPER = new ObjectMapper();

    private String baseUrl;
    private String projectId;
    private int defaultFetchSize = 100;

    public BaseService(String baseUrl, String projectId) {
        this.baseUrl = baseUrl;
        this.projectId = projectId;

        if (LOG.isDebugEnabled()) {
            LOG.debug("Blockfrost URL : " + baseUrl);
        }
    }

    protected Retrofit getRetrofit() {
        return new Retrofit.Builder()
                .addCallAdapterFactory( RateLimiterCallAdapter.of(RateLimitHelper.rateLimiter()))
                .baseUrl(getBaseUrl())
                .addConverterFactory(JacksonConverterFactory.create())
                .client(NetworkHelper.getInstance().getOkHttpClient())
                .build();

    }

    public String getBaseUrl() {
        return baseUrl;
    }

    public String getProjectId() {
        return projectId;
    }


    protected  T processResponse(Response response) throws APIException, IOException {
        if (response.isSuccessful()) {
            return response.body();
        } else {
            String errorMessage = response.errorBody().string();
            int errorCode = response.code();
            try {
                ResponseError responseError = OBJECT_MAPPER.readValue(response.errorBody().bytes(), ResponseError.class);
                errorMessage = responseError.getError() + " : " + responseError.getMessage();
            } catch (Exception exp ){
            }
            throw new APIException(errorMessage, errorCode);
        }
    }

    protected  T getApiClient(Class clazz) {
        return getRetrofit().create(clazz);
    }

    protected  boolean fetchData(List>> completableFutureList, List returnList ) throws Exception {

        boolean noMoreData = false;

        CompletableFuture combinedFuture
                = CompletableFuture.allOf(completableFutureList.toArray(new CompletableFuture[completableFutureList.size()]));

        combinedFuture.get();

        for ( CompletableFuture> completableFuture : completableFutureList ){
           List result = completableFuture.join();
           returnList.addAll(result);

            if ( result.size() < getDefaultFetchSize() ){
                noMoreData = true;
            }
        }

        return noMoreData;

    }

    public int getDefaultFetchSize() {
        return defaultFetchSize;
    }

    public void setDefaultFetchSize(int size) {
        defaultFetchSize = size;
    }

}