diff --git a/llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/helpers/SourceStateTracker.java b/llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/helpers/SourceStateTracker.java index 40b317d..5428305 100644 --- a/llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/helpers/SourceStateTracker.java +++ b/llap-server/src/java/org/apache/hadoop/hive/llap/tezplugins/helpers/SourceStateTracker.java @@ -33,7 +33,9 @@ import org.apache.hadoop.hive.llap.tezplugins.LlapTaskCommunicator; import org.apache.tez.dag.api.TaskCommunicatorContext; import org.apache.tez.dag.api.event.VertexState; +import org.apache.tez.mapreduce.input.MRInput; import org.apache.tez.mapreduce.input.MRInputLegacy; +import org.apache.tez.mapreduce.input.MultiMRInput; import org.apache.tez.runtime.api.impl.InputSpec; public class SourceStateTracker { @@ -274,7 +276,8 @@ private void maybeRegisterForVertexUpdates(String sourceName) { private boolean isSourceOfInterest(InputSpec inputSpec) { String inputClassName = inputSpec.getInputDescriptor().getClassName(); // MRInput is not of interest since it'll always be ready. - return !inputClassName.equals(MRInputLegacy.class.getName()); + return !(inputClassName.equals(MRInputLegacy.class.getName()) || inputClassName.equals( + MultiMRInput.class.getName()) || inputClassName.equals(MRInput.class.getName())); } void sendStateUpdateToNode(LlapNodeId nodeId, String sourceName, VertexState state) {