Skip to content

[SPARK-24107][CORE] ChunkedByteBuffer.writeFully method has not reset the limit value #21175

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
jinhai-cloud wants to merge 11 commits into from
Closed

[SPARK-24107][CORE] ChunkedByteBuffer.writeFully method has not reset the limit value #21175

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
fae1814
restore bytes limit value
Apr 26, 2018
623f26d
add ChunkedByteBufferSuite unit test
Apr 27, 2018
5ba6867
Merge branch 'master' into bugfix-ChunkedByteBuffer
Apr 27, 2018
c585131
ChunkedByteBufferSuite improve unit test
Apr 27, 2018
a2a82f1
change assert size
Apr 27, 2018
217ec9d
add SPARK-24107:
Apr 27, 2018
2bc19a3
change ByteArrayWritableChannel's size
Apr 27, 2018
c9a6816
add configure bufferWriteChunkSize
Apr 27, 2018
fa99a19
sc.conf.set spark.buffer.write.chunkSize
Apr 27, 2018
fb527c8
add try finally
Apr 27, 2018
e78ef39
fixed the style
May 2, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion core/src/main/scala/org/apache/spark/util/io/ChunkedByteBuffer.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,10 +63,12 @@ private[spark] class ChunkedByteBuffer(var chunks: Array[ByteBuffer]) {
*/
def writeFully(channel: WritableByteChannel): Unit = {
for (bytes <- getChunks()) {
while (bytes.remaining() > 0) {
val curChunkLimit = bytes.limit()
while (bytes.hasRemaining) {
val ioSize = Math.min(bytes.remaining(), bufferWriteChunkSize)
bytes.limit(bytes.position() + ioSize)
channel.write(bytes)
bytes.limit(curChunkLimit)
Copy link
Contributor

@jiangxb1987 jiangxb1987 Apr 27, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would rewrite this using:

try {
	val ioSize = Math.min(bytes.remaining(), bufferWriteChunkSize)
	bytes.limit(bytes.position() + ioSize)
	channel.write(bytes)
} finally {
	bytes.limit(curChunkLimit)
}

to be safe.

Copy link
Contributor Author

@jinhai-cloud jinhai-cloud Apr 27, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Right. When channel write throw IOException

Copy link
Contributor Author

@jinhai-cloud jinhai-cloud Apr 27, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I have commit this modified

}
}
}
Expand Down
13 changes: 11 additions & 2 deletions core/src/test/scala/org/apache/spark/io/ChunkedByteBufferSuite.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@ package org.apache.spark.io
import java.nio.ByteBuffer

import com.google.common.io.ByteStreams
Copy link
Member

@xuanyuanking xuanyuanking Apr 29, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add an empty line behind 22 to separate spark and third-party group.


import org.apache.spark.SparkFunSuite
import org.apache.spark.{SparkEnv, SparkFunSuite}
import org.apache.spark.internal.config
import org.apache.spark.network.util.ByteArrayWritableChannel
import org.apache.spark.util.io.ChunkedByteBuffer

Expand Down Expand Up @@ -56,6 +56,15 @@ class ChunkedByteBufferSuite extends SparkFunSuite {
assert(chunkedByteBuffer.getChunks().head.position() === 0)
}

test("SPARK-24107: writeFully() write buffer which is larger than bufferWriteChunkSize") {
val bufferWriteChunkSize = Option(SparkEnv.get).map(_.conf.get(config.BUFFER_WRITE_CHUNK_SIZE))
.getOrElse(config.BUFFER_WRITE_CHUNK_SIZE.defaultValue.get).toInt
Copy link
Member

@maropu maropu Apr 27, 2018 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

How about setting this value via spark.buffer.write.chunkSize? e.g.,

spark/core/src/test/scala/org/apache/spark/rdd/PairRDDFunctionsSuite.scala

Line 337 in 109935f

sc.conf.set("spark.default.parallelism", "4")

Copy link
Contributor Author

@jinhai-cloud jinhai-cloud Apr 27, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ok. I have added. Please check

val chunkedByteBuffer = new ChunkedByteBuffer(Array(ByteBuffer.allocate(bufferWriteChunkSize + 8)))
val byteArrayWritableChannel = new ByteArrayWritableChannel(chunkedByteBuffer.size.toInt)
chunkedByteBuffer.writeFully(byteArrayWritableChannel)
assert(byteArrayWritableChannel.length() === chunkedByteBuffer.size)
}

test("toArray()") {
val empty = ByteBuffer.wrap(Array.empty[Byte])
val bytes = ByteBuffer.wrap(Array.tabulate(8)(_.toByte))
Expand Down