diff --git llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/LlapTaskCommunicator.java llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/LlapTaskCommunicator.java index 2304aeb..33003f7 100644 --- llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/LlapTaskCommunicator.java +++ llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/LlapTaskCommunicator.java @@ -103,9 +103,10 @@ public void registerContainerEnd(ContainerId containerId) { public void registerRunningTaskAttempt(final ContainerId containerId, final TaskSpec taskSpec, Map additionalResources, Credentials credentials, - boolean credentialsChanged) { + boolean credentialsChanged, + int priority) { super.registerRunningTaskAttempt(containerId, taskSpec, additionalResources, credentials, - credentialsChanged); + credentialsChanged, priority); SubmitWorkRequestProto requestProto = null; try { requestProto = constructSubmitWorkRequest(containerId, taskSpec); diff --git llap-server/src/java/org/apache/tez/dag/app/rm/LlapTaskSchedulerService.java llap-server/src/java/org/apache/tez/dag/app/rm/LlapTaskSchedulerService.java index 1c417cf..0af5ea4 100644 --- llap-server/src/java/org/apache/tez/dag/app/rm/LlapTaskSchedulerService.java +++ llap-server/src/java/org/apache/tez/dag/app/rm/LlapTaskSchedulerService.java @@ -46,6 +46,7 @@ import org.apache.hadoop.yarn.client.api.YarnClient; import org.apache.hadoop.yarn.exceptions.YarnException; import org.apache.hadoop.hive.llap.daemon.LlapDaemonConfiguration; +import org.apache.tez.dag.api.TaskAttemptEndReason; import org.apache.tez.dag.app.AppContext; @@ -241,7 +242,7 @@ public void allocateTask(Object task, Resource capability, ContainerId container } @Override - public boolean deallocateTask(Object task, boolean taskSucceeded) { + public boolean deallocateTask(Object task, boolean taskSucceeded, TaskAttemptEndReason endReason) { ContainerId containerId = runningTasks.remove(task); if (containerId == null) { LOG.error("Could not determine ContainerId for task: " + task +