-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
separated FireboltEngineServer to 2 implementations
- Loading branch information
1 parent
cc9a376
commit 12f2f66
Showing
9 changed files
with
239 additions
and
188 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
109 changes: 109 additions & 0 deletions
109
src/main/java/com/firebolt/jdbc/service/FireboltEngineApiService.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
| package com.firebolt.jdbc.service; | ||
|
|
||
| import com.firebolt.jdbc.client.account.FireboltAccountClient; | ||
| import com.firebolt.jdbc.client.account.response.FireboltAccountResponse; | ||
| import com.firebolt.jdbc.client.account.response.FireboltDefaultDatabaseEngineResponse; | ||
| import com.firebolt.jdbc.client.account.response.FireboltEngineIdResponse; | ||
| import com.firebolt.jdbc.client.account.response.FireboltEngineResponse; | ||
| import com.firebolt.jdbc.connection.Engine; | ||
| import com.firebolt.jdbc.connection.settings.FireboltProperties; | ||
| import com.firebolt.jdbc.exception.FireboltException; | ||
| import lombok.CustomLog; | ||
| import lombok.RequiredArgsConstructor; | ||
| import org.apache.commons.lang3.StringUtils; | ||
|
|
||
| import java.io.IOException; | ||
| import java.sql.SQLException; | ||
| import java.util.Optional; | ||
| import java.util.Set; | ||
|
|
||
| @RequiredArgsConstructor | ||
| @CustomLog | ||
| public class FireboltEngineApiService implements FireboltEngineService { | ||
| private static final Set<String> ENGINE_NOT_READY_STATUSES = Set.of( | ||
| "ENGINE_STATUS_PROVISIONING_STARTED", "ENGINE_STATUS_PROVISIONING_PENDING", | ||
| "ENGINE_STATUS_PROVISIONING_FINISHED", "ENGINE_STATUS_RUNNING_REVISION_STARTING"); | ||
| private static final String ERROR_NO_ENGINE_ATTACHED = "There is no Firebolt engine running on %s attached to the database %s. To connect first make sure there is a running engine and then try again."; | ||
| private static final String ERROR_NO_ENGINE_WITH_NAME = "There is no Firebolt engine running on %s with the name %s. To connect first make sure there is a running engine and then try again."; | ||
|
|
||
|
|
||
| private final FireboltAccountClient fireboltAccountClient; | ||
|
|
||
| @Override | ||
| public Engine getEngine(FireboltProperties properties) throws SQLException { | ||
| return getEngine(properties.getHttpConnectionUrl(), properties, properties.getAccessToken()); | ||
| } | ||
|
|
||
| /** | ||
| * Returns the engine | ||
| * | ||
| * @param connectionUrl the connection url | ||
| * @param loginProperties properties to login | ||
| * @param accessToken the access token | ||
| * @return the engine | ||
| */ | ||
| private Engine getEngine(String connectionUrl, FireboltProperties loginProperties, String accessToken) | ||
| throws FireboltException { | ||
| String accountId = null; | ||
| Engine engine; | ||
| try { | ||
| if (StringUtils.isNotEmpty(loginProperties.getAccount())) { | ||
| accountId = getAccountId(connectionUrl, loginProperties.getAccount(), accessToken).orElse(null); | ||
| } | ||
| if (StringUtils.isEmpty(loginProperties.getEngine())) { | ||
| engine = getDefaultEngine(connectionUrl, accountId, loginProperties.getDatabase(), accessToken); | ||
| } else { | ||
| engine = getEngineWithName(connectionUrl, accountId, loginProperties.getEngine(), accessToken); | ||
| } | ||
| } catch (FireboltException e) { | ||
| throw e; | ||
| } catch (Exception e) { | ||
| throw new FireboltException("Failed to get engine", e); | ||
| } | ||
| validateEngineIsNotStarting(engine); | ||
| return engine; | ||
| } | ||
|
|
||
| private Engine getEngineWithName(String connectionUrl, String accountId, String engineName, String accessToken) | ||
| throws FireboltException, IOException { | ||
| FireboltEngineIdResponse response = fireboltAccountClient.getEngineId(connectionUrl, accountId, engineName, | ||
| accessToken); | ||
| String engineID = Optional.ofNullable(response).map(FireboltEngineIdResponse::getEngine) | ||
| .map(FireboltEngineIdResponse.Engine::getEngineId).orElseThrow(() -> new FireboltException( | ||
| "Failed to extract engine id field from the server response: the response from the server is invalid.")); | ||
| FireboltEngineResponse fireboltEngineResponse = fireboltAccountClient.getEngine(connectionUrl, accountId, | ||
| engineName, engineID, accessToken); | ||
|
|
||
| return Optional.ofNullable(fireboltEngineResponse).map(FireboltEngineResponse::getEngine) | ||
| .filter(e -> StringUtils.isNotEmpty(e.getEndpoint())) | ||
| .map(e -> new Engine(e.getEndpoint(), e.getCurrentStatus(), engineName, null, engineID)) | ||
| .orElseThrow(() -> new FireboltException( | ||
| String.format(ERROR_NO_ENGINE_WITH_NAME, connectionUrl, engineName))); | ||
| } | ||
|
|
||
| private Engine getDefaultEngine(String connectionUrl, String accountId, String database, String accessToken) | ||
| throws FireboltException, IOException { | ||
| FireboltDefaultDatabaseEngineResponse defaultEngine = fireboltAccountClient | ||
| .getDefaultEngineByDatabaseName(connectionUrl, accountId, database, accessToken); | ||
|
|
||
| return Optional.ofNullable(defaultEngine).map(FireboltDefaultDatabaseEngineResponse::getEngineUrl) | ||
| .map(url -> new Engine(url, "running", null, database, null)).orElseThrow( | ||
| () -> new FireboltException(String.format(ERROR_NO_ENGINE_ATTACHED, connectionUrl, database))); | ||
| } | ||
|
|
||
| private Optional<String> getAccountId(String connectionUrl, String account, String accessToken) | ||
| throws FireboltException, IOException { | ||
| FireboltAccountResponse fireboltAccountResponse = fireboltAccountClient.getAccount(connectionUrl, account, | ||
| accessToken); | ||
| return Optional.ofNullable(fireboltAccountResponse).map(FireboltAccountResponse::getAccountId); | ||
| } | ||
|
|
||
| private void validateEngineIsNotStarting(Engine engine) throws FireboltException { | ||
| if (StringUtils.isNotEmpty(engine.getId()) && StringUtils.isNotEmpty(engine.getStatus()) | ||
| && ENGINE_NOT_READY_STATUSES.contains(engine.getStatus())) { | ||
| throw new FireboltException(String.format( | ||
| "The engine %s is currently starting. Please wait until the engine is on and then execute the query again.", engine.getName())); | ||
| } | ||
| } | ||
|
|
||
| } |
Oops, something went wrong.