public synchronized void resetState(QueryIdentifierProto currentQueryIdentifierProto) {
    sourceInfoMap.clear();
    nodeInfoMap.clear();
    notificationRegisteredVertices.clear();
    this.currentQueryIdentifier = currentQueryIdentifierProto;
  }
To be invoked after each DAG completes.
llap-tez/src/java/org/apache/hadoop/hive/llap/tezplugins/helpers/SourceStateTracker.java