in aws-sagemaker-image/src/main/java/software/amazon/sagemaker/image/CreateHandler.java [114:145]
private boolean stabilizedOnCreate(
final CreateImageRequest createImageRequest,
final CreateImageResponse createImageResponse,
final ProxyClient<SageMakerClient> proxyClient,
final ResourceModel model,
final CallbackContext callbackContext) {
if (model.getImageArn() == null) {
model.setImageArn(createImageResponse.imageArn());
}
final ImageStatus imageStatus= proxyClient.injectCredentialsAndInvokeV2(
Translator.translateToReadRequest(model), proxyClient.client()::describeImage).imageStatus();
switch (imageStatus) {
case CREATE_FAILED:
throw new CfnGeneralServiceException(String.format("%s [%s] failed to create.", ResourceModel.TYPE_NAME,
model.getImageArn()));
case CREATED:
logger.log(String.format("%s [%s] has been stabilized with status %s.", ResourceModel.TYPE_NAME,
model.getImageArn(), imageStatus));
return true;
case CREATING:
logger.log(String.format("%s [%s] is stabilizing %s.", ResourceModel.TYPE_NAME,
model.getImageArn(), imageStatus));
return false;
default:
throw new CfnGeneralServiceException(
String.format("Stabilizing of %s failed with an unexpected status %s",
model.getImageArn(), imageStatus));
}
}