{
checkNotNull(sourceId, "sourceId is null");
checkNotNull(split, "split is null");
if (split instanceof CollocatedSplit) {
CollocatedSplit collocatedSplit = (CollocatedSplit) split;
// unwind collocated splits
for (Entry<PlanNodeId, Split> entry : collocatedSplit.getSplits().entrySet()) {
addSplit(entry.getKey(), entry.getValue());
}
}
else {
SourceOperator sourceOperator = sourceOperators.get(sourceId);