io.elastic.sailor.impl.ApiClientImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sailor-jvm Show documentation
Sailor for JVM
There is a newer version: 4.0.3
package io.elastic.sailor.impl;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.google.inject.name.Named;
import io.elastic.sailor.ApiClient;
import io.elastic.sailor.Constants;
import io.elastic.sailor.Step;
import io.elastic.sailor.impl.HttpUtils.BasicAuthorizationHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.json.Json;
import javax.json.JsonObject;

@Singleton
public class ApiClientImpl implements ApiClient {

    private static final Logger logger = LoggerFactory.getLogger(ApiClientImpl.class.getName());

    private final BasicAuthorizationHandler authorizationHandler;
    private final String apiBaseUri;
    private final int retryCount;
    private boolean putIncomingMessageIntoPassThrough;

    @Inject
    public ApiClientImpl(@Named(Constants.ENV_VAR_API_URI) final String apiUri,
                         @Named(Constants.ENV_VAR_API_USERNAME) final String apiUser,
                         @Named(Constants.ENV_VAR_API_KEY) final String apiKey,
                         @Named(Constants.ENV_VAR_API_REQUEST_RETRY_ATTEMPTS) final int retryCount,
                         @Named(Constants.ENV_VAR_NO_SELF_PASSTRHOUGH) boolean putIncomingMessageIntoPassThrough) {

        this.authorizationHandler = new BasicAuthorizationHandler(apiUser, apiKey);
        this.apiBaseUri = String.format("%s", apiUri);
        this.retryCount = retryCount;
        this.putIncomingMessageIntoPassThrough = putIncomingMessageIntoPassThrough;
    }

    @Override
    public Step retrieveFlowStep(final String taskId, final String stepId) {
        final String uri = String.format("%s/v1/tasks/%s/steps/%s", this.apiBaseUri, taskId, stepId);

        logger.info("Retrieving step data for user {} at: {}", this.authorizationHandler.getUsername(), uri);

        final JsonObject step = HttpUtils.getJson(uri, authorizationHandler, this.retryCount);

        return new Step(step, putIncomingMessageIntoPassThrough);
    }

    @Override
    public JsonObject updateAccount(final String accountId, final JsonObject body) {
        final String uri = String.format("%s/v1/accounts/%s", this.apiBaseUri, accountId);

        logger.info("Updating account for user {} at: {}", this.authorizationHandler.getUsername(), uri);

        return HttpUtils.putJson(uri, body, authorizationHandler, this.retryCount);

    }

    @Override
    public void storeStartupState(final String flowId, final JsonObject body) {
        final String uri = getStartupStateUrl(flowId);

        HttpUtils.postJson(uri, body, authorizationHandler, this.retryCount);
    }

    @Override
    public JsonObject retrieveStartupState(final String flowId) {
        final String uri = getStartupStateUrl(flowId);

        final JsonObject state = HttpUtils.getJson(uri, authorizationHandler, this.retryCount);

        if (state == null) {
            return Json.createObjectBuilder().build();
        }

        return state;
    }

    @Override
    public void deleteStartupState(final String flowId) {
        final String uri = getStartupStateUrl(flowId);

        HttpUtils.delete(uri, authorizationHandler, this.retryCount);
    }

    private String getStartupStateUrl(final String flowId) {
        return String.format("%s/sailor-support/hooks/task/%s/startup/data", this.apiBaseUri, flowId);
    }
}