use of com.vmware.photon.controller.model.constants.PhotonModelConstants.DISK_LINK in project photon-model by vmware.
the class VSphereVMDiskContext method populateVMDiskContextThen.
/**
* Populates the given initial context and invoke the onSuccess handler when built. At every
* step, if failure occurs the VSphereVMDiskContext's errorHandler is invoked to cleanup.
*/
protected static void populateVMDiskContextThen(Service service, VSphereVMDiskContext ctx, Consumer<VSphereVMDiskContext> onSuccess) {
if (ctx.computeDesc == null) {
URI computeUri = createInventoryUri(service.getHost(), UriUtils.extendUriWithQuery(ctx.request.resourceReference, UriUtils.URI_PARAM_ODATA_EXPAND, Boolean.TRUE.toString()));
AdapterUtils.getServiceState(service, computeUri, op -> {
ctx.computeDesc = op.getBody(ComputeStateWithDescription.class);
if (CustomProperties.of(ctx.computeDesc).getString(CustomProperties.MOREF, null) == null) {
ctx.fail(new IllegalStateException(String.format("VM Moref is not defined in resource %s", ctx.computeDesc.documentSelfLink)));
return;
}
populateVMDiskContextThen(service, ctx, onSuccess);
}, ctx.errorHandler);
return;
}
if (ctx.diskState == null) {
URI diskUri = createInventoryUri(service.getHost(), DiskService.DiskStateExpanded.buildUri(UriUtils.buildUri(service.getHost(), ctx.request.payload.get(DISK_LINK))));
AdapterUtils.getServiceState(service, diskUri, op -> {
ctx.diskState = op.getBody(DiskService.DiskStateExpanded.class);
// the VM. So CD_ROM can be in any status.
if (!ctx.request.isMockRequest) {
if (ctx.request.operation.equals(ResourceOperation.ATTACH_DISK.operation)) {
EnumSet<DiskService.DiskType> notSupportedTypes = EnumSet.of(DiskService.DiskType.SSD, DiskService.DiskType.NETWORK);
if (notSupportedTypes.contains(ctx.diskState.type)) {
ctx.fail(new IllegalStateException(String.format("Not supported disk type %s.", ctx.diskState.type)));
return;
}
if (ctx.diskState.type == DiskService.DiskType.HDD) {
if (ctx.diskState.status != DiskService.DiskStatus.AVAILABLE) {
ctx.fail(new IllegalStateException(String.format("Disk %s is not in AVAILABLE status to attach to VM.", ctx.diskState.documentSelfLink)));
return;
} else if (CustomProperties.of(ctx.diskState).getString(DISK_FULL_PATH, null) == null || CustomProperties.of(ctx.diskState).getString(DISK_DATASTORE_NAME, null) == null) {
ctx.fail(new IllegalStateException(String.format("Disk %s is missing path details to attach to VM.", ctx.diskState.documentSelfLink)));
return;
}
}
} else {
// Allowing only HDD based disk to be detached
if (ctx.diskState.type != DiskService.DiskType.HDD) {
ctx.fail(new IllegalStateException(String.format("Not supported disk type %s for detach.", ctx.diskState.type)));
return;
}
if (ctx.diskState.status != DiskService.DiskStatus.ATTACHED) {
ctx.fail(new IllegalStateException(String.format("Disk %s is not in ATTACHED status to detach from VM.", ctx.diskState.documentSelfLink)));
return;
}
}
}
// fetch the content from the content service
if (ctx.diskState.type == DiskService.DiskType.CDROM) {
String contentUriStr = CustomProperties.of(ctx.diskState).getString(DISK_CONTENT_LINK, null);
if (contentUriStr != null) {
ctx.contentLink = contentUriStr;
URI contentUri = PhotonModelUriUtils.createInventoryUri(service.getHost(), UriUtils.buildUri(service.getHost(), contentUriStr));
AdapterUtils.getServiceState(service, contentUri, MEDIA_TYPE_APPLICATION_OCTET_STREAM, op2 -> {
ctx.contentToUpload = op2.getBody(byte[].class);
populateVMDiskContextThen(service, ctx, onSuccess);
}, ctx.errorHandler);
} else {
populateVMDiskContextThen(service, ctx, onSuccess);
}
} else {
populateVMDiskContextThen(service, ctx, onSuccess);
}
}, ctx.errorHandler);
return;
}
// If it is CD-ROM attach then collect all the disk links objects if insertCDRom is true
if (ctx.computeDiskStates == null) {
Boolean insertCdRom = CustomProperties.of(ctx.diskState).getBoolean(INSERT_CDROM, false);
if (ctx.diskState.type == DiskService.DiskType.CDROM && insertCdRom && ctx.computeDesc.diskLinks != null && !ctx.computeDesc.diskLinks.isEmpty()) {
ctx.computeDiskStates = new ArrayList<>(ctx.computeDesc.diskLinks.size());
// collect disks in parallel
Stream<Operation> opsGetDisk = ctx.computeDesc.diskLinks.stream().map(link -> {
URI diskStateUri = createInventoryUri(service.getHost(), link);
return Operation.createGet(createInventoryUri(service.getHost(), DiskService.DiskStateExpanded.buildUri(diskStateUri)));
});
OperationJoin join = OperationJoin.create(opsGetDisk).setCompletion((os, errors) -> {
if (errors != null && !errors.isEmpty()) {
// fail on first error
ctx.errorHandler.accept(new IllegalStateException("Cannot get disk state", errors.values().iterator().next()));
return;
}
os.values().forEach(op -> ctx.computeDiskStates.add(op.getBody(DiskService.DiskStateExpanded.class)));
populateVMDiskContextThen(service, ctx, onSuccess);
});
join.sendWith(service);
} else {
ctx.computeDiskStates = Collections.emptyList();
populateVMDiskContextThen(service, ctx, onSuccess);
}
return;
}
if (ctx.parentComputeDesc == null && ctx.computeDesc.parentLink != null) {
URI computeUri = createInventoryUri(service.getHost(), UriUtils.extendUriWithQuery(UriUtils.buildUri(service.getHost(), ctx.computeDesc.parentLink), UriUtils.URI_PARAM_ODATA_EXPAND, Boolean.TRUE.toString()));
AdapterUtils.getServiceState(service, computeUri, op -> {
ctx.parentComputeDesc = op.getBody(ComputeStateWithDescription.class);
populateVMDiskContextThen(service, ctx, onSuccess);
}, ctx.errorHandler);
return;
}
if (ctx.vSphereCredentials == null) {
if (IAAS_API_ENABLED) {
if (ctx.operation == null) {
ctx.fail(new IllegalArgumentException("Caller operation cannot be empty"));
return;
}
SessionUtil.retrieveExternalToken(service, ctx.operation.getAuthorizationContext()).whenComplete((authCredentialsServiceState, throwable) -> {
if (throwable != null) {
ctx.errorHandler.accept(throwable);
return;
}
ctx.vSphereCredentials = authCredentialsServiceState;
populateVMDiskContextThen(service, ctx, onSuccess);
});
} else {
if (ctx.parentComputeDesc.description.authCredentialsLink == null) {
ctx.fail(new IllegalStateException(String.format("authCredentialsLink is not defined in resource %s", ctx.parentComputeDesc.description.documentSelfLink)));
return;
}
URI credUri = createInventoryUri(service.getHost(), ctx.parentComputeDesc.description.authCredentialsLink);
AdapterUtils.getServiceState(service, credUri, op -> {
ctx.vSphereCredentials = op.getBody(AuthCredentialsServiceState.class);
populateVMDiskContextThen(service, ctx, onSuccess);
}, ctx.errorHandler);
}
return;
}
if (ctx.datacenterMoRef == null) {
try {
String regionId = ctx.diskState.regionId;
if (regionId == null || regionId.isEmpty()) {
if (ctx.computeDesc.regionId != null) {
regionId = ctx.computeDesc.regionId;
} else if (ctx.parentComputeDesc.regionId != null) {
regionId = ctx.parentComputeDesc.regionId;
}
}
ctx.datacenterMoRef = VimUtils.convertStringToMoRef(regionId);
} catch (IllegalArgumentException ex) {
ctx.fail(ex);
return;
}
populateVMDiskContextThen(service, ctx, onSuccess);
return;
}
if (ctx.computePlacementHost == null) {
String placementLink = CustomProperties.of(ctx.computeDesc).getString(ComputeProperties.COMPUTE_HOST_LINK_PROP_NAME);
// compute host link will be not null here.
URI expandedPlacementUri = UriUtils.extendUriWithQuery(PhotonModelUriUtils.createInventoryUri(service.getHost(), placementLink), UriUtils.URI_PARAM_ODATA_EXPAND, Boolean.TRUE.toString());
expandedPlacementUri = PhotonModelUriUtils.createInventoryUri(service.getHost(), expandedPlacementUri);
AdapterUtils.getServiceState(service, expandedPlacementUri, op -> {
ctx.computePlacementHost = op.getBody(ComputeStateWithDescription.class);
if (ctx.computePlacementHost.groupLinks != null) {
ctx.computeGroupLinks = ctx.computePlacementHost.groupLinks.stream().filter(link -> link.contains(PREFIX_DATASTORE)).collect(Collectors.toSet());
}
populateVMDiskContextThen(service, ctx, onSuccess);
}, ctx.errorHandler);
return;
}
// populate datastore name
if (ctx.datastoreName == null) {
if (ctx.diskState.customProperties != null && ctx.diskState.customProperties.get(DISK_DATASTORE_NAME) != null) {
ctx.datastoreName = ctx.diskState.customProperties.get(DISK_DATASTORE_NAME);
populateVMDiskContextThen(service, ctx, onSuccess);
} else if (ctx.diskState.storageDescription != null) {
ctx.datastoreName = ctx.diskState.storageDescription.id;
populateVMDiskContextThen(service, ctx, onSuccess);
} else if (ctx.diskState.resourceGroupStates != null && !ctx.diskState.resourceGroupStates.isEmpty()) {
// There will always be only one resource group state existing for a disk
ResourceGroupService.ResourceGroupState resource = ctx.diskState.resourceGroupStates.iterator().next();
ClientUtils.getDatastoresForProfile(service, resource.documentSelfLink, ctx.diskState.endpointLink, ctx.diskState.tenantLinks, ctx.errorHandler, (result) -> {
if (result.documents != null && result.documents.size() > 0) {
// pick the first datastore and proceed.
ctx.datastoreName = Utils.fromJson(result.documents.values().iterator().next(), StorageDescriptionService.StorageDescription.class).id;
} else {
// Since no result found default to the available datastore.
ctx.datastoreName = "";
}
populateVMDiskContextThen(service, ctx, onSuccess);
});
} else if (ctx.computeGroupLinks != null) {
// try to get the datastore form the placement link of compute
String datastoreLink = ctx.computeGroupLinks.iterator().next();
URI dsUri = PhotonModelUriUtils.createInventoryUri(service.getHost(), UriUtils.buildUri(service.getHost(), datastoreLink));
AdapterUtils.getServiceState(service, dsUri, op -> {
ResourceGroupService.ResourceGroupState rgState = op.getBody(ResourceGroupService.ResourceGroupState.class);
ctx.datastoreName = rgState.id;
populateVMDiskContextThen(service, ctx, onSuccess);
}, ctx.errorHandler);
} else {
ctx.datastoreName = "";
populateVMDiskContextThen(service, ctx, onSuccess);
}
return;
}
// context populated, invoke handler
onSuccess.accept(ctx);
}
Aggregations