.setSlaveId(offer.getSlaveId()) .setCommand(builder) .addResources(Protos.Resource.newBuilder() .setName(CPU) .setType(Protos.Value.Type.SCALAR) .setScalar(Protos.Value.Scalar.newBuilder().setValue(igniteTask.cpuCores()))) .addResources(Protos.Resource.newBuilder() .setName(MEM) .setType(Protos.Value.Type.SCALAR) .setScalar(Protos.Value.Scalar.newBuilder().setValue(igniteTask.mem()))) .addResources(Protos.Resource.newBuilder() .setName(DISK) .setType(Protos.Value.Type.SCALAR)
static void setProtos(PodInfoBuilder podInfoBuilder, Protos.Resource resource, Optional<String> taskName) { if (taskName.isPresent()) { Protos.TaskInfo.Builder taskBuilder = podInfoBuilder.getTaskBuilder(taskName.get()); taskBuilder.addResources(resource); } else { Protos.ExecutorInfo.Builder executorBuilder = podInfoBuilder.getExecutorBuilder().get(); executorBuilder.addResources(resource); } }
.setTaskId(taskId) .setSlaveId(offer.offer.getSlaveId()) .addResources( Resource.newBuilder().setName("cpus").setType(Value.Type.SCALAR) .setRole(offer.cpuRole) .setScalar(Value.Scalar.newBuilder().setValue(taskCpus))) .addResources( Resource.newBuilder().setName("mem").setType(Value.Type.SCALAR) .setRole(offer.memRole) .setScalar(Value.Scalar.newBuilder().setValue(taskMem))) .addResources( Resource.newBuilder().setName("disk").setType(Value.Type.SCALAR) .setRole(offer.diskRole) .setScalar(Value.Scalar.newBuilder().setValue(taskDisk))) .addResources( Resource .newBuilder()
.setTaskId(taskId) .setSlaveId(offer.getSlaveId()) .addResources( Resource .newBuilder() .setBegin(reportAddress.getPort()) .setEnd(reportAddress.getPort())))) .addResources( Resource .newBuilder() .setRole(cpuRole) .setScalar(Value.Scalar.newBuilder().setValue(taskCpus - containerCpus))) .addResources( Resource .newBuilder()
taskBuilder.addResources( Resource .newBuilder()
taskBuilder.addResources(resource);
.setName(id) .setSlaveId(offer.getSlaveId()) .addResources(Resource.newBuilder() .setName("mem") .setType(Type.SCALAR) .build()) .build()) .addResources(Resource.newBuilder() .setName("cpus") .setType(Type.SCALAR)
if (r.getName().equals("cpus") && cpusNeeded > 0) { double cpus = Math.min(r.getScalar().getValue(), cpusNeeded); builder.addResources( Resource .newBuilder() } else if (r.getName().equals("mem") && memNeeded > 0) { double mem = Math.min(r.getScalar().getValue(), memNeeded); builder.addResources( Resource .newBuilder() } else if (r.getName().equals("disk") && diskNeeded > 0) { double disk = Math.min(r.getScalar().getValue(), diskNeeded); builder.addResources( Resource .newBuilder()
builder.addResources(ResourceUtils.getDesiredMountVolume(role, principal, diskMb, CassandraConfig.VOLUME_PATH)); } else { builder.addResources(ResourceUtils.getDesiredRootVolume(role, principal, diskMb, CassandraConfig.VOLUME_PATH)); builder.addResources(ResourceUtils.getDesiredRanges(role, principal, "ports", Algorithms.createRanges(ports)));
.setTaskId(taskId) .setSlaveId(offer.getSlaveId()) .addResources( Resource .newBuilder() Value.Scalar.newBuilder() .setValue(request.request.getCpus()).build()).build()) .addResources( Resource .newBuilder()
.setTaskId(taskId) .setSlaveId(offer.getSlaveId()) .addResources(Protos.Resource.newBuilder() .setName("cpus") .setType(Protos.Value.Type.SCALAR) .setScalar(Protos.Value.Scalar.newBuilder().setValue(1))) .addResources(Protos.Resource.newBuilder() .setName("mem") .setType(Protos.Value.Type.SCALAR)
private static Protos.TaskInfo getTask( String podType, int podIndex, String taskName, int resourceCount) { String fullTaskName = String.format("%s-%s", PodInstance.getName(podType, podIndex), taskName); Protos.TaskInfo.Builder builder = Protos.TaskInfo.newBuilder() .setTaskId(TestConstants.TASK_ID) .setName(fullTaskName) .setSlaveId(TestConstants.AGENT_ID); builder.setLabels(new TaskLabelWriter(builder) .setType(podType) .setIndex(podIndex) .toProto()); for (int i = 0; i < resourceCount; ++i) { builder.addResources(ResourceTestUtils.getReservedCpus(5, String.format("%s-resource%d", fullTaskName, i))); } return builder.build(); } }
private Protos.TaskInfo buildCustomizedExecutorTaskInfo(final TaskContext taskContext, final CloudAppConfiguration appConfig, final CloudJobConfiguration jobConfig, final ShardingContexts shardingContexts, final Protos.Offer offer, final Protos.CommandInfo command) { Protos.TaskInfo.Builder result = Protos.TaskInfo.newBuilder().setTaskId(Protos.TaskID.newBuilder().setValue(taskContext.getId()).build()) .setName(taskContext.getTaskName()).setSlaveId(offer.getSlaveId()) .addResources(buildResource("cpus", jobConfig.getCpuCount(), offer.getResourcesList())) .addResources(buildResource("mem", jobConfig.getMemoryMB(), offer.getResourcesList())) .setData(ByteString.copyFrom(new TaskInfoData(shardingContexts, jobConfig).serialize())); Protos.ExecutorInfo.Builder executorBuilder = Protos.ExecutorInfo.newBuilder().setExecutorId(Protos.ExecutorID.newBuilder() .setValue(taskContext.getExecutorId(jobConfig.getAppName()))).setCommand(command) .addResources(buildResource("cpus", appConfig.getCpuCount(), offer.getResourcesList())) .addResources(buildResource("mem", appConfig.getMemoryMB(), offer.getResourcesList())); if (env.getJobEventRdbConfiguration().isPresent()) { executorBuilder.setData(ByteString.copyFrom(SerializationUtils.serialize(env.getJobEventRdbConfigurationMap()))).build(); } return result.setExecutor(executorBuilder.build()).build(); }
private Protos.TaskInfo buildCommandExecutorTaskInfo(final TaskContext taskContext, final CloudJobConfiguration jobConfig, final ShardingContexts shardingContexts, final Protos.Offer offer, final Protos.CommandInfo command) { Protos.TaskInfo.Builder result = Protos.TaskInfo.newBuilder().setTaskId(Protos.TaskID.newBuilder().setValue(taskContext.getId()).build()) .setName(taskContext.getTaskName()).setSlaveId(offer.getSlaveId()) .addResources(buildResource("cpus", jobConfig.getCpuCount(), offer.getResourcesList())) .addResources(buildResource("mem", jobConfig.getMemoryMB(), offer.getResourcesList())) .setData(ByteString.copyFrom(new TaskInfoData(shardingContexts, jobConfig).serialize())); return result.setCommand(command).build(); }