@Override
public void launchTask(final ExecutorDriver executorDriver, final Protos.TaskInfo taskInfo) {
LOGGER.info("Launching task in PinUserProfileExecutor ...");
Protos.TaskStatus taskStatus =
Protos.TaskStatus.newBuilder()
.setTaskId(taskInfo.getTaskId())
.setState(Protos.TaskState.TASK_RUNNING)
.build();
executorDriver.sendStatusUpdate(taskStatus);
String url = taskInfo.getData().toStringUtf8();
byte[] message = new byte[0];
try {
message = ("userprofile :" + getUserProfileInfo(url)).getBytes();
} catch (IOException e) {
LOGGER.error("Error parsing the Pinterest URL :" + e.getMessage());
}
LOGGER.info("Sending framework message and marking task finished." + getClass().getName());
executorDriver.sendFrameworkMessage(message);
taskStatus =
Protos.TaskStatus.newBuilder()
.setTaskId(taskInfo.getTaskId())
.setState(Protos.TaskState.TASK_FINISHED)
.build();
executorDriver.sendStatusUpdate(taskStatus);
}
private boolean isHostAlreadyRunningTask(Protos.Offer offer) {
Boolean result = false;
List<Protos.TaskInfo> stateList = clusterState.getTaskList();
for (Protos.TaskInfo t : stateList) {
if (t.getSlaveId().equals(offer.getSlaveId())) {
result = true;
}
}
return result;
}
private static double getDiskFrom(Protos.TaskInfo taskInfo) {
for (Protos.Resource resource : taskInfo.getResourcesList()) {
if (resource.getName() == "disk") {
return resource.getScalar().getValue();
}
}
return -1;
}
/**
* Construct the Mesos TaskInfo in Protos to launch basing on the LaunchableTask
*
* @param heronConfig the heron config
* @param heronRuntime the heron runtime
* @return Mesos TaskInfo in Protos to launch
*/
public Protos.TaskInfo constructMesosTaskInfo(Config heronConfig, Config heronRuntime) {
// String taskIdStr, BaseContainer task, Offer offer
String taskIdStr = this.taskId;
Protos.TaskID mesosTaskID = Protos.TaskID.newBuilder().setValue(taskIdStr).build();
Protos.TaskInfo.Builder taskInfo =
Protos.TaskInfo.newBuilder().setName(baseContainer.name).setTaskId(mesosTaskID);
Protos.Environment.Builder environment = Protos.Environment.newBuilder();
// If the job defines custom environment variables, add them to the builder
// Don't add them if they already exist to prevent overwriting the defaults
Set<String> builtinEnvNames = new HashSet<>();
for (Protos.Environment.Variable variable : environment.getVariablesList()) {
builtinEnvNames.add(variable.getName());
}
for (BaseContainer.EnvironmentVariable ev : baseContainer.environmentVariables) {
environment.addVariables(
Protos.Environment.Variable.newBuilder().setName(ev.name).setValue(ev.value));
}
taskInfo
.addResources(scalarResource(TaskResources.CPUS_RESOURCE_NAME, baseContainer.cpu))
.addResources(scalarResource(TaskResources.MEM_RESOURCE_NAME, baseContainer.memInMB))
.addResources(scalarResource(TaskResources.DISK_RESOURCE_NAME, baseContainer.diskInMB))
.addResources(
rangeResource(
TaskResources.PORT_RESOURCE_NAME,
this.freePorts.get(0),
this.freePorts.get(this.freePorts.size() - 1)))
.setSlaveId(this.offer.getSlaveId());
int containerIndex = TaskUtils.getContainerIndexForTaskId(taskIdStr);
String commandStr = executorCommand(heronConfig, heronRuntime, containerIndex);
Protos.CommandInfo.Builder command = Protos.CommandInfo.newBuilder();
List<Protos.CommandInfo.URI> uriProtos = new ArrayList<>();
for (String uri : baseContainer.dependencies) {
uriProtos.add(Protos.CommandInfo.URI.newBuilder().setValue(uri).setExtract(true).build());
}
command
.setValue(commandStr)
.setShell(baseContainer.shell)
.setEnvironment(environment)
.addAllUris(uriProtos);
if (!baseContainer.runAsUser.isEmpty()) {
command.setUser(baseContainer.runAsUser);
}
taskInfo.setCommand(command);
return taskInfo.build();
}
private static int getPortsFrom(Protos.TaskInfo taskInfo) {
double portCount = 0;
for (Protos.Resource resource : taskInfo.getResourcesList()) {
if (resource.getName() == "ports") {
for (Protos.Value.Range range : resource.getRanges().getRangeList()) {
portCount = 1 + range.getEnd() - range.getBegin();
}
}
}
return (int) portCount;
}
@Test
public void willLaunchTaskFromValidOffer() throws Exception {
Protos.Offer offer = TestHelper.createDummyOffer();
Protos.TaskInfo task =
TestHelper.createDummyTask(
"task", builder -> builder.setTaskId(Protos.TaskID.newBuilder().setValue(taskId)));
when(uuidSupplier.get()).thenReturn(uuid);
OfferEvaluation offerEvaluation =
OfferEvaluation.accept(
"test",
taskId,
offer,
Collections.emptyMap(),
Collections.emptyList(),
Collections.emptyList());
when(offerStrategyFilter.evaluate(taskId, offer)).thenReturn(offerEvaluation);
when(taskMaterializer.createProposal(offerEvaluation))
.thenReturn(new TaskProposal(offer, task));
when(taskInfoFactory.create(
taskId,
offer,
Collections.emptyList(),
new ExecutionParameters(
Collections.emptyMap(), Collections.emptyList(), Collections.emptyList())))
.thenReturn(task);
scheduler.resourceOffers(schedulerDriver, Collections.singletonList(offer));
verify(schedulerDriver, never()).declineOffer(any(Protos.OfferID.class));
verify(schedulerDriver)
.launchTasks(Collections.singleton(offer.getId()), Collections.singleton(task));
ArgumentCaptor<Protos.TaskInfo> taskInfoArgumentCaptor =
ArgumentCaptor.forClass(Protos.TaskInfo.class);
verify(stateRepository).store(taskInfoArgumentCaptor.capture());
Protos.TaskInfo taskInfo = taskInfoArgumentCaptor.getValue();
assertEquals(taskId, taskInfo.getTaskId().getValue());
}
/**
* Accept offers as advised by the launch coordinator.
*
* <p>Acceptance is routed through the RM to update the persistent state before forwarding the
* message to Mesos.
*/
private void acceptOffers(AcceptOffers msg) {
try {
List<TaskMonitor.TaskGoalStateUpdated> toMonitor = new ArrayList<>(msg.operations().size());
// transition the persistent state of some tasks to Launched
for (Protos.Offer.Operation op : msg.operations()) {
if (op.getType() != Protos.Offer.Operation.Type.LAUNCH) {
continue;
}
for (Protos.TaskInfo info : op.getLaunch().getTaskInfosList()) {
MesosWorkerStore.Worker worker = workersInNew.remove(extractResourceID(info.getTaskId()));
assert (worker != null);
worker = worker.launchWorker(info.getSlaveId(), msg.hostname());
workerStore.putWorker(worker);
workersInLaunch.put(extractResourceID(worker.taskID()), worker);
LOG.info(
"Launching Mesos task {} on host {}.",
worker.taskID().getValue(),
worker.hostname().get());
toMonitor.add(new TaskMonitor.TaskGoalStateUpdated(extractGoalState(worker)));
}
}
// tell the task router about the new plans
for (TaskMonitor.TaskGoalStateUpdated update : toMonitor) {
taskRouter.tell(update, self());
}
// send the acceptance message to Mesos
schedulerDriver.acceptOffers(msg.offerIds(), msg.operations(), msg.filters());
} catch (Exception ex) {
fatalError("unable to accept offers", ex);
}
}
@Override
public void resourceOffers(SchedulerDriver schedulerDriver, List<Protos.Offer> list) {
double CPUS_PER_TASK = 1;
double MEM_PER_TASK = 128;
for (Protos.Offer offer : list) {
List<Protos.TaskInfo> taskInfoList = new ArrayList<Protos.TaskInfo>();
double offerCpus = 0;
double offerMem = 0;
for (Protos.Resource resource : offer.getResourcesList()) {
if (resource.getName().equals("cpus")) {
offerCpus += resource.getScalar().getValue();
} else if (resource.getName().equals("mem")) {
offerMem += resource.getScalar().getValue();
}
}
LOGGER.info(
"Received Offer : "
+ offer.getId().getValue()
+ " with cpus = "
+ offerCpus
+ " and mem ="
+ offerMem);
double remainingCpus = offerCpus;
double remainingMem = offerMem;
if (launchedTasks < totalTasks
&& remainingCpus >= CPUS_PER_TASK
&& remainingMem >= MEM_PER_TASK) {
Protos.TaskID taskID =
Protos.TaskID.newBuilder().setValue(Integer.toString(launchedTasks++)).build();
LOGGER.info(
"Launching task :"
+ taskID.getValue()
+ " using the offer : "
+ offer.getId().getValue());
Protos.TaskInfo piTaskInfo =
Protos.TaskInfo.newBuilder()
.setName("task " + taskID.getValue())
.setTaskId(taskID)
.setSlaveId(offer.getSlaveId())
.addResources(
Protos.Resource.newBuilder()
.setName("cpus")
.setType(Protos.Value.Type.SCALAR)
.setScalar(Protos.Value.Scalar.newBuilder().setValue(CPUS_PER_TASK)))
.addResources(
Protos.Resource.newBuilder()
.setName("mem")
.setType(Protos.Value.Type.SCALAR)
.setScalar(Protos.Value.Scalar.newBuilder().setValue(MEM_PER_TASK)))
.setExecutor(Protos.ExecutorInfo.newBuilder(piExecutor))
.build();
taskID = Protos.TaskID.newBuilder().setValue(Integer.toString(launchedTasks++)).build();
LOGGER.info(
"Launching task :"
+ taskID.getValue()
+ " using the offer : "
+ offer.getId().getValue());
taskInfoList.add(piTaskInfo);
}
schedulerDriver.launchTasks(offer.getId(), taskInfoList);
}
}
@Override
public void resourceOffers(SchedulerDriver driver, List<Protos.Offer> offers) {
long masterCpu = Configuration.getInt(PropertyKey.INTEGRATION_MASTER_RESOURCE_CPU);
long masterMem =
Configuration.getBytes(PropertyKey.INTEGRATION_MASTER_RESOURCE_MEM) / Constants.MB;
long workerCpu = Configuration.getInt(PropertyKey.INTEGRATION_WORKER_RESOURCE_CPU);
long workerMem =
Configuration.getBytes(PropertyKey.INTEGRATION_WORKER_RESOURCE_MEM) / Constants.MB;
LOG.info(
"Master launched {}, master count {}, "
+ "requested master cpu {} mem {} MB and required master hostname {}",
mMasterLaunched,
mMasterCount,
masterCpu,
masterMem,
mRequiredMasterHostname);
for (Protos.Offer offer : offers) {
Protos.Offer.Operation.Launch.Builder launch = Protos.Offer.Operation.Launch.newBuilder();
double offerCpu = 0;
double offerMem = 0;
for (Protos.Resource resource : offer.getResourcesList()) {
if (resource.getName().equals(Constants.MESOS_RESOURCE_CPUS)) {
offerCpu += resource.getScalar().getValue();
} else if (resource.getName().equals(Constants.MESOS_RESOURCE_MEM)) {
offerMem += resource.getScalar().getValue();
} else {
// Other resources are currently ignored.
}
}
LOG.info(
"Received offer {} on host {} with cpus {} and mem {} MB and hasMasterPorts {}",
offer.getId().getValue(),
offer.getHostname(),
offerCpu,
offerMem,
OfferUtils.hasAvailableMasterPorts(offer));
Protos.ExecutorInfo.Builder executorBuilder = Protos.ExecutorInfo.newBuilder();
List<Protos.Resource> resources;
if (!mMasterLaunched
&& offerCpu >= masterCpu
&& offerMem >= masterMem
&& mMasterCount
< Configuration.getInt(PropertyKey.INTEGRATION_MESOS_ALLUXIO_MASTER_NODE_COUNT)
&& OfferUtils.hasAvailableMasterPorts(offer)
&& (mRequiredMasterHostname == null
|| mRequiredMasterHostname.equals(offer.getHostname()))) {
LOG.debug("Creating Alluxio Master executor");
executorBuilder
.setName("Alluxio Master Executor")
.setSource("master")
.setExecutorId(Protos.ExecutorID.newBuilder().setValue("master"))
.addAllResources(getExecutorResources())
.setCommand(
Protos.CommandInfo.newBuilder()
.setValue(createStartAlluxioCommand("alluxio-master-mesos.sh"))
.addAllUris(getExecutorDependencyURIList())
.setEnvironment(
Protos.Environment.newBuilder()
.addVariables(
Protos.Environment.Variable.newBuilder()
.setName("ALLUXIO_UNDERFS_ADDRESS")
.setValue(Configuration.get(PropertyKey.UNDERFS_ADDRESS))
.build())
.build()));
// pre-build resource list here, then use it to build Protos.Task later.
resources = getMasterRequiredResources(masterCpu, masterMem);
mMasterHostname = offer.getHostname();
mTaskName = Configuration.get(PropertyKey.INTEGRATION_MESOS_ALLUXIO_MASTER_NAME);
mMasterCount++;
mMasterTaskId = mLaunchedTasks;
} else if (mMasterLaunched
&& !mWorkers.contains(offer.getHostname())
&& offerCpu >= workerCpu
&& offerMem >= workerMem
&& OfferUtils.hasAvailableWorkerPorts(offer)) {
LOG.debug("Creating Alluxio Worker executor");
final String memSize = FormatUtils.getSizeFromBytes((long) workerMem * Constants.MB);
executorBuilder
.setName("Alluxio Worker Executor")
.setSource("worker")
.setExecutorId(Protos.ExecutorID.newBuilder().setValue("worker"))
.addAllResources(getExecutorResources())
.setCommand(
Protos.CommandInfo.newBuilder()
.setValue(createStartAlluxioCommand("alluxio-worker-mesos.sh"))
.addAllUris(getExecutorDependencyURIList())
.setEnvironment(
Protos.Environment.newBuilder()
.addVariables(
Protos.Environment.Variable.newBuilder()
.setName("ALLUXIO_MASTER_HOSTNAME")
.setValue(mMasterHostname)
.build())
.addVariables(
Protos.Environment.Variable.newBuilder()
.setName("ALLUXIO_WORKER_MEMORY_SIZE")
.setValue(memSize)
.build())
.addVariables(
Protos.Environment.Variable.newBuilder()
.setName("ALLUXIO_UNDERFS_ADDRESS")
.setValue(Configuration.get(PropertyKey.UNDERFS_ADDRESS))
.build())
.build()));
// pre-build resource list here, then use it to build Protos.Task later.
resources = getWorkerRequiredResources(workerCpu, workerMem);
mWorkers.add(offer.getHostname());
mTaskName = Configuration.get(PropertyKey.INTEGRATION_MESOS_ALLUXIO_WORKER_NAME);
} else {
// The resource offer cannot be used to start either master or a worker.
LOG.info("Declining offer {}", offer.getId().getValue());
driver.declineOffer(offer.getId());
continue;
}
Protos.TaskID taskId =
Protos.TaskID.newBuilder().setValue(String.valueOf(mLaunchedTasks)).build();
LOG.info("Launching task {} using offer {}", taskId.getValue(), offer.getId().getValue());
Protos.TaskInfo task =
Protos.TaskInfo.newBuilder()
.setName(mTaskName)
.setTaskId(taskId)
.setSlaveId(offer.getSlaveId())
.addAllResources(resources)
.setExecutor(executorBuilder)
.build();
launch.addTaskInfos(Protos.TaskInfo.newBuilder(task));
mLaunchedTasks++;
// NOTE: We use the new API `acceptOffers` here to launch tasks.
// The 'launchTasks' API will be deprecated.
List<Protos.OfferID> offerIds = new ArrayList<Protos.OfferID>();
offerIds.add(offer.getId());
List<Protos.Offer.Operation> operations = new ArrayList<Protos.Offer.Operation>();
Protos.Offer.Operation operation =
Protos.Offer.Operation.newBuilder()
.setType(Protos.Offer.Operation.Type.LAUNCH)
.setLaunch(launch)
.build();
operations.add(operation);
Protos.Filters filters = Protos.Filters.newBuilder().setRefuseSeconds(1).build();
driver.acceptOffers(offerIds, operations, filters);
}
}