Resource.Builder cpusResource = Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.CPUS).setScalar(Scalar.newBuilder().setValue(cpus)); Resource.Builder memoryResources = Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.MEMORY).setScalar(Scalar.newBuilder().setValue(memory)); Resource.Builder diskResources = Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.DISK).setScalar(Scalar.newBuilder().setValue(disk)); if(role.isPresent()) { cpusResource = cpusResource.setRole(role.get()); memoryResources = memoryResources.setRole(role.get()); diskResources = diskResources.setRole(role.get());
public static List<Resource> combineResources(List<List<Resource>> resourcesList) { List<Resource> resources = new ArrayList<>(); for (List<Resource> resourcesToAdd : resourcesList) { for (Resource resource : resourcesToAdd) { Optional<Resource> matched = getMatchingResource(resource, resources); if (!matched.isPresent()) { resources.add(resource); } else { int index = resources.indexOf(matched.get()); Resource.Builder resourceBuilder = resource.toBuilder().clone(); if (resource.hasScalar()) { resourceBuilder.setScalar(resource.toBuilder().getScalarBuilder().setValue(resource.getScalar().getValue() + matched.get().getScalar().getValue()).build()); resources.set(index, resourceBuilder.build()); } else if (resource.hasRanges()) { Ranges.Builder newRanges = Ranges.newBuilder(); resource.getRanges().getRangeList().forEach(newRanges::addRange); matched.get().getRanges().getRangeList().forEach(newRanges::addRange); resourceBuilder.setRanges(newRanges); resources.set(index, resourceBuilder.build()); } else { throw new IllegalStateException(String.format("Can't subtract non-scalar or range resources %s", formatForLogging(resource))); } } } } return resources; }
.setFrameworkId(org.apache.mesos.v1.Protos.FrameworkID.newBuilder().setValue(frameworkId)) .addResources(org.apache.mesos.v1.Protos.Resource.newBuilder() .setName("cpus") .setRole("*") .setType(org.apache.mesos.v1.Protos.Value.Type.SCALAR) .setScalar(org.apache.mesos.v1.Protos.Value.Scalar.newBuilder().setValue(cpus))) .addResources(org.apache.mesos.v1.Protos.Resource.newBuilder() .setName("mem") .setRole("*") .setType(org.apache.mesos.v1.Protos.Value.Type.SCALAR) .setScalar(org.apache.mesos.v1.Protos.Value.Scalar.newBuilder().setValue(mem))) .addResources(org.apache.mesos.v1.Protos.Resource.newBuilder() .setName("disk") .setRole("*") .setType(org.apache.mesos.v1.Protos.Value.Type.SCALAR) .setScalar(org.apache.mesos.v1.Protos.Value.Scalar.newBuilder().setValue(disk))) .build() ))
.setFrameworkId(org.apache.mesos.v1.Protos.FrameworkID.newBuilder().setValue(frameworkId)) .addResources(org.apache.mesos.v1.Protos.Resource.newBuilder() .setName("cpus") .setRole("*") .setType(org.apache.mesos.v1.Protos.Value.Type.SCALAR) .setScalar(org.apache.mesos.v1.Protos.Value.Scalar.newBuilder().setValue(cpus))) .addResources(org.apache.mesos.v1.Protos.Resource.newBuilder() .setName("mem") .setRole("*") .setType(org.apache.mesos.v1.Protos.Value.Type.SCALAR) .setScalar(org.apache.mesos.v1.Protos.Value.Scalar.newBuilder().setValue(mem))) .addResources(org.apache.mesos.v1.Protos.Resource.newBuilder() .setName("disk") .setRole("*") .setType(org.apache.mesos.v1.Protos.Value.Type.SCALAR) .setScalar(org.apache.mesos.v1.Protos.Value.Scalar.newBuilder().setValue(disk))) .build() ))
Resource.Builder cpusResource = Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.CPUS).setScalar(Scalar.newBuilder().setValue(cpus)); Resource.Builder memoryResources = Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.MEMORY).setScalar(Scalar.newBuilder().setValue(memory)); Resource.Builder diskResources = Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.DISK).setScalar(Scalar.newBuilder().setValue(disk)); if(role.isPresent()) { cpusResource = cpusResource.setRole(role.get()); memoryResources = memoryResources.setRole(role.get()); diskResources = diskResources.setRole(role.get());
public static List<Resource> combineResources(List<List<Resource>> resourcesList) { List<Resource> resources = new ArrayList<>(); for (List<Resource> resourcesToAdd : resourcesList) { for (Resource resource : resourcesToAdd) { Optional<Resource> matched = getMatchingResource(resource, resources); if (!matched.isPresent()) { resources.add(resource); } else { int index = resources.indexOf(matched.get()); Resource.Builder resourceBuilder = resource.toBuilder().clone(); if (resource.hasScalar()) { resourceBuilder.setScalar(resource.toBuilder().getScalarBuilder().setValue(resource.getScalar().getValue() + matched.get().getScalar().getValue()).build()); resources.set(index, resourceBuilder.build()); } else if (resource.hasRanges()) { Ranges.Builder newRanges = Ranges.newBuilder(); resource.getRanges().getRangeList().forEach(newRanges::addRange); matched.get().getRanges().getRangeList().forEach(newRanges::addRange); resourceBuilder.setRanges(newRanges); resources.set(index, resourceBuilder.build()); } else { throw new IllegalStateException(String.format("Can't subtract non-scalar or range resources %s", formatForLogging(resource))); } } } } return resources; }
/** * <pre> * The guarantee that these resources are allocatable for the above role. * NOTE: `guarantee.role` should not specify any role except '*', * because quota does not reserve specific resources. * </pre> * * <code>repeated .mesos.v1.Resource guarantee = 3;</code> */ public Builder addGuarantee( int index, org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (guaranteeBuilder_ == null) { ensureGuaranteeIsMutable(); guarantee_.add(index, builderForValue.build()); onChanged(); } else { guaranteeBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * <pre> * The requested guarantee that these resources will be allocatable for * the above role. * </pre> * * <code>repeated .mesos.v1.Resource guarantee = 3;</code> */ public Builder addGuarantee( org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (guaranteeBuilder_ == null) { ensureGuaranteeIsMutable(); guarantee_.add(builderForValue.build()); onChanged(); } else { guaranteeBuilder_.addMessage(builderForValue.build()); } return this; } /**
/** * <pre> * The requested guarantee that these resources will be allocatable for * the above role. * </pre> * * <code>repeated .mesos.v1.Resource guarantee = 3;</code> */ public Builder setGuarantee( int index, org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (guaranteeBuilder_ == null) { ensureGuaranteeIsMutable(); guarantee_.set(index, builderForValue.build()); onChanged(); } else { guaranteeBuilder_.setMessage(index, builderForValue.build()); } return this; } /**
@Test public void testGetPortByIndex() throws Exception{ taskResources = new Resources(1, 1, 4, 0); final Protos.Resource portsResource = Protos.Resource.newBuilder() .setName("ports") .setType(Protos.Value.Type.RANGES) .setRanges(Protos.Value.Ranges.newBuilder() .addRange( Protos.Value.Range.newBuilder() .setBegin(31003) .setEnd(31006).build()) .build()).build(); final SingularityRequest request = new SingularityRequestBuilder("test", RequestType.WORKER).build(); final SingularityDeploy deploy = new SingularityDeployBuilder("test", "1") .setCommand(Optional.of("/bin/echo")) .setArguments(Optional.of(Collections.singletonList("wat"))) .build(); final SingularityTaskRequest taskRequest = new SingularityTaskRequest(request, deploy, pendingTask); final SingularityMesosTaskHolder task = builder.buildTask(offerHolder, Collections.singletonList(portsResource), taskRequest, taskResources, executorResources); assertEquals(31005L, task.getTask().getPortByIndex(2).get().longValue()); }
/** * <pre> * The total resources provided by this resource provider. * </pre> * * <code>repeated .mesos.v1.Resource resources = 2;</code> */ public Builder addResources( int index, org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (resourcesBuilder_ == null) { ensureResourcesIsMutable(); resources_.add(index, builderForValue.build()); onChanged(); } else { resourcesBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * <pre> * The requested guarantee that these resources will be allocatable for * the above role. * </pre> * * <code>repeated .mesos.v1.Resource guarantee = 3;</code> */ public Builder addGuarantee( int index, org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (guaranteeBuilder_ == null) { ensureGuaranteeIsMutable(); guarantee_.add(index, builderForValue.build()); onChanged(); } else { guaranteeBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * <pre> * The guarantee that these resources are allocatable for the above role. * NOTE: `guarantee.role` should not specify any role except '*', * because quota does not reserve specific resources. * </pre> * * <code>repeated .mesos.v1.Resource guarantee = 3;</code> */ public Builder setGuarantee( int index, org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (guaranteeBuilder_ == null) { ensureGuaranteeIsMutable(); guarantee_.set(index, builderForValue.build()); onChanged(); } else { guaranteeBuilder_.setMessage(index, builderForValue.build()); } return this; } /**
/** * <pre> * The guarantee that these resources are allocatable for the above role. * NOTE: `guarantee.role` should not specify any role except '*', * because quota does not reserve specific resources. * </pre> * * <code>repeated .mesos.v1.Resource guarantee = 3;</code> */ public Builder addGuarantee( org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (guaranteeBuilder_ == null) { ensureGuaranteeIsMutable(); guarantee_.add(builderForValue.build()); onChanged(); } else { guaranteeBuilder_.addMessage(builderForValue.build()); } return this; } /**
/** * <pre> * The total resources provided by this resource provider. * </pre> * * <code>repeated .mesos.v1.Resource resources = 2;</code> */ public Builder addResources( org.apache.mesos.v1.Protos.Resource.Builder builderForValue) { if (resourcesBuilder_ == null) { ensureResourcesIsMutable(); resources_.add(builderForValue.build()); onChanged(); } else { resourcesBuilder_.addMessage(builderForValue.build()); } return this; } /**
@Test public void testResourceAddition() { List<List<Resource>> toAdd = ImmutableList.of( ImmutableList.of( Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.CPUS).setScalar(Scalar.newBuilder().setValue(1)).build(), Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.MEMORY).setScalar(Scalar.newBuilder().setValue(1024)).build() ), ImmutableList.of( Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.CPUS).setScalar(Scalar.newBuilder().setValue(2)).build(), Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.MEMORY).setScalar(Scalar.newBuilder().setValue(1024)).build() ), ImmutableList.of( Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.CPUS).setScalar(Scalar.newBuilder().setValue(3)).build(), Resource.newBuilder().setType(Type.SCALAR).setName(MesosUtils.MEMORY).setScalar(Scalar.newBuilder().setValue(1024)).build() ) ); List<Resource> combined = MesosUtils.combineResources(toAdd); Assert.assertEquals(6, MesosUtils.getNumCpus(combined, Optional.absent()), 0.1); Assert.assertEquals(3072, MesosUtils.getMemory(combined, Optional.absent()), 0.1); }