Protos.Resource.DiskInfo.Builder diskBuilder = builder.getDiskBuilder(); diskBuilder .getVolumeBuilder()
public static Protos.Resource getUnreservedMountVolume(double diskSize, Optional<String> profile) { Protos.Resource.Builder builder = getUnreservedDisk(diskSize).toBuilder(); builder.getDiskBuilder().setSource(TestConstants.MOUNT_DISK_SOURCE); if (profile.isPresent()) { builder.getDiskBuilder().getSourceBuilder().setProfile(profile.get()); } return builder.build(); }
@SuppressWarnings("deprecation") // for Resource.setRole() private static Protos.Resource toUnreservedResource(String resourceName, Protos.Value value, boolean isMountDisk) { Protos.Resource.Builder resourceBuilder = Protos.Resource.newBuilder() .setRole("*") .setName(resourceName) .setType(value.getType()); switch (value.getType()) { case SCALAR: resourceBuilder.setScalar(value.getScalar()); break; case RANGES: resourceBuilder.setRanges(value.getRanges()); break; case SET: resourceBuilder.setSet(value.getSet()); break; default: throw new IllegalArgumentException("Unsupported value type: " + value); } if (isMountDisk) { resourceBuilder.getDiskBuilder().getSourceBuilder() .setType(Protos.Resource.DiskInfo.Source.Type.MOUNT) .getMountBuilder().setRoot("test-mount-root"); } return resourceBuilder.build(); } }
public static Protos.Resource getReservedMountVolume( double diskSize, Optional<String> profile, String resourceId, String persistenceId) { Protos.Resource.Builder builder = getUnreservedMountVolume(diskSize, profile).toBuilder(); builder.getDiskBuilder().getPersistenceBuilder() .setId(persistenceId) .setPrincipal(TestConstants.PRINCIPAL); builder.getDiskBuilder().getVolumeBuilder() .setContainerPath(TestConstants.CONTAINER_PATH) .setMode(Volume.Mode.RW); return addReservation(builder, resourceId).build(); }