com.spotinst.sdkjava.model.SpotinstMrScalerAwsClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spotinst-sdk-java Show documentation
A Spotinst Java SDK
There is a newer version: 1.0.121
package com.spotinst.sdkjava.model;

import com.spotinst.sdkjava.exception.HttpError;
import com.spotinst.sdkjava.exception.SpotinstHttpException;
import com.spotinst.sdkjava.model.api.mrScaler.aws.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;


public class SpotinstMrScalerAwsClient {
    private static final Logger LOGGER = LoggerFactory.getLogger(SpotinstSpectrumClient.class);

    //region Members
    private String                           authToken;
    private String                           account;
    private ISpotinstMrScalerAwsRepo         spotinstMrScalerRepo;
    private ISpotinstMrScalerOperatorAwsRepo spotinstMrScalerOperatorAwsRepo;

    public ISpotinstMrScalerAwsRepo getSpotinstMrScalerRepo() {
        return this.spotinstMrScalerRepo;
    }

    public void setSpotinstMrScalerRepo() {
        this.spotinstMrScalerRepo = SpotinstRepoManager.getInstance().getSpotinstMrScalerAwsRepo();
    }

    public ISpotinstMrScalerOperatorAwsRepo getSpotinstMrScalerOperatorAwsRepo() {
        return spotinstMrScalerOperatorAwsRepo;
    }

    public void setSpotinstMrScalerOperatorAwsRepo() {
        this.spotinstMrScalerOperatorAwsRepo = SpotinstRepoManager.getInstance().getSpotinstMrScalerOperatorAwsRepo();
    }

    /**
     * This Object is used to store the users account and token information and then make requests to the
     * AWS MrScaler endpoints.
     *
     * @param authToken User Spotinst API token
     * @param account   User Spotinst account ID
     */
    public SpotinstMrScalerAwsClient(String authToken, String account) {
        this.authToken = authToken;
        this.account = account;

        setSpotinstMrScalerRepo();
        setSpotinstMrScalerOperatorAwsRepo();
    }


    // endregion

    // region Methods

    /**
     * This method is used to create an AWS MrScaler Cluster
     *
     * @param mrScalerCreationRequest ApiMrScalerAwsCreationRequest Object that contains the requested create params
     * @return ApiMrScalerAws
     */
    public ApiMrScalerAws createMrScaler(ApiMrScalerAwsCreationRequest mrScalerCreationRequest) {
        ApiMrScalerAws retVal;

        ApiMrScalerAws mrScalerToCreate = mrScalerCreationRequest.getMrScaler();

        RepoGenericResponse creationResponse =
                getSpotinstMrScalerRepo().create(mrScalerToCreate, authToken, account);
        if (creationResponse.isRequestSucceed()) {
            retVal = creationResponse.getValue();
        }
        else {
            List httpExceptions = creationResponse.getHttpExceptions();
            HttpError       httpException  = httpExceptions.get(0);
            LOGGER.error(String.format("Error encountered while attempting to create MrScaler. Code: %s. Message: %s.",
                                       httpException.getCode(), httpException.getMessage()));
            throw new SpotinstHttpException(httpException.getMessage());
        }
        return retVal;
    }

    /**
     * This method is used to update an exiting AWS MrScaler Cluster
     *
     * @param mrScalerUpdateRequest ApiMrScalerAwsUpdateRequest Object that contains the requested update params
     * @param clusterId             String for the cluster Id to update
     * @return Boolean
     */
    public Boolean updateMrScaler(ApiMrScalerAwsUpdateRequest mrScalerUpdateRequest, String clusterId) {
        Boolean retVal;

        ApiMrScalerAws mrScalerToUpdate = mrScalerUpdateRequest.getMrScaler();

        RepoGenericResponse updateResponse =
                getSpotinstMrScalerRepo().update(clusterId, mrScalerToUpdate, authToken, account);
        if (updateResponse.isRequestSucceed()) {
            retVal = updateResponse.getValue();
        }
        else {
            List httpExceptions = updateResponse.getHttpExceptions();
            HttpError       httpException  = httpExceptions.get(0);
            LOGGER.error(String.format("Error encountered while attempting to create mrScaler. Code: %s. Message: %s.",
                                       httpException.getCode(), httpException.getMessage()));
            throw new SpotinstHttpException(httpException.getMessage());
        }
        return retVal;
    }

