com.bloxbean.cardano.client.backend.gql.BaseGqlService Maven / Gradle / Ivy

Go to download
package com.bloxbean.cardano.client.backend.gql;

import com.apollographql.apollo.ApolloCall;
import com.apollographql.apollo.ApolloClient;
import com.apollographql.apollo.api.Mutation;
import com.apollographql.apollo.api.Query;
import com.apollographql.apollo.api.Response;
import com.apollographql.apollo.exception.ApolloException;
import com.bloxbean.cardano.client.backend.gql.adapter.AddHeadersInterceptor;
import com.bloxbean.cardano.client.backend.gql.adapter.JSONCustomTypeAdapter;
import com.bloxbean.cardano.client.api.model.Result;
import com.bloxbean.cardano.client.util.JsonUtil;
import com.bloxbean.cardano.gql.type.CustomType;
import okhttp3.OkHttpClient;
import org.jetbrains.annotations.NotNull;
import reactor.core.publisher.Mono;

import java.time.Duration;
import java.util.Collections;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;

import static com.bloxbean.cardano.client.backend.gql.util.HttpClientConstants.GQL_CALL_TIMEOUT_SEC;
import static com.bloxbean.cardano.client.backend.gql.util.HttpClientConstants.GQL_READ_TIMEOUT_SEC;

public class BaseGqlService {
    protected String gqlUrl;
    protected ApolloClient apolloClient;

    public BaseGqlService(String gqlUrl) {
        this(gqlUrl, Collections.EMPTY_MAP);
    }

    public BaseGqlService(String gqlUrl, Map headers) {
        this.gqlUrl = gqlUrl;

        OkHttpClient.Builder okHttpClientBuilder = new OkHttpClient().newBuilder();
        okHttpClientBuilder.callTimeout( Duration.ofSeconds(GQL_CALL_TIMEOUT_SEC));
        okHttpClientBuilder.readTimeout(Duration.ofSeconds(GQL_READ_TIMEOUT_SEC));

        if(headers != null && headers.size() > 0) {
            okHttpClientBuilder.addInterceptor(new AddHeadersInterceptor(headers));
        }

        ApolloClient.Builder builder = ApolloClient.builder();
        builder.okHttpClient(okHttpClientBuilder.build());
        apolloClient = builder
                .serverUrl(gqlUrl)
                .addCustomTypeAdapter(CustomType.JSON, new JSONCustomTypeAdapter())
                .build();
    }

    public BaseGqlService(String gqlUrl, OkHttpClient okHttpClient) {
        apolloClient = ApolloClient.builder()
                .serverUrl(gqlUrl)
                .okHttpClient(okHttpClient)
                .addCustomTypeAdapter(CustomType.JSON, new JSONCustomTypeAdapter())
                .build();
    }

    protected  T execute(Query operation) {
        CompletableFuture future = new CompletableFuture<>();
        apolloClient.query(operation).enqueue(new ApolloCall.Callback() {
            @Override
            public void onResponse(@NotNull Response response) {
                if (response.hasErrors()) {
                    String errors = response.getErrors().stream().map(Object::toString).collect(Collectors.joining(", "));
                    future.completeExceptionally(new ApolloException(errors));
                    return;
                }
                future.complete(response.getData());
            }

            @Override
            public void onFailure(@NotNull ApolloException e) {
                future.completeExceptionally(e);
            }
        });

        return Mono.fromFuture(future).block();
    }

    protected  T executeMutatation(Mutation operation) {
        CompletableFuture future = new CompletableFuture<>();
        apolloClient.mutate(operation).enqueue(new ApolloCall.Callback() {
            @Override
            public void onResponse(@NotNull Response response) {
                if (response.hasErrors()) {
                    String errors = response.getErrors().stream().map(Object::toString).collect(Collectors.joining(", "));
                    future.completeExceptionally(new ApolloException(errors));
                    return;
                }
                future.complete(response.getData());
            }

            @Override
            public void onFailure(@NotNull ApolloException e) {
                future.completeExceptionally(e);
            }
        });

        return Mono.fromFuture(future).block();
    }

    public  Result processSuccessResult(T t) {
        return Result.success(JsonUtil.getPrettyJson(t))
                .withValue(t)
                .code(200);
    }
}