Skip to content

Commit 9336c15

Browse files
committed
Disable use DownloadCallback fetch big blocks
1 parent d7da2b9 commit 9336c15

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

common/network-shuffle/src/main/java/org/apache/spark/network/shuffle/OneForOneBlockFetcher.java

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -119,12 +119,7 @@ public void onSuccess(ByteBuffer response) {
119119
// Immediately request all chunks -- we expect that the total size of the request is
120120
// reasonable due to higher level chunking in [[ShuffleBlockFetcherIterator]].
121121
for (int i = 0; i < streamHandle.numChunks; i++) {
122-
if (shuffleFiles != null) {
123-
client.stream(OneForOneStreamManager.genStreamChunkId(streamHandle.streamId, i),
124-
new DownloadCallback(shuffleFiles[i], i));
125-
} else {
126-
client.fetchChunk(streamHandle.streamId, i, chunkCallback);
127-
}
122+
client.fetchChunk(streamHandle.streamId, i, chunkCallback);
128123
}
129124
} catch (Exception e) {
130125
logger.error("Failed while starting block fetches after success", e);

0 commit comments

Comments
 (0)