void resetResultPartitionID(ResultPartitionID[] partitionIds) { Map<IntermediateResultPartitionID, IntermediateResultPartition> newResultPartitions = new LinkedHashMap<>(resultPartitions.size()); Iterator<IntermediateResultPartition> iterator = resultPartitions.values().iterator(); for (int i = 0; i < resultPartitions.size(); i++) { IntermediateResultPartition resultPartition = iterator.next(); IntermediateResultPartitionID originId = resultPartition.getPartitionId(); resultPartition.setPartitionId(partitionIds[i].getPartitionId()); resultPartition.getIntermediateResult().resetLookupHelper(originId, partitionIds[i].getPartitionId()); newResultPartitions.put(resultPartition.getPartitionId(), resultPartition); } this.resultPartitions = newResultPartitions; }