    /**
     * This method is used to delete an exiting AWS MrScaler Cluster
     *
     * @param mrScalerDeletionRequest ApiMrScalerAwsDeleteRequest Object that contains the requested delete mrScaler ID
     * @return Boolean
     */
    public Boolean deleteMrScaler(ApiMrScalerAwsDeleteRequest mrScalerDeletionRequest) {
        Boolean retVal;
        String  mrScalerToDeleteId = mrScalerDeletionRequest.getMrScalerId();
        RepoGenericResponse mrScalerDeletionResponse =
                getSpotinstMrScalerRepo().delete(mrScalerToDeleteId, authToken, account);
        if (mrScalerDeletionResponse.isRequestSucceed()) {
            retVal = mrScalerDeletionResponse.getValue();
        }
        else {
            List httpExceptions = mrScalerDeletionResponse.getHttpExceptions();
            HttpError       httpException  = httpExceptions.get(0);
            LOGGER.error(String.format("Error encountered while attempting to delete mrScaler. Code: %s. Message: %s.",
                                       httpException.getCode(), httpException.getMessage()));
            throw new SpotinstHttpException(httpException.getMessage());
        }

        return retVal;
    }

    /**
     * This method is used to get an exiting AWS MrScaler Cluster
     *
     * @param mrScalerAwsGetRequest ApiMrScalerAwsGetRequest Object that contains the requested get mrScaler ID
     * @return ApiMrScalerAws
     */
    public ApiMrScalerAws getMrScaler(ApiMrScalerAwsGetRequest mrScalerAwsGetRequest) {
        ApiMrScalerAws retVal;
        String         clusterToGet = mrScalerAwsGetRequest.getMrScalerId();
        RepoGenericResponse mrScalerGetRes =
                getSpotinstMrScalerRepo().get(clusterToGet, authToken, account);
        if (mrScalerGetRes.isRequestSucceed()) {
            retVal = mrScalerGetRes.getValue();
        }
        else {
            List httpExceptions = mrScalerGetRes.getHttpExceptions();
            HttpError       httpException  = httpExceptions.get(0);
            LOGGER.error(String.format("Error encountered while attempting to get mrScaler. Code: %s. Message: %s.",
                                       httpException.getCode(), httpException.getMessage()));
            throw new SpotinstHttpException(httpException.getMessage());
        }

        return retVal;
    }

    /**
     * This method is used to get all AWS MrScaler Clusters
     *
     * @return List
     */
    public List getAllMrScalers() {
        List retVal;
        RepoGenericResponse> getAllRes =
                getSpotinstMrScalerRepo().getAll(null, authToken, account);

        if (getAllRes.isRequestSucceed()) {
            retVal = getAllRes.getValue();
        }
        else {
            List httpExceptions = getAllRes.getHttpExceptions();
            HttpError       httpException  = httpExceptions.get(0);
            LOGGER.error(
                    String.format("Error encountered while attempting to get all mrScalers. Code: %s. Message: %s.",
                                  httpException.getCode(), httpException.getMessage()));
            throw new SpotinstHttpException(httpException.getMessage());
        }

        return retVal;
    }

    /**
     * This method is used to create an AWS MrScaler Operator.
     * if MrScaler does not exist - it will create a new one
     * otherwise - it will return the MrScaler ID which is connected to this operator name and its cluster status.
     *
     * @param mrScalerOperatorRequest ApiMrScalerOpeartorAwsRequest Object that contains the requested operator params
     * @return ApiMrScalerOperatorAwsResponse
     */
    public ApiMrScalerOperatorAwsResponse createMrScalerOperator(
            ApiMrScalerOperatorAwsRequest mrScalerOperatorRequest) {
        ApiMrScalerOperatorAwsResponse retVal;
        ApiMrScalerOperatorAws apiMrScalerOperatorAws = mrScalerOperatorRequest.getMrScalerOperator();

        RepoGenericResponse operatorResponse =
                getSpotinstMrScalerOperatorAwsRepo().create(apiMrScalerOperatorAws, authToken, account);
        if (operatorResponse.isRequestSucceed()) {
            retVal = operatorResponse.getValue();
        }
        else {
            List httpExceptions = operatorResponse.getHttpExceptions();
            HttpError       httpException  = httpExceptions.get(0);
            LOGGER.error(String.format(
                    "Error encountered while attempting to create MrScaler operator. Code: %s. Message: %s.",
                    httpException.getCode(), httpException.getMessage()));
            throw new SpotinstHttpException(httpException.getMessage());
        }
        return retVal;
    }

    // endregion

}