arrow

ARROW-12334: [Rust] [Ballista] Aggregate queries producing incorrect results

The function that calculated job status from the task status was aggregating all of the partition locations. This is incorrect. Only the partitions of the last stage should be collected.

@andygrove, could you confirm that my assumption that the output of any query is contained in the last stage? If there is a possibility of having multiple output stages, then this fix is incorrect.

Closes #10083 from edrevo/fix-shuffle-reads

Authored-by: Ximo Guanter <ximo.guanter@gmail.com>
Signed-off-by: Andy Grove <andygrove73@gmail.com>