use of com.google.api.services.serviceusage.v1beta1.model.Operation in project terra-workspace-manager by DataBiosphere.
the class CreateAiNotebookInstanceStep method doStep.
@Override
public StepResult doStep(FlightContext flightContext) throws InterruptedException, RetryException {
final GcpCloudContext gcpCloudContext = flightContext.getWorkingMap().get(ControlledResourceKeys.GCP_CLOUD_CONTEXT, GcpCloudContext.class);
String projectId = gcpCloudContext.getGcpProjectId();
InstanceName instanceName = resource.toInstanceName(projectId);
Instance instance = createInstanceModel(flightContext, projectId, petEmail, workspaceUserFacingId, cliConfiguration.getServerName());
AIPlatformNotebooksCow notebooks = crlService.getAIPlatformNotebooksCow();
try {
OperationCow<Operation> creationOperation;
try {
creationOperation = notebooks.operations().operationCow(notebooks.instances().create(instanceName, instance).execute());
} catch (GoogleJsonResponseException e) {
// retry.
if (HttpStatus.CONFLICT.value() == e.getStatusCode()) {
logger.debug("Notebook instance {} already created.", instanceName.formatName());
return StepResult.getStepResultSuccess();
} else if (HttpStatus.BAD_REQUEST.value() == e.getStatusCode()) {
// Don't retry bad requests, which won't change. Instead fail faster.
return new StepResult(StepStatus.STEP_RESULT_FAILURE_FATAL, e);
}
return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
}
GcpUtils.pollUntilSuccess(creationOperation, Duration.ofSeconds(20), Duration.ofMinutes(12));
} catch (IOException e) {
return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
}
return StepResult.getStepResultSuccess();
}
use of com.google.api.services.serviceusage.v1beta1.model.Operation in project terra-workspace-manager by DataBiosphere.
the class CreateAiNotebookInstanceStep method undoStep.
@Override
public StepResult undoStep(FlightContext flightContext) throws InterruptedException {
final GcpCloudContext gcpCloudContext = flightContext.getWorkingMap().get(ControlledResourceKeys.GCP_CLOUD_CONTEXT, GcpCloudContext.class);
InstanceName instanceName = resource.toInstanceName(gcpCloudContext.getGcpProjectId());
AIPlatformNotebooksCow notebooks = crlService.getAIPlatformNotebooksCow();
try {
OperationCow<Operation> deletionOperation;
try {
deletionOperation = notebooks.operations().operationCow(notebooks.instances().delete(instanceName).execute());
} catch (GoogleJsonResponseException e) {
// The AI notebook instance may never have been created or have already been deleted.
if (e.getStatusCode() == HttpStatus.NOT_FOUND.value()) {
logger.debug("No notebook instance {} to delete.", instanceName.formatName());
return StepResult.getStepResultSuccess();
}
return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
}
GcpUtils.pollUntilSuccess(deletionOperation, Duration.ofSeconds(20), Duration.ofMinutes(12));
} catch (IOException | RetryException e) {
return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
}
return StepResult.getStepResultSuccess();
}
use of com.google.api.services.serviceusage.v1beta1.model.Operation in project terra-workspace-manager by DataBiosphere.
the class DeleteAiNotebookInstanceStep method doStep.
@Override
public StepResult doStep(FlightContext flightContext) throws InterruptedException, RetryException {
final GcpCloudContext gcpCloudContext = flightContext.getWorkingMap().get(ControlledResourceKeys.GCP_CLOUD_CONTEXT, GcpCloudContext.class);
InstanceName instanceName = resource.toInstanceName(gcpCloudContext.getGcpProjectId());
AIPlatformNotebooksCow notebooks = crlService.getAIPlatformNotebooksCow();
try {
Optional<Operation> rawOperation = deleteIfFound(instanceName, notebooks);
if (rawOperation.isEmpty()) {
logger.info("Notebook instance {} already deleted", instanceName.formatName());
return StepResult.getStepResultSuccess();
}
GcpUtils.pollUntilSuccess(notebooks.operations().operationCow(rawOperation.get()), Duration.ofSeconds(20), Duration.ofMinutes(10));
} catch (IOException e) {
return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
}
return StepResult.getStepResultSuccess();
}
use of com.google.api.services.serviceusage.v1beta1.model.Operation in project terra-workspace-manager by DataBiosphere.
the class CompleteTransferOperationStep method getTransferOperationResult.
/**
* Poll for completion of the named transfer operation and return the result.
*
* @param transferJobName - name of job owning the transfer operation
* @param operationName - server-generated name of running operation
* @return StepResult indicating success or failure
* @throws IOException
* @throws InterruptedException
*/
private StepResult getTransferOperationResult(String transferJobName, String operationName) throws IOException, InterruptedException {
// Now that we have an operation name, we can poll the operations endpoint for completion
// information.
int attempts = 0;
Operation operation;
do {
operation = storagetransfer.transferOperations().get(operationName).execute();
if (operation == null) {
throw new RuntimeException(String.format("Failed to get transfer operation with name %s", operationName));
} else if (operation.getDone() != null && operation.getDone()) {
break;
} else {
// operation is not started or is in progress
TimeUnit.MILLISECONDS.sleep(OPERATIONS_POLL_INTERVAL.toMillis());
attempts++;
logger.debug("Attempted to get transfer operation {} {} times", operationName, attempts);
}
} while (attempts < MAX_ATTEMPTS);
if (MAX_ATTEMPTS <= attempts) {
final String message = "Timed out waiting for operation result.";
logger.info(message);
return new StepResult(StepStatus.STEP_RESULT_FAILURE_FATAL, new StorageTransferServiceTimeoutException(message));
}
logger.info("Operation {} in transfer job {} has completed", operationName, transferJobName);
// Inspect the completed operation for success
if (operation.getError() != null) {
logger.warn("Error in transfer operation {}: {}", operationName, operation.getError());
final RuntimeException e = new RuntimeException("Failed transfer with error " + operation.getError().toString());
return new StepResult(StepStatus.STEP_RESULT_FAILURE_FATAL, e);
} else {
logger.debug("Completed operation metadata: {}", operation.getMetadata());
return StepResult.getStepResultSuccess();
}
}
use of com.google.api.services.serviceusage.v1beta1.model.Operation in project terra-cloud-resource-lib by DataBiosphere.
the class CloudResourceManagerCowTest method createGetDeleteProject.
@Test
public void createGetDeleteProject() throws Exception {
CloudResourceManagerCow managerCow = defaultManager();
String projectId = ProjectUtils.randomProjectId();
assertThrows(GoogleJsonResponseException.class, () -> managerCow.projects().get(projectId).execute());
Operation operation = managerCow.projects().create(defaultProject(projectId)).execute();
OperationTestUtils.pollAndAssertSuccess(managerCow.operations().operationCow(operation), Duration.ofSeconds(5), Duration.ofSeconds(30));
Project project = managerCow.projects().get(projectId).execute();
assertEquals(projectId, project.getProjectId());
assertEquals("ACTIVE", project.getState());
Operation deleteOperation = managerCow.projects().delete(projectId).execute();
OperationTestUtils.pollAndAssertSuccess(managerCow.operations().operationCow(deleteOperation), Duration.ofSeconds(5), Duration.ofSeconds(30));
// After "deletion," the project still exists for up to 30 days where it can be recovered.
project = managerCow.projects().get(projectId).execute();
assertEquals("DELETE_REQUESTED", project.getState());
}
Aggregations