com.spotinst.sdkjava.model.SpotinstMrScalerAwsClient Maven / Gradle / Ivy
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
}