Examples with AZURE_STORAGE_ACCOUNT_URI com.vmware.photon.controller.model.adapters.azure.constants.AzureConstants.AZURE_STORAGE_ACCOUNT_URI used on opensource projects

Example 1 with AZURE_STORAGE_ACCOUNT_URI

use of com.vmware.photon.controller.model.adapters.azure.constants.AzureConstants.AZURE_STORAGE_ACCOUNT_URI in project photon-model by vmware.

the class AzureComputeEnumerationAdapterService method queryForDiagnosticStorageDescriptions.

/**
 * Get all storage descriptions responsible for diagnostics of given VMs.
 */
private void queryForDiagnosticStorageDescriptions(EnumerationContext ctx, ComputeEnumerationSubStages next) {
    List<String> diagnosticStorageAccountUris = new ArrayList<>();
    String storageAccountProperty = QuerySpecification.buildCompositeFieldName(StorageDescription.FIELD_NAME_CUSTOM_PROPERTIES, AZURE_STORAGE_ACCOUNT_URI);
    ctx.virtualMachines.keySet().stream().filter(instanceId -> ctx.virtualMachines.get(instanceId) != null && ctx.virtualMachines.get(instanceId).diagnosticsProfile() != null && ctx.virtualMachines.get(instanceId).diagnosticsProfile().bootDiagnostics() != null && ctx.virtualMachines.get(instanceId).diagnosticsProfile().bootDiagnostics().storageUri() != null).forEach(instanceId -> {
        diagnosticStorageAccountUris.add(ctx.virtualMachines.get(instanceId).diagnosticsProfile().bootDiagnostics().storageUri());
    });
    if (diagnosticStorageAccountUris.isEmpty()) {
        ctx.subStage = next;
        handleSubStage(ctx);
        return;
    }
    Query.Builder qBuilder = Query.Builder.create().addKindFieldClause(StorageDescription.class).addInClause(storageAccountProperty, diagnosticStorageAccountUris);
    QueryTop<StorageDescription> queryDiskStates = new QueryTop<>(getHost(), qBuilder.build(), StorageDescription.class, ctx.parentCompute.tenantLinks, // endpointLink
    null, ctx.parentCompute.documentSelfLink).setClusterType(ServiceTypeCluster.INVENTORY_SERVICE);
    queryDiskStates.queryDocuments(storageDesc -> {
        ctx.storageDescriptions.put(storageDesc.customProperties.get(AZURE_STORAGE_ACCOUNT_URI), storageDesc);
    }).thenRun(() -> logFine(() -> String.format("Found %d matching diagnostics storage accounts", ctx.storageDescriptions.size()))).whenComplete(thenHandleSubStage(ctx, next));
}