incubator-uniffle
incubator-uniffle copied to clipboard
[Improvement][Aqe] Avoid calling `getShuffleResult` multiple times
When we use AQE, we may call shuffleWriteClient.getShuffleResult multiple times. But if both partition 1 and partition 2 are on the server A, we call getShuffleResult(partition 1) to get data form server A, and then we call getShuffleResult(partition 2) to get data form server A, it's not necassray. We can get getShuffleResult(partition 1, partition 2) instead.