/** * <code>optional .mesos.SlaveID slave_id = 1;</code> */ public Builder mergeSlaveId(org.apache.mesos.Protos.SlaveID value) { if (slaveIdBuilder_ == null) { if (((bitField0_ & 0x00000001) == 0x00000001) && slaveId_ != null && slaveId_ != org.apache.mesos.Protos.SlaveID.getDefaultInstance()) { slaveId_ = org.apache.mesos.Protos.SlaveID.newBuilder(slaveId_).mergeFrom(value).buildPartial(); } else { slaveId_ = value; } onChanged(); } else { slaveIdBuilder_.mergeFrom(value); } bitField0_ |= 0x00000001; return this; } /**
/** * <code>optional .mesos.SlaveID slave_id = 2;</code> */ public Builder mergeSlaveId(org.apache.mesos.Protos.SlaveID value) { if (slaveIdBuilder_ == null) { if (((bitField0_ & 0x00000002) == 0x00000002) && slaveId_ != null && slaveId_ != org.apache.mesos.Protos.SlaveID.getDefaultInstance()) { slaveId_ = org.apache.mesos.Protos.SlaveID.newBuilder(slaveId_).mergeFrom(value).buildPartial(); } else { slaveId_ = value; } onChanged(); } else { slaveIdBuilder_.mergeFrom(value); } bitField0_ |= 0x00000002; return this; } /**
public void sendFrameworkMessage(String executorId, String slaveId, byte[] data) { driver.get().sendFrameworkMessage(Protos.ExecutorID.newBuilder().setValue(executorId).build(), Protos.SlaveID.newBuilder().setValue(slaveId).build(), data); }
public static Protos.Offer generateReplacementOffer( String frameworkId, Protos.TaskInfo taskInfo, Protos.TaskInfo templateTaskInfo) { final String offerUUID = UUID.randomUUID().toString(); return Protos.Offer .newBuilder() .setId(Protos.OfferID.newBuilder().setValue(offerUUID)) .setFrameworkId(Protos.FrameworkID.newBuilder().setValue(frameworkId)) .setSlaveId(Protos.SlaveID.newBuilder().setValue(taskInfo.getSlaveId().getValue())) .setHostname("127.0.0.1") .addAllResources(taskInfo.getResourcesList()) .addAllResources(taskInfo.getExecutor().getResourcesList()) .addAllResources(templateTaskInfo.getResourcesList()) .build(); }
@Test public void shouldCreateArguments() { Configuration configuration = new Configuration(ZookeeperCLIParameter.ZOOKEEPER_MESOS_URL, "aa"); final ClusterState clusterState = Mockito.mock(ClusterState.class); final int port = 1234; final Protos.DiscoveryInfo discoveryInfo = Protos.DiscoveryInfo.newBuilder().setPorts(Protos.Ports.newBuilder() .addPorts(Protos.Port.newBuilder().setNumber(port)) .addPorts(Protos.Port.newBuilder().setNumber(port))) .setVisibility(Protos.DiscoveryInfo.Visibility.EXTERNAL) .build(); Protos.SlaveID slaveID = Protos.SlaveID.newBuilder().setValue("SLAVE").build(); final List<String> arguments = configuration.esArguments(clusterState, discoveryInfo, slaveID); String allArgs = arguments.toString(); assertTrue(allArgs.contains(Integer.toString(port))); }
private Protos.Offer getTestOffer() { return Protos.Offer.newBuilder() .setId(Protos.OfferID.newBuilder() .setValue("test-offer-id") .build()) .setFrameworkId(Protos.FrameworkID.newBuilder() .setValue("test-framework-id") .build()) .setSlaveId(Protos.SlaveID.newBuilder() .setValue("test-slave-id") .build()) .setHostname(testHostName) .build(); }
public static Protos.SlaveID getSlaveId() { return Protos.SlaveID.newBuilder().setValue(SLAVE_ID).build(); }
private Protos.Offer getOffer(Protos.FrameworkID frameworkId) { return Protos.Offer.newBuilder() .setId(Protos.OfferID.newBuilder().setValue(UUID.randomUUID().toString())) .setSlaveId(Protos.SlaveID.newBuilder().setValue(SLAVEID)) .setFrameworkId(frameworkId) .setHostname("localhost") .addAllResources(asList( Resources.singlePortRange(9200, "some-framework-role"), Resources.singlePortRange(9300, "some-framework-role"), Resources.cpus(1.0, "some-framework-role"), Resources.disk(2.0, "some-framework-role"), Resources.mem(3.0, "some-framework-role"))) .build(); }
private Protos.TaskInfo createTaskInfo(Protos.TaskID taskId, ByteString data) { Protos.DiscoveryInfo.Builder discovery = Protos.DiscoveryInfo.newBuilder() .setPorts(Protos.Ports.newBuilder() .addPorts(Discovery.CLIENT_PORT_INDEX, Protos.Port.newBuilder().setNumber(9001).setName(Discovery.CLIENT_PORT_NAME)) .addPorts(Discovery.TRANSPORT_PORT_INDEX, Protos.Port.newBuilder().setNumber(9002).setName(Discovery.TRANSPORT_PORT_NAME))) .setVisibility(Protos.DiscoveryInfo.Visibility.EXTERNAL); return Protos.TaskInfo.newBuilder() .setName("Name") .setTaskId(taskId) .setSlaveId(Protos.SlaveID.newBuilder().setValue("SlaveID").build()) .setData(data) .setDiscovery(discovery) .build(); }
private static Offer getOffer(String agent) { Offer.Builder o = OfferTestUtils.getEmptyOfferBuilder() .setSlaveId(SlaveID.newBuilder().setValue(agent)); OfferTestUtils.addResource(o, "a"); OfferTestUtils.addResource(o, "b"); return o.build(); }
private void stopExecutors(final String appName) { try { Collection<MesosStateService.ExecutorStateInfo> executorBriefInfo = mesosStateService.executors(appName); for (MesosStateService.ExecutorStateInfo each : executorBriefInfo) { producerManager.sendFrameworkMessage(ExecutorID.newBuilder().setValue(each.getId()).build(), SlaveID.newBuilder().setValue(each.getSlaveId()).build(), "STOP".getBytes()); } } catch (final JSONException ex) { throw new JobSystemException(ex); } } }
public static Protos.TaskInfo createTask(String taskName) { return Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(CommonIdUtils.toTaskId(TestConstants.SERVICE_NAME, taskName)) .setSlaveId(Protos.SlaveID.newBuilder().setValue("ignored")) // proto field required .build(); }
private static Protos.Offer getOffer(int id) { return Protos.Offer.newBuilder() .setId(Protos.OfferID.newBuilder().setValue(Integer.toString(id))) .setFrameworkId(Protos.FrameworkID.newBuilder().setValue("test-framework-id").build()) .setSlaveId(Protos.SlaveID.newBuilder().setValue("test-slave-id").build()) .setHostname("test-hostname") .build(); }
public static Protos.TaskInfo createTask(String taskName) { return Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(CommonIdUtils.toTaskId(TestConstants.SERVICE_NAME, taskName)) .setSlaveId(Protos.SlaveID.newBuilder().setValue("ignored")) // proto field required .build(); }
public static Protos.TaskStatus getDefaultTaskStatus(Protos.TaskState state, Double timestamp) { Protos.SlaveID slaveID = Protos.SlaveID.newBuilder().setValue(SLAVE_ID).build(); Protos.ExecutorID executorID = Protos.ExecutorID.newBuilder().setValue(EXECUTOR_ID).build(); Protos.TaskID taskID = Protos.TaskID.newBuilder().setValue(TASK_ID).build(); return Protos.TaskStatus.newBuilder().setSlaveId(slaveID).setTaskId(taskID).setExecutorId(executorID).setTimestamp(timestamp) .setState(state).build(); }
private Protos.Offer.Builder baseOfferBuilder(String slaveId) { return Protos.Offer.newBuilder() .setId(Protos.OfferID.newBuilder().setValue("offerId").build()) .setFrameworkId(Protos.FrameworkID.newBuilder().setValue("testframework").build()) .setHostname("localhost") .setSlaveId(Protos.SlaveID.newBuilder().setValue(slaveId).build()); }
@Override public Protos.TaskStatus apply(final TaskContext input) { return Protos.TaskStatus.newBuilder() .setTaskId(Protos.TaskID.newBuilder().setValue(input.getId()).build()) .setSlaveId(Protos.SlaveID.newBuilder().setValue(input.getSlaveId()).build()) .setState(Protos.TaskState.TASK_RUNNING).build(); } }));
private static Protos.TaskInfo getTestTaskInfo() { return Protos.TaskInfo.newBuilder() .setName(testTaskName) .setTaskId(Protos.TaskID.newBuilder().setValue(testTaskId)) .setSlaveId(Protos.SlaveID.newBuilder().setValue(testAgentId)) .build(); } }
private Protos.TaskInfo createTask(String hostname) throws InvalidProtocolBufferException { return Protos.TaskInfo.newBuilder() .setName("Test") .setTaskId(Protos.TaskID.newBuilder().setValue("TestId").build()) .setSlaveId(Protos.SlaveID.newBuilder().setValue(hostname).build()) .build(); }
public static Protos.Offer.Builder newOfferBuilder(String offerId, String hostname, String slave, Protos.FrameworkID frameworkID) { Protos.OfferID offerID = Protos.OfferID.newBuilder().setValue(offerId).build(); Protos.SlaveID slaveID = Protos.SlaveID.newBuilder().setValue(slave).build(); return Protos.Offer.newBuilder().setId(offerID).setFrameworkId(frameworkID).setSlaveId(slaveID).setHostname(hostname); }